Update app.py

app.py

@@ -6,23 +6,22 @@ import plotly.graph_objects as go
 # Function to plot the map
 def plot_map(data):
     fig = px.choropleth(locations=data['State'], locationmode="USA-states", scope="usa")
-    for index, row in data.iterrows():
+    grouped_data = data.groupby('State')
+    for state, group in grouped_data:
+        top_corp = group.nlargest(1, 'Revenue')
+        text_label = f"{top_corp['Corporation'].iloc[0]} - ${top_corp['Revenue'].iloc[0]}B"
+        lon = group['Longitude'].mean()
+        lat = group['Latitude'].mean()
         fig.add_trace(go.Scattergeo(
-            lon=[row['Longitude']],
-            lat=[row['Latitude']],
-            text=f"{row['Corporation']} - ${row['Revenue']}B",
+            lon=[lon],
+            lat=[lat],
+            text=text_label,
             mode='text',
         ))
-    fig.update_layout(title="Top Corporations by State in the United States")
+    fig.update_layout(title="Top Corporation by State in the United States")
     return fig
 
-# 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')
+st.title('Top Corporation by State in the United States 🏢')
 
 # Upload CSV
 uploaded_files = st.file_uploader("Choose a CSV file", accept_multiple_files=True, type="csv")
@@ -30,23 +29,11 @@ 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 🗺️')
 
-# Download CSV button
-download_csv_button = st.button('Download Top Two Corporations CSV 📥')
-
-if display_map_button or download_csv_button:
+if display_map_button:
     if uploaded_files:
         for uploaded_file in uploaded_files:
             data = pd.read_csv(uploaded_file)
-            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"
-                )
+            st.write(f"Map for {uploaded_file.name} 🌐")
+            st.plotly_chart(plot_map(data))
     else:
-        st.write("Please upload a CSV file to proceed.")
+        st.write("Please upload a CSV file to proceed. 📑")