Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import json
 import uvicorn
 import pandas as pd
-import numpy as np
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel, ConfigDict
@@ -33,7 +32,7 @@ app.add_middleware(
 # ==========================================
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-ARTIFACTS_DIR = os.path.normpath(os.path.join(BASE_DIR, "..", "artifacts_final"))
+ARTIFACTS_DIR = os.path.join(BASE_DIR, "artifacts_final")  # Hugging Face compatible path
 
 print("ARTIFACTS DIR:", ARTIFACTS_DIR)
 print("EXISTS:", os.path.exists(ARTIFACTS_DIR))
@@ -56,17 +55,22 @@ try:
 
     # B. Load Metadata
     meta_path = os.path.join(ARTIFACTS_DIR, "catboost_metadata.json")
+    if not os.path.exists(meta_path):
+        raise FileNotFoundError(f"Metadata not found at {meta_path}")
+
     with open(meta_path, "r") as f:
         metadata = json.load(f)
 
-    # ✅ Correct key from your metadata
-    TOP_FEATURES = metadata["features_used"]
-    THRESHOLD = metadata["thresholds"]["optimal_balanced"]
+    TOP_FEATURES = metadata.get("features_used", [])
+    THRESHOLD = metadata.get("thresholds", {}).get("optimal_balanced", 0.5)
 
     print(f" Metadata Loaded. Threshold set to: {THRESHOLD}")
 
     # C. Load UI Schema
     ui_path = os.path.join(ARTIFACTS_DIR, "model_ui_schema.json")
+    if not os.path.exists(ui_path):
+        raise FileNotFoundError(f"UI schema not found at {ui_path}")
+
     with open(ui_path, "r") as f:
         ui_schema = json.load(f)
     print(" UI Schema Loaded.")