Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import csv
 
 # Define the state populations and family sizes
 state_data = {
@@ -29,22 +30,35 @@ TRANSPORTATION_ICON = '🚗'
 
 def main():
     st.title('State Comparison')
-    
+
+    # Consolidate the state data and spending data into a list of dictionaries
+    state_list = []
+    for state, data in state_data.items():
+        state_dict = {
+            'state': state,
+            'population': data['population'],
+            'family_size': data['family_size'],
+            'education_spending': spending_data[state]['education'],
+            'healthcare_spending': spending_data[state]['healthcare'],
+            'transportation_spending': spending_data[state]['transportation']
+        }
+        state_list.append(state_dict)
+
+    # Save the data to a CSV file and provide a download link
+    with open('state_data.csv', mode='w', newline='') as file:
+        writer = csv.DictWriter(file, fieldnames=['state', 'population', 'family_size', 'education_spending', 'healthcare_spending', 'transportation_spending'])
+        writer.writeheader()
+        for state in state_list:
+            writer.writerow(state)
+    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"]}')
-    
+
     # 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"]}')
-        
-
-if __name__ == '__main__':
-    main()
+    for state