Update app.py

app.py

@@ -57,7 +57,6 @@ states = {
     'Wyoming': {'region': 'West', 'population': 578759, 'area': 253335}
 }
 
-
 # Create a function to calculate population density
 def calculate_density(population, area):
     return population / area
@@ -90,19 +89,19 @@ dataframes = []
 for region, data in regions.items():
     population = data['population']
     area = data['area']
-    states = data['states']
+    states_in_region = data['states']
     densities = []
-    for state in states:
-        state_data = states_dict[state]
+    for state in states_in_region:
+        state_data = states[state]
         state_population = state_data['population']
         state_area = state_data['area']
         state_density = calculate_density(state_population, state_area)
         densities.append(state_density)
-    df = pd.DataFrame({'State': states, 'Population Density': densities})
+    df = pd.DataFrame({'State': states_in_region, 'Population Density': densities})
     dataframes.append(df)
 
     plot_graph(df, region)
 
 # Use Streamlit to display dataframes
 for df in dataframes:
-    st.write(df)
+    st.write(df)