Update app.py

app.py

@@ -158,66 +158,71 @@ def create_wind_rose(df, ax):
 
 def create_plots(df):
     """Create all weather plots including SWE estimates"""
-    # Create figure with adjusted GridSpec for better plot alignment
-    fig = plt.figure(figsize=(20, 20))  # Increased height to accommodate separated plots
-    gs = GridSpec(5, 1, figure=fig)  # Changed to 5 rows, 1 column
+    # Create figure with adjusted height and spacing
+    fig = plt.figure(figsize=(20, 24))
+    
+    # Calculate height ratios for different plots
+    height_ratios = [1, 1, 1, 1, 1]  # Equal height for all plots
+    gs = GridSpec(5, 1, figure=fig, height_ratios=height_ratios)
+    gs.update(hspace=0.4)  # Increase vertical spacing between plots
     
     # Temperature plot
     ax1 = fig.add_subplot(gs[0])
     ax1.plot(df['datetime'], df['temp'], label='Temperature', color='red')
     ax1.plot(df['datetime'], df['wind_chill'], label='Wind Chill', color='blue')
-    ax1.set_title('Temperature and Wind Chill Over Time')
+    ax1.set_title('Temperature and Wind Chill Over Time', pad=20)
     ax1.set_xlabel('Date')
     ax1.set_ylabel('Temperature (°F)')
     ax1.legend()
     ax1.grid(True)
-    plt.setp(ax1.xaxis.get_majorticklabels(), rotation=45)
+    ax1.tick_params(axis='x', rotation=45)
     
     # Wind speed plot
     ax2 = fig.add_subplot(gs[1])
     ax2.plot(df['datetime'], df['wind_speed'], label='Wind Speed', color='blue')
     ax2.plot(df['datetime'], df['wind_gust'], label='Wind Gust', color='orange')
-    ax2.set_title('Wind Speed and Gusts Over Time')
+    ax2.set_title('Wind Speed and Gusts Over Time', pad=20)
     ax2.set_xlabel('Date')
     ax2.set_ylabel('Wind Speed (mph)')
     ax2.legend()
     ax2.grid(True)
-    plt.setp(ax2.xaxis.get_majorticklabels(), rotation=45)
+    ax2.tick_params(axis='x', rotation=45)
     
-    # Snow depth plot - Now on its own row
+    # Snow depth plot
     ax3 = fig.add_subplot(gs[2])
     ax3.plot(df['datetime'], df['snow_depth'], color='blue', label='Snow Depth')
-    ax3.set_title('Snow Depth Over Time')
+    ax3.set_title('Snow Depth Over Time', pad=20)
     ax3.set_xlabel('Date')
     ax3.set_ylabel('Snow Depth (inches)')
     ax3.grid(True)
-    plt.setp(ax3.xaxis.get_majorticklabels(), rotation=45)
+    ax3.tick_params(axis='x', rotation=45)
     
-    # Daily new snow bar plot - Now on its own row
+    # Daily new snow bar plot
     ax4 = fig.add_subplot(gs[3])
     daily_snow = df.groupby('date')['snowfall_3hr'].sum()
     ax4.bar(daily_snow.index, daily_snow.values, color='blue')
-    ax4.set_title('Daily New Snow')
+    ax4.set_title('Daily New Snow', pad=20)
     ax4.set_xlabel('Date')
     ax4.set_ylabel('New Snow (inches)')
-    plt.setp(ax4.xaxis.get_majorticklabels(), rotation=45)
+    ax4.tick_params(axis='x', rotation=45)
     
     # SWE bar plot
     ax5 = fig.add_subplot(gs[4])
     daily_swe = df.groupby('date')['swe'].mean()
     ax5.bar(daily_swe.index, daily_swe.values, color='lightblue')
-    ax5.set_title('Snow/Water Equivalent')
+    ax5.set_title('Snow/Water Equivalent', pad=20)
     ax5.set_xlabel('Date')
     ax5.set_ylabel('SWE (inches)')
-    plt.setp(ax5.xaxis.get_majorticklabels(), rotation=45)
+    ax5.tick_params(axis='x', rotation=45)
     
-    plt.tight_layout()
+    # Adjust layout
+    plt.subplots_adjust(top=0.95, bottom=0.05, left=0.1, right=0.95)
     
     # Create separate wind rose figure
     fig_rose = plt.figure(figsize=(10, 10))
     ax_rose = WindroseAxes.from_ax(fig=fig_rose)
     create_wind_rose(df, ax_rose)
-    plt.tight_layout()
+    fig_rose.subplots_adjust(top=0.95, bottom=0.05, left=0.1, right=0.95)
     
     return fig, fig_rose