Update app.py

app.py

@@ -65,13 +65,41 @@ def create_choropleth_map(metric, start_date, end_date):
         ax.text(0.5, 0.5, "Data not loaded", ha='center', va='center')
         return fig
 
-    start_date = pd.to_datetime(start_date)
-    end_date = pd.to_datetime(end_date)
+    try:
+        # Handle both string and datetime inputs
+        if start_date is None or end_date is None:
+            # Use full date range if dates are None
+            start_date = panel_df['month'].min()
+            end_date = panel_df['month'].max()
+        else:
+            start_date = pd.to_datetime(start_date)
+            end_date = pd.to_datetime(end_date)
+    except Exception as e:
+        print(f"Date parsing error: {e}")
+        # Fallback to full range
+        start_date = panel_df['month'].min()
+        end_date = panel_df['month'].max()
     
     filtered_df = panel_df[(panel_df['month'] >= start_date) & (panel_df['month'] <= end_date)]
     
+    if len(filtered_df) == 0:
+        fig, ax = plt.subplots(1, 1, figsize=(10, 10))
+        ax.text(0.5, 0.5, f"No data found for date range\n{start_date.date()} to {end_date.date()}", 
+                ha='center', va='center', fontsize=12)
+        ax.set_title('No Data Available', fontsize=15)
+        ax.set_axis_off()
+        return fig
+    
     geoid_totals = filtered_df.groupby('GEOID')[metric].sum().reset_index()
     
+    if len(geoid_totals) == 0 or geoid_totals[metric].sum() == 0:
+        fig, ax = plt.subplots(1, 1, figsize=(10, 10))
+        ax.text(0.5, 0.5, f"No {metric.replace('_', ' ').title()} data for selected period", 
+                ha='center', va='center', fontsize=12)
+        ax.set_title('No Data Available', fontsize=15)
+        ax.set_axis_off()
+        return fig
+    
     merged_gdf = tracts_gdf.merge(geoid_totals, on='GEOID', how='left').fillna(0)
     
     fig, ax = plt.subplots(1, 1, figsize=(10, 10))
@@ -93,13 +121,39 @@ def create_time_series_plot(metric, start_date, end_date):
         ax.text(0.5, 0.5, "Data not loaded", ha='center', va='center')
         return fig
 
-    start_date = pd.to_datetime(start_date)
-    end_date = pd.to_datetime(end_date)
+    try:
+        # Handle both string and datetime inputs
+        if start_date is None or end_date is None:
+            # Use full date range if dates are None
+            start_date = panel_df['month'].min()
+            end_date = panel_df['month'].max()
+        else:
+            start_date = pd.to_datetime(start_date)
+            end_date = pd.to_datetime(end_date)
+    except Exception as e:
+        print(f"Date parsing error: {e}")
+        # Fallback to full range
+        start_date = panel_df['month'].min()
+        end_date = panel_df['month'].max()
 
     filtered_df = panel_df[(panel_df['month'] >= start_date) & (panel_df['month'] <= end_date)]
     
+    if len(filtered_df) == 0:
+        fig, ax = plt.subplots(figsize=(12, 6))
+        ax.text(0.5, 0.5, f"No data found for date range\n{start_date.date()} to {end_date.date()}", 
+                ha='center', va='center', fontsize=12)
+        ax.set_title('No Data Available', fontsize=15)
+        return fig
+    
     monthly_totals = filtered_df.groupby('month')[metric].sum()
     
+    if len(monthly_totals) == 0 or monthly_totals.sum() == 0:
+        fig, ax = plt.subplots(figsize=(12, 6))
+        ax.text(0.5, 0.5, f"No {metric.replace('_', ' ').title()} data for selected period", 
+                ha='center', va='center', fontsize=12)
+        ax.set_title('No Data Available', fontsize=15)
+        return fig
+    
     fig, ax = plt.subplots(figsize=(12, 6))
     monthly_totals.plot(ax=ax)
     ax.set_title(f'Monthly Total of {metric.replace("_", " ").title()}', fontsize=15)