Update app.py

app.py

@@ -1,6 +1,5 @@
 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
@@ -30,10 +29,8 @@ 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.")
 
-# Log dependency loading
-logger.info("Loading Gemini dependencies...")
+# Initialize Gemini LLM
 try:
-    # Initialize Gemini LLM
     llm = ChatGoogleGenerativeAI(
         model="gemini-1.5-flash",
         google_api_key=os.getenv("GOOGLE_API_KEY")
@@ -43,8 +40,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")
@@ -131,6 +128,4 @@ async def ask_question(query: str):
 async def health_check():
     """Health check endpoint."""
     logger.info("Health check requested.")
-    # Add delay to ensure server stability
-    time.sleep(1)
     return {"status": "API is running"}