Upload 2 files

app.py

@@ -79,6 +79,18 @@ def cron_trigger(background_tasks: BackgroundTasks):
     return {"status": "triggered", "message": "Update and forecast pipeline started in the background."}
 
 # ── NEW: T5 Forecaster Endpoints ─────────────────────────────────────────────
+import math
+from fastapi.responses import JSONResponse
+
+def _sanitize_for_json(obj):
+    """Recursively replace NaN/Inf floats with None for JSON compliance."""
+    if isinstance(obj, dict):
+        return {k: _sanitize_for_json(v) for k, v in obj.items()}
+    elif isinstance(obj, list):
+        return [_sanitize_for_json(v) for v in obj]
+    elif isinstance(obj, float) and (math.isnan(obj) or math.isinf(obj)):
+        return None
+    return obj
 
 @app.get("/t5/predictions")
 def get_t5_predictions():
@@ -88,6 +100,9 @@ def get_t5_predictions():
     
     with open(T5_PREDICTIONS_FILE, "r") as f:
         data = json.load(f)
+    
+    # Sanitize NaN/Inf values that break FastAPI's JSON serializer
+    data = _sanitize_for_json(data)
         
     return data
 

t5_engine.py

@@ -1,5 +1,6 @@
 import os
 import json
+import math
 import time
 import requests
 import pandas as pd
@@ -8,6 +9,18 @@ from zoneinfo import ZoneInfo
 import pandas_market_calendars as mcal
 import numpy as np
 
+
+def _sanitize_value(v):
+    """Replace NaN/Inf floats with 0 so JSON serialization doesn't break."""
+    if isinstance(v, float) and (math.isnan(v) or math.isinf(v)):
+        return 0.0
+    return v
+
+
+def _sanitize_dict(d):
+    """Recursively sanitize a dict of float values."""
+    return {k: _sanitize_dict(v) if isinstance(v, dict) else _sanitize_value(v) for k, v in d.items()}
+
 IST = ZoneInfo("Asia/Kolkata")
 BASE_DIR = os.path.dirname(__file__)
 DATA_DIR = os.path.join(BASE_DIR, "data")
@@ -130,6 +143,9 @@ def run_t5_pipeline():
         features['hl_spread'] = (high_5m - low_5m) / open_5m if open_5m else 0
         features['prev_ret'] = (prev_close - prev_open) / prev_open if prev_open else 0
         
+        # Sanitize NaN/Inf values that break JSON serialization
+        features = _sanitize_dict(features)
+        
         # Evaluate rule
         rule_item = rules_dict.get(ticker)
         if rule_item: