Update app.py

app.py

@@ -60,17 +60,39 @@ else:
 
 def create_choropleth_map(metric, start_date, end_date):
     """Creates a choropleth map for a given metric and date range."""
+    print(f"DEBUG: create_choropleth_map called with metric={metric}, start_date={start_date}, end_date={end_date}")
+    
     if panel_df is None or 'DUMMY' in panel_df['GEOID'].tolist():
         fig, ax = plt.subplots()
         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)
+    # Parse dates - handle both string and datetime inputs
+    try:
+        if isinstance(start_date, str):
+            start_date = pd.to_datetime(start_date)
+        if isinstance(end_date, str):
+            end_date = pd.to_datetime(end_date)
+        
+        print(f"DEBUG: Parsed dates - start: {start_date}, end: {end_date}")
+        
+    except Exception as e:
+        print(f"DEBUG: Date parsing error: {e}")
+        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)]
+    print(f"DEBUG: Filtered dataframe length: {len(filtered_df)}")
+    
+    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", ha='center', va='center')
+        ax.set_title('No Data Available', fontsize=15)
+        ax.set_axis_off()
+        return fig
     
     geoid_totals = filtered_df.groupby('GEOID')[metric].sum().reset_index()
+    print(f"DEBUG: GEOID totals shape: {geoid_totals.shape}")
     
     merged_gdf = tracts_gdf.merge(geoid_totals, on='GEOID', how='left').fillna(0)
     
@@ -88,17 +110,38 @@ def create_choropleth_map(metric, start_date, end_date):
 
 def create_time_series_plot(metric, start_date, end_date):
     """Creates a time series plot for a given metric and date range."""
+    print(f"DEBUG: create_time_series_plot called with metric={metric}, start_date={start_date}, end_date={end_date}")
+    
     if panel_df is None or 'DUMMY' in panel_df['GEOID'].tolist():
         fig, ax = plt.subplots()
         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)
+    # Parse dates - handle both string and datetime inputs
+    try:
+        if isinstance(start_date, str):
+            start_date = pd.to_datetime(start_date)
+        if isinstance(end_date, str):
+            end_date = pd.to_datetime(end_date)
+            
+        print(f"DEBUG: Parsed dates - start: {start_date}, end: {end_date}")
+        
+    except Exception as e:
+        print(f"DEBUG: Date parsing error: {e}")
+        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)]
+    print(f"DEBUG: Filtered dataframe length: {len(filtered_df)}")
+    
+    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", ha='center', va='center')
+        ax.set_title('No Data Available', fontsize=15)
+        return fig
     
     monthly_totals = filtered_df.groupby('month')[metric].sum()
+    print(f"DEBUG: Monthly totals shape: {monthly_totals.shape}")
     
     fig, ax = plt.subplots(figsize=(12, 6))
     monthly_totals.plot(ax=ax)
@@ -192,49 +235,58 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     choices=['crime_total', 'sr311_total', 'dob_permits_total'],
                     value='crime_total'
                 )
-                min_date = panel_df['month'].min().to_pydatetime()
-                max_date = panel_df['month'].max().to_pydatetime()
+                
+                # Get date range from data
+                min_date = panel_df['month'].min().strftime('%Y-%m-%d')
+                max_date = panel_df['month'].max().strftime('%Y-%m-%d')
 
-                start_date_picker = gr.DateTime(label="Start Date", value=min_date, include_time=False)
-                end_date_picker = gr.DateTime(label="End Date", value=max_date, include_time=False)
+                start_date_picker = gr.Textbox(
+                    label="Start Date (YYYY-MM-DD)", 
+                    value=min_date,
+                    placeholder="2023-01-01"
+                )
+                end_date_picker = gr.Textbox(
+                    label="End Date (YYYY-MM-DD)", 
+                    value=max_date,
+                    placeholder="2023-12-31"
+                )
                 
                 update_button = gr.Button("Update Dashboard")
 
             with gr.Column(scale=3):
                 gr.Markdown("### Visualizations")
-                # Initialize plots with default values
-                initial_map = create_choropleth_map('crime_total', min_date, max_date)
-                initial_ts = create_time_series_plot('crime_total', min_date, max_date)
-                
-                map_plot = gr.Plot(value=initial_map)
-                ts_plot = gr.Plot(value=initial_ts)
+                map_plot = gr.Plot()
+                ts_plot = gr.Plot()
         
-        # Function to update both plots
-        def update_plots(metric, start_date, end_date):
+        # Function to update both plots at once
+        def update_dashboard(metric, start_date, end_date):
+            print(f"DEBUG: update_dashboard called with {metric}, {start_date}, {end_date}")
             map_fig = create_choropleth_map(metric, start_date, end_date)
             ts_fig = create_time_series_plot(metric, start_date, end_date)
             return map_fig, ts_fig
         
-        # Update plots when button is clicked
+        # Update on button click
         update_button.click(
-            fn=update_plots,
+            fn=update_dashboard,
             inputs=[metric_selector, start_date_picker, end_date_picker],
             outputs=[map_plot, ts_plot]
         )
         
-        # Also update plots when any input changes
+        # Also trigger updates when inputs change
         metric_selector.change(
-            fn=update_plots,
+            fn=update_dashboard,
             inputs=[metric_selector, start_date_picker, end_date_picker],
             outputs=[map_plot, ts_plot]
         )
+        
         start_date_picker.change(
-            fn=update_plots,
+            fn=update_dashboard,
             inputs=[metric_selector, start_date_picker, end_date_picker],
             outputs=[map_plot, ts_plot]
         )
+        
         end_date_picker.change(
-            fn=update_plots,
+            fn=update_dashboard,
             inputs=[metric_selector, start_date_picker, end_date_picker],
             outputs=[map_plot, ts_plot]
         )