Update app.py

app.py

@@ -43,12 +43,12 @@ async def lifespan(app: FastAPI):
         if not GOOGLE_API_KEY:
             raise RuntimeError("CRITICAL: Missing GOOGLE_API_KEY in environment secrets!")
     
-    # Load models into the shared dictionary
-    ml_models["embedder"] = HuggingFaceEmbeddings(model_name="BAAI/bge-base-en-v1.5")
-    cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
-    ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=5)
-    ml_models["llm"] = ChatGoogleGenerativeAI(model="gemini-1.5-flash", api_key=GOOGLE_API_KEY)
-    ml_models["prompt_template"] = PromptTemplate.from_template(
+        # Load models into the shared dictionary
+        ml_models["embedder"] = HuggingFaceEmbeddings(model_name="BAAI/bge-base-en-v1.5")
+        cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
+        ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=5)
+        ml_models["llm"] = ChatGoogleGenerativeAI(model="gemini-1.5-flash", api_key=GOOGLE_API_KEY)
+        ml_models["prompt_template"] = PromptTemplate.from_template(
          """
         You are an expert decision maker Assistant in the domain such as insurance, legal compliance, human resources, and contract management.
 
@@ -85,12 +85,13 @@ Else:
 ##NOTE : Do  not mention document id or its page number just mention clauses if applicable .
         """
     )
-    print("✅ Models and prompt loaded successfully!")
-    
+        print("✅ Models and prompt loaded successfully!")
+    except Exception as e:
+        print("❌ Lifespan error:", str(e))
+        raise e
+
     yield
-    
-    # Code below yield runs on shutdown (optional)
-    print("Shutting down and cleaning up.")
+    print("🧹 Cleaning up.")
     ml_models.clear()
 
 # --- 2. FastAPI App Instance ---