Update app.py

app.py

@@ -165,13 +165,17 @@ with gr.Blocks(title="Retail Finance: Spend Forecast") as demo:
         "Upload your transactions CSV (columns: `date, amount, category, income`) or use demo data. "
         "The model forecasts **next-month spend** and flags **overspend risk**."
     )
+    with gr.Row():
+    file = gr.File(label="Upload CSV (optional)")
+    budget = gr.Number(value=2500, label="Monthly budget (positive number)")
+    income_override = gr.Number(value=None, label="Expected income next month (optional)")
     with gr.Row():
         file = gr.File(label="Upload CSV (optional)")
         budget = gr.Number(value=2500, label="Monthly budget (positive number)")
     btn = gr.Button("Run Forecast")
     summary = gr.Dataframe(label="Summary")
     monthly_table = gr.Dataframe(label="Monthly aggregates used by the model")
-    btn.click(safe_run, inputs=[file, budget], outputs=[summary, monthly_table])
+   btn.click(safe_run, inputs=[file, budget, income_override], outputs=[summary, monthly_table])
 
 if __name__ == "__main__":
     demo.launch()