Update app.py

app.py

@@ -1,5 +1,6 @@
 # Import necessary libraries
 import pandas as pd
+import streamlit as st
 import matplotlib.pyplot as plt
 
 # Create a dictionary of states with their region, population, and area
@@ -56,6 +57,7 @@ states = {
     'Wyoming': {'region': 'West', 'population': 578759, 'area': 253335}
 }
 
+
 # Create a function to calculate population density
 def calculate_density(population, area):
     return population / area
@@ -63,7 +65,7 @@ def calculate_density(population, area):
 # Create a function to plot the graph
 def plot_graph(df, region):
     plt.figure(figsize=(10, 5))
-    plt.bar(df.index, df['Population Density'])
+    plt.bar(df['State'], df['Population Density'])
     plt.title(f'Population Density of States in {region} Region')
     plt.xlabel('State')
     plt.ylabel('Population Density')
@@ -83,16 +85,24 @@ for state, data in states.items():
         regions[region]['area'] += area
     regions[region]['states'].append(state)
 
-# Calculate the population density for each state in each region
+# Calculate the population density for each state in each region and create dataframes
+dataframes = []
 for region, data in regions.items():
     population = data['population']
     area = data['area']
     states = data['states']
-    density = []
+    densities = []
     for state in states:
-        state_population = states[state]['population']
-        state_area = states[state]['area']
+        state_data = states_dict[state]
+        state_population = state_data['population']
+        state_area = state_data['area']
         state_density = calculate_density(state_population, state_area)
-        density.append(state_density)
-    df = pd.DataFrame({'State': states, 'Population Density': density})
-    plot_graph(df, region)
+        densities.append(state_density)
+    df = pd.DataFrame({'State': states, 'Population Density': densities})
+    dataframes.append(df)
+
+    plot_graph(df, region)
+
+# Use Streamlit to display dataframes
+for df in dataframes:
+    st.write(df)