Add editable planting schedule to Declining Increment tab with live dashboard updates

dashboard/app.py

@@ -275,16 +275,57 @@ with gr.Blocks() as demo:
     gr.Markdown("# Mangrove ER Model Dashboard\nDeclining Increment Model Only")
     with gr.Tabs():
         with gr.Tab("Declining Increment"):
-            gr.Markdown("## Declining Increment Model Results (base config)")
-            carbon_plot, annual_plot, biomass_plot, summary, results, species, survival = run_model(MODEL_CONFIGS["Declining Increment"])
-            growth_fig = create_growth_increment_plots(yaml.safe_load(open(MODEL_CONFIGS["Declining Increment"])), model_type="declining_increment")
+            gr.Markdown("## Planting Schedule (ha per year)")
+            year_1 = gr.Number(value=2500, label="Year 1 Area (ha)")
+            year_2 = gr.Number(value=2500, label="Year 2 Area (ha)")
+            year_3 = gr.Number(value=0, label="Year 3 Area (ha)")
+            year_4 = gr.Number(value=0, label="Year 4 Area (ha)")
+            year_5 = gr.Number(value=0, label="Year 5 Area (ha)")
+            update_btn = gr.Button("Update Results", variant="primary")
             with gr.Row():
-                carbon_plot_comp = gr.Plot(value=carbon_plot)
-                biomass_plot_comp = gr.Plot(value=biomass_plot)
-                annual_plot_comp = gr.Plot(value=annual_plot)
-            gr.Textbox(value=summary, label="Summary", lines=12)
-            gr.Dataframe(value=results, label="Project Results (Annual)")
-            gr.Dataframe(value=species, label="Species Results (Annual)")
-            gr.Dataframe(value=survival, label="Surviving Trees Table")
-            # gr.Plot(value=growth_fig, label="Growth & Increment Curves")
+                carbon_plot = gr.Plot()
+                biomass_plot = gr.Plot()
+                annual_plot = gr.Plot()
+            summary_box = gr.Textbox(label="Summary", lines=12)
+            results_box = gr.Dataframe(label="Project Results (Annual)")
+            species_box = gr.Dataframe(label="Species Results (Annual)")
+            survival_box = gr.Dataframe(label="Surviving Trees Table")
+
+            def update_declining_increment(y1, y2, y3, y4, y5):
+                config = update_planting_schedule(MODEL_CONFIGS["Declining Increment"], [y1, y2, y3, y4, y5])
+                import tempfile
+                import yaml
+                with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as tmp:
+                    yaml.dump(config, tmp)
+                    tmp_path = tmp.name
+                model = ERModel(Path(tmp_path))
+                results, species_results = model.run()
+                plots = create_all_plots(results, species_results, config)
+                summary = create_summary(results, species_results, config, model)
+                survival_table = create_survival_table(model)
+                results_fmt = results.copy()
+                species_results_fmt = species_results.copy()
+                for col in results_fmt.columns:
+                    if results_fmt[col].dtype in [float, int]:
+                        results_fmt[col] = results_fmt[col].apply(lambda x: f"{x:,.2f}" if isinstance(x, float) else f"{x:,}")
+                for col in species_results_fmt.columns:
+                    if species_results_fmt[col].dtype in [float, int]:
+                        species_results_fmt[col] = species_results_fmt[col].apply(lambda x: f"{x:,.2f}" if isinstance(x, float) else f"{x:,}")
+                return plots[0], plots[2], plots[1], summary, results_fmt.head(30), species_results_fmt.head(30), survival_table.head(30)
+
+            update_btn.click(
+                update_declining_increment,
+                inputs=[year_1, year_2, year_3, year_4, year_5],
+                outputs=[carbon_plot, biomass_plot, annual_plot, summary_box, results_box, species_box, survival_box]
+            )
+
+            # Show initial results
+            c, b, a, summary, r, s, surv = update_declining_increment(2500, 2500, 0, 0, 0)
+            carbon_plot.value = c
+            biomass_plot.value = b
+            annual_plot.value = a
+            summary_box.value = summary
+            results_box.value = r
+            species_box.value = s
+            survival_box.value = surv
     demo.launch(share=True)