Update app.py

app.py

@@ -201,7 +201,7 @@ def irr_sql(cols: List[str]) -> str:
 # =========================
 # Dashboard callback
 # =========================
-def run_dashboard(scenario: str) -> Tuple[str, str, str, str, str, Any, pd.DataFrame, pd.DataFrame, str, pd.DataFrame]:
+def run_dashboard(scenario: str, runoff_pct: float, rate_shock_bps_input: float) -> Tuple[str, str, str, str, str, Any, pd.DataFrame, pd.DataFrame, str, pd.DataFrame]:
     """
     Returns:
       status, as_of, a1_text, a2_text, a3_text, figure, ladder_df, irr_df,
@@ -213,14 +213,14 @@ def run_dashboard(scenario: str) -> Tuple[str, str, str, str, str, Any, pd.DataF
         # --- Scenario Application ---
         # Create a temporary view with scenario adjustments.
         # Subsequent queries will use this stressed view.
-        stressed_view_fqn = f"{VIEW_FQN}_stressed"
+        stressed_view_fqn = "positions_v_stressed"
         runoff_factor = 1.0
         rate_shock_bps = 0.0
 
-        if scenario == "Liquidity Stress: High Deposit Runoff":
-            runoff_factor = 0.8 # 20% runoff
-        elif scenario == "IRR Stress: Rate Shock (+200bps)":
-            rate_shock_bps = 200.0
+        if scenario == "Liquidity Stress: High Deposit Runoff" and runoff_pct > 0:
+            runoff_factor = (100.0 - runoff_pct) / 100.0
+        elif scenario == "IRR Stress: Rate Shock" and rate_shock_bps_input != 0:
+            rate_shock_bps = rate_shock_bps_input
 
         scenario_sql = f"""
         CREATE OR REPLACE TEMP VIEW {stressed_view_fqn} AS
@@ -311,7 +311,7 @@ def run_dashboard(scenario: str) -> Tuple[str, str, str, str, str, Any, pd.DataF
         explain_text += "*   **Seasonal Pattern:** Analysis not possible without relevant time-series features in the source data."
 
         # Add scenario explanation for IRR stress
-        if scenario == "IRR Stress: Rate Shock (+200bps)" and not irr.empty:
+        if scenario == "IRR Stress: Rate Shock" and rate_shock_bps != 0 and not irr.empty:
             net_bpv = irr["BPV (DV01)"].sum()
             eve_impact = net_bpv * rate_shock_bps
             eve_impact_mn = eve_impact / 1_000_000
@@ -368,10 +368,22 @@ with gr.Blocks(title=APP_TITLE) as demo:
 
     scenario_dd = gr.Dropdown(
         label="Select Stress Scenario",
-        choices=["Baseline", "Liquidity Stress: High Deposit Runoff", "IRR Stress: Rate Shock (+200bps)"],
+        choices=["Baseline", "Liquidity Stress: High Deposit Runoff", "IRR Stress: Rate Shock"],
         value="Baseline"
     )
 
+    with gr.Accordion("Stress Scenario Parameters", open=False):
+        runoff_slider = gr.Slider(
+            label="Deposit Runoff (%)",
+            minimum=0, maximum=100, step=1, value=20,
+            info="For Liquidity Stress: Percentage of key deposits that run off."
+        )
+        shock_slider = gr.Slider(
+            label="Rate Shock (bps)",
+            minimum=-500, maximum=500, step=25, value=200,
+            info="For IRR Stress: Parallel shift in the yield curve in basis points."
+        )
+
     with gr.Row():
         as_of = gr.Textbox(label="As of date", interactive=False)
 
@@ -396,7 +408,7 @@ with gr.Blocks(title=APP_TITLE) as demo:
 
     refresh_btn.click(
         fn=run_dashboard,
-        inputs=[scenario_dd],
+        inputs=[scenario_dd, runoff_slider, shock_slider],
         outputs=[status, as_of, a1, a2, a3, chart, ladder_df, irr_df, explain_text, drivers_df],
     )