Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import logging
+import time
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import JSONResponse
 from langchain_google_genai import ChatGoogleGenerativeAI, GoogleGenerativeAIEmbeddings
@@ -29,8 +30,10 @@ if not os.getenv("GOOGLE_API_KEY"):
     logger.error("GOOGLE_API_KEY environment variable not set.")
     raise ValueError("GOOGLE_API_KEY environment variable not set.")
 
-# Initialize Gemini LLM
+# Log dependency loading
+logger.info("Loading Gemini dependencies...")
 try:
+    # Initialize Gemini LLM
     llm = ChatGoogleGenerativeAI(
         model="gemini-1.5-flash",
         google_api_key=os.getenv("GOOGLE_API_KEY")
@@ -40,8 +43,8 @@ except Exception as e:
     logger.error(f"Failed to initialize Gemini LLM: {str(e)}")
     raise
 
-# Initialize embeddings
 try:
+    # Initialize embeddings
     embeddings = GoogleGenerativeAIEmbeddings(
         model="models/embedding-001",
         google_api_key=os.getenv("GOOGLE_API_KEY")
@@ -128,4 +131,6 @@ async def ask_question(query: str):
 async def health_check():
     """Health check endpoint."""
     logger.info("Health check requested.")
+    # Add slight delay to ensure stability
+    time.sleep(0.5)
     return {"status": "API is running"}