cleanup: remove debug tracing, restore clean health endpoint

app/main.py

@@ -1,8 +1,6 @@
 """StockPro ML Backend — FastAPI app for Hugging Face Spaces."""
-import traceback
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
 
 from app.routers import predict, anomaly
 
@@ -23,10 +21,6 @@ app.include_router(predict.router, tags=["prediction"])
 app.include_router(anomaly.router, tags=["anomaly"])
 
 
-@app.exception_handler(Exception)
-async def global_exception_handler(request: Request, exc: Exception):
-    return JSONResponse(status_code=500, content={"error": traceback.format_exc()})
-
 
 @app.get("/")
 def root():
@@ -35,44 +29,4 @@ def root():
 
 @app.get("/health")
 def health():
-    import os
-    import app.config as cfg
-    from huggingface_hub import hf_hub_download
-
-    model_path = os.path.join(cfg.MODEL_DIR, "tft_stock.ckpt")
-    params_path = os.path.join(cfg.MODEL_DIR, "dataset_params.pt")
-
-    download_error = None
-    if not os.path.exists(model_path) and cfg.HF_TOKEN and cfg.MODEL_REPO:
-        try:
-            os.makedirs(cfg.MODEL_DIR, exist_ok=True)
-            hf_hub_download(
-                repo_id=cfg.MODEL_REPO,
-                filename="tft_stock.ckpt",
-                token=cfg.HF_TOKEN,
-                local_dir=cfg.MODEL_DIR,
-            )
-            hf_hub_download(
-                repo_id=cfg.MODEL_REPO,
-                filename="dataset_params.pt",
-                token=cfg.HF_TOKEN,
-                local_dir=cfg.MODEL_DIR,
-            )
-            hf_hub_download(
-                repo_id=cfg.MODEL_REPO,
-                filename="ddg_da.pt",
-                token=cfg.HF_TOKEN,
-                local_dir=cfg.MODEL_DIR,
-            )
-        except Exception as e:
-            download_error = str(e)
-
-    return {
-        "status": "ok",
-        "hf_token_set": bool(cfg.HF_TOKEN),
-        "hf_model_repo": cfg.MODEL_REPO or "(not set)",
-        "model_exists": os.path.exists(model_path),
-        "params_exists": os.path.exists(params_path),
-        "model_dir": cfg.MODEL_DIR,
-        "download_error": download_error,
-    }
+    return {"status": "ok"}

app/routers/predict.py

@@ -75,9 +75,10 @@ async def predict(req: PredictRequest, x_api_key: str = Header(default="")):
             dataset_params_path=DATASET_PARAMS_PATH,
             symbol=data["symbol"],
         )
+    except ValueError as e:
+        raise HTTPException(status_code=422, detail=str(e))
     except Exception as e:
-        import traceback
-        raise HTTPException(status_code=500, detail=traceback.format_exc())
+        raise HTTPException(status_code=500, detail=str(e))
 
     # ── DDG-DA: concept drift detection ──
     # Drift score and flag are added to the response for transparency.