Upload app.py

app.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import asyncio
-from datetime import date
+from datetime import date, datetime, time
 
 import sys
 from pathlib import Path
@@ -10,7 +10,7 @@ from fastapi import FastAPI, Query
 from fastapi.middleware.cors import CORSMiddleware
 
 sys.path.insert(0, str(Path(__file__).resolve().parent))
-from nifty_backend.runtime import dashboard_payload, latest_saved_prediction, refresh_daily_data, refresh_first5_prediction, seconds_until_next_ist_run
+from nifty_backend.runtime import dashboard_payload, latest_saved_prediction, refresh_daily_data, refresh_first5_prediction, seconds_until_next_ist_run, IST
 
 
 app = FastAPI(title="NIFTY 50 Forecaster Backend")
@@ -23,21 +23,52 @@ app.add_middleware(
 )
 
 
+market_status = "Waiting for next session"
+
 async def daily_ist_refresh_loop() -> None:
+    global market_status
     while True:
-        await asyncio.sleep(seconds_until_next_ist_run())
+        # Wait until 9:00 AM IST
+        await asyncio.sleep(seconds_until_next_ist_run(time(9, 0)))
+        market_status = "Market Pre-Open"
+        print("[scheduler] 9:00 AM IST - Market Pre-Open", flush=True)
+
+        # Wait until 9:15 AM IST
+        await asyncio.sleep(seconds_until_next_ist_run(time(9, 15)))
+        market_status = "Market Officially Opened"
+        print("[scheduler] 9:15 AM IST - Market Officially Opened", flush=True)
+
+        # Wait until 9:20 AM IST
+        await asyncio.sleep(seconds_until_next_ist_run(time(9, 20)))
+        market_status = "Fetching T+5 Prediction Data..."
+        print("[scheduler] 9:20 AM IST - Fetching Data", flush=True)
+
         try:
             await asyncio.to_thread(refresh_first5_prediction)
+            market_status = "Prediction Ready"
         except Exception as exc:
             print(f"[scheduler] first5 refresh failed: {exc}", flush=True)
+            market_status = "Prediction Failed"
+            
         try:
             await asyncio.to_thread(refresh_daily_data)
         except Exception as exc:
             print(f"[scheduler] daily refresh failed: {exc}", flush=True)
 
-
 @app.on_event("startup")
 async def start_scheduler() -> None:
+    global market_status
+    # Initialize correct status on startup based on current time
+    now = datetime.now(IST).time()
+    if now < time(9, 0):
+        market_status = "Waiting for 9:00 AM"
+    elif now < time(9, 15):
+        market_status = "Market Pre-Open"
+    elif now < time(9, 20):
+        market_status = "Market Officially Opened"
+    else:
+        market_status = "Prediction Ready"
+        
     asyncio.create_task(daily_ist_refresh_loop())
 
 
@@ -53,7 +84,12 @@ def root() -> dict[str, str]:
 
 @app.get("/dashboard")
 def dashboard() -> dict:
-    return dashboard_payload()
+    global market_status
+    payload = dashboard_payload()
+    if "data_status" not in payload:
+        payload["data_status"] = {}
+    payload["data_status"]["market_status"] = market_status
+    return payload
 
 
 @app.get("/cron/keepalive")