Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import csv
+import base64
 
 # Define the state populations and family sizes
 state_data = {
@@ -50,21 +51,22 @@ def main():
         writer.writeheader()
         for state in state_list:
             writer.writerow(state)
+    with open('state_data.csv', mode='rb') as file:
+        b64 = base64.b64encode(file.read()).decode('utf-8')
     st.markdown(f'<a href="data:file/csv;base64,{b64}" download="state_data.csv">Download State Data CSV File</a>', unsafe_allow_html=True)
 
     # Display state populations and family sizes
     st.header('Population and Family Size')
     for state, data in state_data.items():
         st.subheader(f'{POPULATION_ICON} {state}')
-        st.write(f'Population: {data["population"]}')
-        st.write(f'Family Size: {data["family_size"]}')
+    st.write(f'Population: {data["population"]}')
+    st.write(f'Family Size: {data["family_size"]}')
 
-    # Display state spending data
-    st.header('State Spending')
-    for state, data in spending_data.items():
-        st.subheader(state)
-        st.write(f'{EDUCATION_ICON} Education: {data["education"]}')
-        st.write(f'{HEALTHCARE_ICON} Healthcare: {data["healthcare"]}')
-        st.write(f'{TRANSPORTATION_ICON} Transportation: {data["transportation"]}')
+# Display state spending data
+st.header('State Spending')
+for state, data in spending_data.items():
+    st.subheader(state)
+    st.write(f'{EDUCATION_ICON} Education: {data["education"]}')
+    st.write(f'{HEALTHCARE_ICON} Healthcare: {data["healthcare"]}')
+    st.write(f'{TRANSPORTATION_ICON} Transportation: {data["transportation"]}')
 
-main()