Update app.py

app.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, HTTPException, Header
+from contextlib import asynccontextmanager
 import pandas as pd
 import os
 import joblib
@@ -7,7 +8,14 @@ import requests
 from huggingface_hub import hf_hub_download
 from prometheus_client import Counter, Histogram, Gauge, generate_latest
 
-app = FastAPI(title="Loan Approval API", version="1.0")
+# Global variables
+model = None
+encoders = {}
+scaler = None
+feature_columns = []
+categorical_columns = []
+boolean_columns = []
+loaded = False
 
 # Environment variables
 API_KEY = os.getenv("API_KEY", "test-key-123")
@@ -19,18 +27,10 @@ REQS = Counter("pred_requests_total", "Total prediction requests")
 LAT = Histogram("pred_request_latency_seconds", "Request latency")
 LATEST = Gauge("latest_prediction", "Last predicted value")
 
-# Global variables
-model = None
-encoders = {}
-scaler = None
-feature_columns = []
-categorical_columns = []
-boolean_columns = []
-loaded = False
 
-
-@app.on_event("startup")
-async def load_model():
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Startup
     global model, encoders, scaler, feature_columns, categorical_columns, boolean_columns, loaded
     
     print("🚀 Starting FastAPI application...")
@@ -66,7 +66,15 @@ async def load_model():
             print(f" Model load error: {ex}")
             loaded = False
     
-    print(" FastAPI app initialized")
+    print(" FastAPI app initialized and ready!")
+    
+    yield  # Server runs here
+    
+    # Shutdown (cleanup if needed)
+    print("🛑 Shutting down...")
+
+
+app = FastAPI(title="Loan Approval API", version="1.0", lifespan=lifespan)
 
 
 @app.get("/")