Upload 5 files

app.py

@@ -7,18 +7,12 @@ from zoneinfo import ZoneInfo
 from data_updater import update_daily_data, is_trading_day
 from forecaster_engine import generate_predictions
 
-from t5_engine import generate_t5_predictions
-
 IST = ZoneInfo("Asia/Kolkata")
 MARKET_CLOSE_BUFFER = time(15, 45) # Update runs after 3:45 PM
-T5_RUN_BUFFER = time(9, 21) # T+5 runs after 9:21 AM
 PREDICTIONS_FILE = os.path.join(os.path.dirname(__file__), "predictions.json")
-PREDICTIONS_FILE_T5 = os.path.join(os.path.dirname(__file__), "predictions_t5.json")
 
 app = FastAPI(title="HF NIFTY Forecaster Backend")
 
-import threading
-
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -26,50 +20,16 @@ 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):
+def run_update_pipeline():
+    try:
+        # Step 1: Update data
+        res = update_daily_data()
+        if res.get("status") == "error":
+            print(f"Update failed: {res.get('reason')}")
             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:
-            # Step 1: Generate T+5 predictions
-            generate_t5_predictions()
-        else:
-            # Step 1: Update daily data
-            res = update_daily_data()
-            if res.get("status") == "error":
-                print(f"Update failed: {res.get('reason')}")
-                return
-                
-            # Step 2: Generate T+1 predictions
-            generate_predictions()
+        # Step 2: Generate predictions
+        generate_predictions()
     except Exception as e:
         print(f"Pipeline error: {e}")
 
@@ -83,16 +43,6 @@ def get_predictions():
         
     return data
 
-@app.get("/t5-predictions")
-def get_t5_predictions():
-    if not os.path.exists(PREDICTIONS_FILE_T5):
-        raise HTTPException(status_code=404, detail="T+5 Predictions not yet generated")
-    
-    with open(PREDICTIONS_FILE_T5, "r") as f:
-        data = json.load(f)
-        
-    return data
-
 @app.post("/cron/update")
 def cron_trigger(background_tasks: BackgroundTasks):
     now = datetime.now(IST)
@@ -103,17 +53,14 @@ def cron_trigger(background_tasks: BackgroundTasks):
     if not is_trading_day(today):
         return {"status": "skipped", "reason": f"{today} is a holiday or weekend"}
         
-    # Determine which pipeline to run
-    if current_time >= MARKET_CLOSE_BUFFER:
-        # Run standard T+1 end-of-day pipeline
-        background_tasks.add_task(run_update_pipeline, is_t5=False)
-        return {"status": "triggered", "message": "End-of-day T+1 update and forecast pipeline started in the background."}
-    elif current_time >= T5_RUN_BUFFER and current_time < time(10, 0):
-        # 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."}
-    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)."}
+    # 2. Check if it's past 3:45 PM
+    if current_time < MARKET_CLOSE_BUFFER:
+        return {"status": "skipped", "reason": "Market is still open or buffer not reached. Runs after 3:45 PM IST."}
+        
+    # Trigger the full pipeline in the background so Netlify doesn't timeout
+    background_tasks.add_task(run_update_pipeline)
+    
+    return {"status": "triggered", "message": "Update and forecast pipeline started in the background."}
 
 @app.get("/health")
 def health_check():

requirements.txt

@@ -5,5 +5,3 @@ requests
 pandas_market_calendars
 pyarrow
 fastparquet
-scikit-learn
-joblib