Update api.py

api.py

@@ -1,4 +1,3 @@
-
 # Install necessary libraries if not already present
 # These lines will be executed as shell commands by %%writefile
 # !pip install duckduckgo_search dateparser
@@ -33,6 +32,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
 from fastapi import FastAPI, Request, HTTPException, Depends, Security
 from fastapi.security.api_key import APIKeyHeader
 from dotenv import load_dotenv # For loading environment variables from a .env file
+from fastapi.responses import JSONResponse # Import JSONResponse
 
 # Load environment variables from .env file (if it exists)
 load_dotenv()
@@ -498,7 +498,7 @@ Query: {query}
     for question in cleaned_questions:
         print(f"\nAnalyzing question for tool determination: '{question}'")
         determined_tools[question] = determine_tool_usage(question)
-        print(f"Determined tool for '{question}': '{tool}'")
+        print(f"Determined tool for '{question}': '{determined_tools[question]}'") # Corrected print statement
 
     print("\nSummary of determined tools per question:")
     for question, tool in determined_tools.items():
@@ -583,10 +583,44 @@ async def chat_endpoint(request: Request, api_key: str = Depends(get_api_key)):
         print(traceback.format_exc())
         raise HTTPException(status_code=500, detail=f"Internal server error: {e}")
 
-# Optional: Root endpoint for health check
+# Health Check Endpoint
+@app.get("/health")
+async def health_check():
+    """
+    Health check endpoint to verify the application is running and essential components are loaded.
+    """
+    status = {
+        "status": "ok",
+        "llm_client_initialized": client is not None,
+        "business_info_loaded": business_info_available,
+        "spacy_loaded": nlp is not None,
+        "embedder_loaded": embedder is not None,
+        "reranker_loaded": reranker is not None,
+        "secrets_loaded": {
+            "HF_TOKEN": HF_TOKEN is not None,
+            "SHEET_ID": SHEET_ID is not None,
+            "GOOGLE_BASE64_CREDENTIALS": GOOGLE_BASE64_CREDENTIALS is not None,
+            "API_KEY": API_KEY is not None,
+        }
+    }
+    unhealthy_components = [key for key, value in status.items() if isinstance(value, bool) and not value]
+    if status["secrets_loaded"] and not all(status["secrets_loaded"].values()):
+         unhealthy_components.append("secrets_loaded (partial)")
+
+    if unhealthy_components:
+         status["status"] = "unhealthy"
+         status["unhealthy_components"] = unhealthy_components
+         return JSONResponse(status=503, content=status) # Return 503 Service Unavailable if unhealthy
+
+    return status # Return 200 OK if healthy
+
+
+# Optional: Root endpoint for basic info
 @app.get("/")
 async def read_root():
-    # Add a check for essential components
+    """
+    Root endpoint providing basic application information and status.
+    """
     status = {
         "message": "LLM with Tools API is running",
         "llm_client_initialized": client is not None,