Update app.py

app.py

@@ -8,6 +8,8 @@ from plotly.subplots import make_subplots
 import plotly.express as px
 import datetime
 import numpy as np
+import matplotlib.pyplot as plt
+import seaborn as sns
 
 # Function to fetch renewable energy data
 def get_renewable_energy_data(city_code):
@@ -244,6 +246,13 @@ if st.button('Calculate Optimal Energy Mix'):
         hourly_data['Hour'] = hourly_data['Time'].dt.hour
         
         for column in hourly_data.columns[2:]:
+            # Create a pivot table to reshape data into hours of the day vs days of the year
             pivot_df = hourly_data.pivot_table(index='Hour', columns='Date', values=column, aggfunc=np.mean)
-            fig_heatmap = px.imshow(pivot_df, title=f"Heatmap of {column}", labels={'color': column}, template='plotly_white', aspect='auto')
-            st.plotly_chart(fig_heatmap, use_container_width=True, height=800)
+            # Plot using seaborn to create a heatmap similar to the provided image
+            plt.figure(figsize=(12, 6))
+            sns.heatmap(pivot_df, cmap='viridis', cbar_kws={'label': f'{column} (kW)'})
+            plt.title(f'Heatmap of {column}')
+            plt.xlabel('Days of the Year')
+            plt.ylabel('Hours of the Day')
+            st.pyplot(plt.gcf())
+            plt.close()