Update app.py

app.py

@@ -1,6 +1,33 @@
 import streamlit as st
-import folium
-import numpy as np
+import csv
+import base64
+
+# Define the state populations and family sizes
+state_data = {
+    'California': {'population': 39538223, 'family_size': 3.3},
+    'Texas': {'population': 29145505, 'family_size': 3.4},
+    'Florida': {'population': 21538187, 'family_size': 3.0},
+    'New York': {'population': 19849399, 'family_size': 3.1},
+    'Minnesota': {'population': 5700671, 'family_size': 2.5},
+    'Wisconsin': {'population': 5897473, 'family_size': 2.6},
+}
+
+# Define the state spending data
+spending_data = {
+    'California': {'education': 2500, 'healthcare': 3000, 'transportation': 1500},
+    'Texas': {'education': 2000, 'healthcare': 2500, 'transportation': 1000},
+    'Florida': {'education': 1500, 'healthcare': 2000, 'transportation': 750},
+    'New York': {'education': 3000, 'healthcare': 3500, 'transportation': 2000},
+    'Minnesota': {'education': 1000, 'healthcare': 1500, 'transportation': 500},
+    'Wisconsin': {'education': 1250, 'healthcare': 1750, 'transportation': 750},
+}
+
+# Define the emoji icons
+POPULATION_ICON = '👥'
+FAMILY_SIZE_ICON = '👨‍👩‍👧‍👦'
+EDUCATION_ICON = '🏫'
+HEALTHCARE_ICON = '🏥'
+TRANSPORTATION_ICON = '🚗'
 
 def main():
     st.title('State Comparison')
@@ -24,33 +51,21 @@ 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)
-
-    # Create a map with pie charts for each state
-    m = folium.Map(location=[37.0902, -95.7129], zoom_start=4)
+    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():
-        lat, lon = get_coordinates(state)
-        popup_text = f'<b>{state}</b><br>Population: {data["population"]}<br>Family Size: {data["family_size"]}'
-        chart_data = [spending_data[state]['education'], spending_data[state]['healthcare'], spending_data[state]['transportation']]
-        chart_labels = ['Education', 'Healthcare', 'Transportation']
-        colors = ['blue', 'red', 'green']
-        chart = create_pie_chart(chart_data, chart_labels, colors)
-        folium.Marker(location=[lat, lon], popup=popup_text, icon=folium.Icon(color='gray', icon='info-sign')).add_to(m)
-        folium.Marker(location=[lat, lon], icon=folium.Icon(color='cadetblue', icon='chart-pie'), tooltip='Click for chart').add_to(m).add_child(chart)
-
-    st.markdown(folium_map_to_html(m), unsafe_allow_html=True)
-
-def get_coordinates(state):
-    # Replace this with your own function to get the coordinates for each state
-    # This is just a placeholder
-    return np.random.randint(-120, -70), np.random.randint(25, 50)
-
-def create_pie_chart(data, labels, colors):
-    chart = folium.plugins.PieChart(data, labels=labels, colors=colors, radius=30, weight=0.5)
-    return chart
-
-def folium_map_to_html(m):
-    return m.get_root().render()
+        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"]}')