Update app.py

app.py

@@ -33,11 +33,11 @@ st.write(filtered_df)
 avg_prices = filtered_df.groupby(['neighbourhood', 'room_type'])['price'].mean().reset_index()
 
 # Pivot the DataFrame to have neighbourhoods as index and room types as columns
-avg_prices_pivot = avg_prices.pivot(index='neighbourhood', values='price')
+avg_prices_pivot = avg_prices.pivot(index='neighbourhood', columns='room_type', values='price')
 
-# Plot a histogram for each neighbourhood
+# Plot a bar chart for average price of both room types in each neighbourhood
 if not avg_prices_pivot.empty:
-    st.write('Histogram of Average Price for Each Neighbourhood:')
+    st.write('Bar Chart of Average Price for Each Neighbourhood and Room Type:')
     fig, ax = plt.subplots()
     avg_prices_pivot.plot(kind='bar', ax=ax)
     ax.set_xlabel('Neighbourhood')