Update app.py

app.py

@@ -65,41 +65,13 @@ def create_choropleth_map(metric, start_date, end_date):
         ax.text(0.5, 0.5, "Data not loaded", ha='center', va='center')
         return fig
 
-    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()
+    start_date = pd.to_datetime(start_date)
+    end_date = pd.to_datetime(end_date)
     
     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))
@@ -121,20 +93,8 @@ 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
 
-    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()
+    start_date = pd.to_datetime(start_date)
+    end_date = pd.to_datetime(end_date)
 
     filtered_df = panel_df[(panel_df['month'] >= start_date) & (panel_df['month'] <= end_date)]