Update app.py

app.py

@@ -1,25 +1,61 @@
 import streamlit as st
-import geopandas as gpd
-import matplotlib.pyplot as plt
-
-# Function to plot the map with labels
-def plot_map():
-    us_map = gpd.read_file('path_to_shapefile.shp')
-    corporations_data = {
-        'Minnesota': [('CorpA', 100), ('CorpB', 50)],
-        'California': [('TechCorp', 300), ('HealthCorp', 200)],
-        # Add more states and corporations as needed
-    }
-    us_map.plot()
-    for state, data in corporations_data.items():
-        geometry = us_map[us_map['STATE_NAME'] == state].geometry.iloc[0]
-        centroid = geometry.centroid
-        for i, (corp, revenue) in enumerate(data):
-            plt.text(centroid.x, centroid.y - i * 0.1, f'{corp}: ${revenue}B', fontsize=10)
-    plt.axis('off')
-    plt.show()
+import pandas as pd
+import plotly.express as px
+import csv
+import base64
+
+# Default data for top two corporations per state
+corporations_data = {
+    'Minnesota': [('CorpA', 100), ('CorpB', 50)],
+    'California': [('TechCorp', 300), ('HealthCorp', 200)],
+    # Add more states and corporations as needed
+}
+
+# Function to create a CSV file from the corporations_data dictionary
+def create_csv():
+    csv_file = 'corporations_data.csv'
+    with open(csv_file, 'w', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerow(['State', 'Corporation', 'Revenue'])
+        for state, data in corporations_data.items():
+            for corp, revenue in data:
+                writer.writerow([state, corp, revenue])
+    return csv_file
+
+# Function to download CSV
+def download_csv(file_name):
+    with open(file_name, 'r') as file:
+        csv_data = file.read()
+    b64 = base64.b64encode(csv_data.encode()).decode()
+    href = f'<a href="data:file/csv;base64,{b64}" download="{file_name}">Download CSV File 📥</a>'
+    return href
+
+# Function to plot the map
+def plot_map(data):
+    fig = px.choropleth(locations=data['State'], locationmode="USA-states", color=data['Revenue'],
+                        scope="usa", title="Top Corporations by State in the United States",
+                        hover_name=data['Corporation'], hover_data=['Revenue'])
+    return fig
+
+# Create CSV from default data
+csv_file = create_csv()
 
 # Streamlit app
 st.title('Top Corporations by State in the United States')
-st.write('This map shows the top 2 corporations in each state with their revenue in billions of dollars for 2021.')
-plot_map()
+
+# Upload CSV
+uploaded_files = st.file_uploader("Choose a CSV file", accept_multiple_files=True, type="csv")
+
+# If files are uploaded, use them to generate maps
+if uploaded_files:
+    for uploaded_file in uploaded_files:
+        data = pd.read_csv(uploaded_file)
+        st.write(f"Map for {uploaded_file.name}")
+        st.plotly_chart(plot_map(data))
+else:
+    # Use default data
+    data = pd.read_csv(csv_file)
+    st.plotly_chart(plot_map(data))
+
+# Download CSV
+st.markdown(download_csv(csv_file), unsafe_allow_html=True)