Update app.py

app.py

@@ -81,7 +81,7 @@ def create_choropleth_map(metric, start_date, end_date):
                     cmap='viridis',
                     legend_kwds={'label': f"Total {metric.replace('_', ' ').title()}",
                                  'orientation': "horizontal"})
-    ax.set_title(f'Spatial Distribution of {metric.replace("_", " ").title()}', fontsize=15)
+    ax.set_title(f'Spatial Distribution of {metric.replace("_', ' ').title()}', fontsize=15)
     ax.set_axis_off()
     plt.tight_layout()
     return fig
@@ -195,48 +195,25 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 min_date = panel_df['month'].min().to_pydatetime()
                 max_date = panel_df['month'].max().to_pydatetime()
 
-                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.DatePicker(label="Start Date", value=min_date)
+                end_date_picker = gr.DatePicker(label="End Date", value=max_date)
                 
                 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)
-        
-        # Function to update both plots
-        def update_plots(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
+                map_plot = gr.Plot()
+                ts_plot = gr.Plot()
         
-        # Update plots when button is clicked
         update_button.click(
-            fn=update_plots,
-            inputs=[metric_selector, start_date_picker, end_date_picker],
-            outputs=[map_plot, ts_plot]
-        )
-        
-        # Also update plots when any input changes
-        metric_selector.change(
-            fn=update_plots,
-            inputs=[metric_selector, start_date_picker, end_date_picker],
-            outputs=[map_plot, ts_plot]
-        )
-        start_date_picker.change(
-            fn=update_plots,
+            fn=create_choropleth_map,
             inputs=[metric_selector, start_date_picker, end_date_picker],
-            outputs=[map_plot, ts_plot]
+            outputs=map_plot
         )
-        end_date_picker.change(
-            fn=update_plots,
+        update_button.click(
+            fn=create_time_series_plot,
             inputs=[metric_selector, start_date_picker, end_date_picker],
-            outputs=[map_plot, ts_plot]
+            outputs=ts_plot
         )
 
     with gr.Tab("Predictive Analytics"):
@@ -281,4 +258,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()