Update app.py

app.py

@@ -5,10 +5,7 @@ import plotly.graph_objects as go
 
 # Function to plot the map
 def plot_map(data):
-    # Create a choropleth map as a base
     fig = px.choropleth(locations=data['State'], locationmode="USA-states", scope="usa")
-    
-    # Add a scatter plot on top to show the corporation labels
     for index, row in data.iterrows():
         fig.add_trace(go.Scattergeo(
             lon=[row['Longitude']],
@@ -16,13 +13,15 @@ def plot_map(data):
             text=f"{row['Corporation']} - ${row['Revenue']}B",
             mode='text',
         ))
-    
-    # Set the title
     fig.update_layout(title="Top Corporations by State in the United States")
-    
     return fig
 
-# Streamlit app
+# Function to generate the CSV file
+def generate_csv(data):
+    top_corporations = data.nlargest(2, 'Revenue')
+    csv_file = top_corporations.to_csv(index=False)
+    return csv_file
+
 st.title('Top Corporations by State in the United States')
 
 # Upload CSV
@@ -31,14 +30,23 @@ uploaded_files = st.file_uploader("Choose a CSV file", accept_multiple_files=Tru
 # Display map button
 display_map_button = st.button('Display Map of CSV Data 🗺️')
 
-# If button is clicked, use uploaded files to generate maps or use the default CSV file
-if display_map_button:
+# Download CSV button
+download_csv_button = st.button('Download Top Two Corporations CSV 📥')
+
+if display_map_button or download_csv_button:
     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))
+            if display_map_button:
+                st.write(f"Map for {uploaded_file.name}")
+                st.plotly_chart(plot_map(data))
+            if download_csv_button:
+                csv_file = generate_csv(data)
+                st.download_button(
+                    label="Download CSV File",
+                    data=csv_file,
+                    file_name="top_two_corporations.csv",
+                    mime="text/csv"
+                )
     else:
-        # Use the default CSV file if no files are uploaded
-        data = pd.read_csv('corporations_data.csv')
-        st.plotly_chart(plot_map(data))
+        st.write("Please upload a CSV file to proceed.")