Update app.py

app.py

@@ -10,7 +10,7 @@ import sys
 import matplotlib.pyplot as plt
 from matplotlib.gridspec import GridSpec
 import matplotlib.dates as mdates
-from windrose import WindRose, WindroseAxes
+from windrose import WindroseAxes
 from datetime import datetime
 
 # Install Playwright browsers on startup
@@ -130,20 +130,22 @@ def parse_weather_data(data):
     
     return df
 
-def create_wind_rose(df, ax):
+def create_wind_rose(df, ax=None):
     """Create a wind rose plot"""
+    if ax is None:
+        fig = plt.figure(figsize=(8, 8))
+        ax = WindroseAxes.from_ax(fig=fig)
+    
     ws = df['wind_speed'].dropna().values
     wd = df['wind_dir_deg'].dropna().values
     
-    # Define the wind speed bins
-    bins = np.array([0, 5, 10, 15, 20, 25, 30, 35, 40])
-    
-    # Create the wind rose
-    ax.bar(wd, ws, nsector=36, normed=True, opening=0.8, 
-           bins=bins, colors=plt.cm.viridis(np.linspace(0, 1, len(bins)-1)))
-    
-    # Set legend
-    ax.set_legend(title='Wind Speed (mph)', bbox_to_anchor=(1.2, 0.5))
+    if len(ws) > 0 and len(wd) > 0:
+        # 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)))
+        ax.set_legend(title='Wind Speed (mph)', bbox_to_anchor=(1.2, 0.5))
+    return ax
 
 def create_daily_wind_roses(df):
     """Create wind roses for each day"""
@@ -154,16 +156,11 @@ def create_daily_wind_roses(df):
         if len(group) > 0:  # Only create rose if we have data
             fig = plt.figure(figsize=(8, 8))
             ax = WindroseAxes.from_ax(fig=fig)
-            ws = group['wind_speed'].dropna().values
-            wd = group['wind_dir_deg'].dropna().values
-            if len(ws) > 0 and len(wd) > 0:
-                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)))
-                ax.set_legend(title='Wind Speed (mph)', bbox_to_anchor=(1.2, 0.5))
+            if create_wind_rose(group, ax):
                 ax.set_title(f'Wind Rose - {date.strftime("%Y-%m-%d")}')
                 plt.tight_layout()
                 roses.append(fig)
+            plt.close(fig)
     
     return roses
 
@@ -231,14 +228,16 @@ def create_plots(df):
     ax4.grid(True, axis='y')
     
     # Add overall wind rose
-    ax5 = fig.add_subplot(gs[4, 0], projection='windrose')
+    fig_wind = plt.figure(figsize=(8, 8))
+    ax5 = WindroseAxes.from_ax(fig=fig_wind)
     create_wind_rose(df, ax5)
     ax5.set_title('Overall Wind Rose')
     
     # Add latest day's wind rose
     latest_date = df['date'].iloc[0]
     latest_data = df[df['date'] == latest_date]
-    ax6 = fig.add_subplot(gs[4, 1], projection='windrose')
+    fig_latest = plt.figure(figsize=(8, 8))
+    ax6 = WindroseAxes.from_ax(fig=fig_latest)
     create_wind_rose(latest_data, ax6)
     ax6.set_title(f'Latest Day Wind Rose\n{latest_date}')
     
@@ -313,6 +312,7 @@ with gr.Blocks(title="Weather Station Data Analyzer") as demo:
             minimum=1,
             maximum=1440
         )
+    
     analyze_btn = gr.Button("Fetch and Analyze Weather Data")
     
     with gr.Row():