Update app.py

app.py

@@ -10,6 +10,7 @@ from tools.report_tool import ReportTool
 from utils.tracing import Tracer
 from utils.config import AppConfig
 from tools.ts_forecast_tool import TimeseriesForecastTool
+from tools.ts_preprocess import build_timeseries
 
 # Optional tiny CPU LLM for planning (can be disabled by not setting ORCHESTRATOR_MODEL)
 llm = None
@@ -78,16 +79,27 @@ def run_agent(message: str, hitl_decision: str = "Approve", reviewer_note: str =
     explain_imgs = {}
     artifacts = {}
 
+    if "sql" in plan["steps"]:
+        sql_df = sql_tool.run(message)
+        # 👉 Build the model-friendly time series (monthly)
+        try:
+            ts_df = build_timeseries(sql_df)
+        except Exception:
+            ts_df = None
+        
     if "sql" in plan["steps"]:
         sql_df = sql_tool.run(message)
         artifacts["sql_rows"] = int(len(sql_df)) if isinstance(sql_df, pd.DataFrame) else 0
 
     if "predict" in plan["steps"]:
-        predict_df = predict_tool.run(sql_df)
+       predict_df = predict_tool.run(sql_df)
 
     if "explain" in plan["steps"]:
         explain_imgs = explain_tool.run(predict_df or sql_df)
 
+    if "forecast" in plan["steps"] and ts_df is not None:
+        fc = ts_tool.zeroshot_forecast(ts_df[["timestamp","portfolio_value","interest_rate"]].dropna())
+
     report_link = None
     if "report" in plan["steps"]:
         report_link = report_tool.render_and_save(