Update app.py

app.py

@@ -140,12 +140,17 @@ def create_wind_rose(df, ax=None):
     wd = df['wind_dir_deg'].dropna().values
     
     if len(ws) > 0 and len(wd) > 0:
+        # Define wind speed bins
+        bins = np.array([0, 5, 10, 15, 20, 25, 30, 35])
+        # Create color map
+        colors = plt.cm.viridis(np.linspace(0, 1, len(bins)-1))
+        
         # Create the wind rose
         ax.bar(wd, ws, nsector=36, normed=True, opening=0.8,
-               bins=np.array([0, 5, 10, 15, 20, 25, 30, 35, 40]),
-               colors=plt.cm.viridis(np.linspace(0, 1, 8)))
+              bins=bins, colors=colors)
         ax.set_legend(title='Wind Speed (mph)', bbox_to_anchor=(1.2, 0.5))
-    return ax
+        return True
+    return False
 
 def create_daily_wind_roses(df):
     """Create wind roses for each day"""
@@ -168,7 +173,7 @@ def create_plots(df):
     """Create all weather plots"""
     # Create main figure with subplots
     fig = plt.figure(figsize=(20, 25))
-    gs = GridSpec(5, 2, figure=fig, height_ratios=[1, 1, 1, 1, 1])
+    gs = GridSpec(4, 2, figure=fig, height_ratios=[1, 1, 1, 1])
     
     # Temperature plot
     ax1 = fig.add_subplot(gs[0, :])
@@ -195,7 +200,7 @@ def create_plots(df):
     plt.setp(ax2.xaxis.get_majorticklabels(), rotation=45, ha='right')
     
     # Snow depth plot
-    ax3 = fig.add_subplot(gs[2, :])
+    ax3 = fig.add_subplot(gs[2, 0])
     ax3.plot(df['datetime'], df['snow_depth'], color='blue', label='Snow Depth')
     ax3.set_title('Snow Depth Over Time', pad=20)
     ax3.set_xlabel('Date')
@@ -205,7 +210,7 @@ def create_plots(df):
     plt.setp(ax3.xaxis.get_majorticklabels(), rotation=45, ha='right')
     
     # Daily new snow bar plot
-    ax4 = fig.add_subplot(gs[3, :])
+    ax4 = fig.add_subplot(gs[2, 1])
     daily_snow = df.groupby('date')['snowfall_24hr'].max()
     bars = ax4.bar(range(len(daily_snow)), daily_snow.values, color='blue', width=0.8)
     
@@ -227,17 +232,15 @@ def create_plots(df):
     ax4.set_ylabel('New Snow (inches)')
     ax4.grid(True, axis='y')
     
-    # Add overall wind rose
-    fig_wind = plt.figure(figsize=(8, 8))
-    ax5 = WindroseAxes.from_ax(fig=fig_wind)
+    # Wind roses
+    ax5 = fig.add_subplot(gs[3, 0], projection='windrose')
     create_wind_rose(df, ax5)
     ax5.set_title('Overall Wind Rose')
     
-    # Add latest day's wind rose
+    # Latest day's wind rose
     latest_date = df['date'].iloc[0]
     latest_data = df[df['date'] == latest_date]
-    fig_latest = plt.figure(figsize=(8, 8))
-    ax6 = WindroseAxes.from_ax(fig=fig_latest)
+    ax6 = fig.add_subplot(gs[3, 1], projection='windrose')
     create_wind_rose(latest_data, ax6)
     ax6.set_title(f'Latest Day Wind Rose\n{latest_date}')
     
@@ -321,8 +324,7 @@ with gr.Blocks(title="Weather Station Data Analyzer") as demo:
     with gr.Row():
         weather_plots = gr.Plot(label="Weather Plots")
     
-    with gr.Row():
-        gallery = gr.Gallery(label="Daily Wind Roses", columns=3, rows=2, height=400)
+    gallery = gr.Gallery(label="Daily Wind Roses", columns=3, rows=2, height=400)
     
     analyze_btn.click(
         fn=analyze_weather_data,