Upload app.py

app.py

@@ -17,6 +17,8 @@ PREDICTIONS_FILE_T5 = os.path.join(os.path.dirname(__file__), "predictions_t5.js
 
 app = FastAPI(title="HF NIFTY Forecaster Backend")
 
+import threading
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -24,6 +26,36 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+@app.on_event("startup")
+def on_startup():
+    def startup_check():
+        now = datetime.now(IST)
+        today = now.date()
+        current_time = now.time()
+        
+        if not is_trading_day(today):
+            return
+            
+        # If past 09:21 and T+5 predictions are missing, generate them
+        if current_time >= T5_RUN_BUFFER and not os.path.exists(PREDICTIONS_FILE_T5):
+            print("Startup: T+5 predictions missing and it's past 09:21. Backfilling...")
+            try:
+                generate_t5_predictions()
+            except Exception as e:
+                print(f"Startup T+5 generation failed: {e}")
+                
+        # If past 15:45 and EOD predictions are missing, generate them
+        if current_time >= MARKET_CLOSE_BUFFER and not os.path.exists(PREDICTIONS_FILE):
+            print("Startup: EOD predictions missing and it's past 15:45. Backfilling...")
+            try:
+                update_daily_data()
+                generate_predictions()
+            except Exception as e:
+                print(f"Startup EOD generation failed: {e}")
+
+    # Run in a background thread so it doesn't block Uvicorn startup
+    threading.Thread(target=startup_check, daemon=True).start()
+
 def run_update_pipeline(is_t5=False):
     try:
         if is_t5:
@@ -80,10 +112,6 @@ def cron_trigger(background_tasks: BackgroundTasks):
         # Run T+5 pipeline at 09:21
         background_tasks.add_task(run_update_pipeline, is_t5=True)
         return {"status": "triggered", "message": "Morning T+5 forecast pipeline started in the background."}
-    elif not os.path.exists(PREDICTIONS_FILE_T5):
-        # Fallback: if file is missing (e.g. container restart), generate T+5 immediately
-        background_tasks.add_task(run_update_pipeline, is_t5=True)
-        return {"status": "triggered", "message": "Missing T+5 predictions. Generating them now in the background."}
     else:
         return {"status": "skipped", "reason": f"Current time {current_time} is not in the execution windows (09:21-10:00 for T+5, after 15:45 for T+1)."}