Update main.py

main.py

@@ -34,12 +34,19 @@ def load_components():
     global llm, embeddings, vectorstore, retriever, chain
 
     try:
+        # Ensure API key is provided
+        api_key = os.getenv("API_KEY")
+        if not api_key:
+            logger.error("API_KEY environment variable is not set or empty.")
+            raise RuntimeError("API_KEY environment variable is not set or empty.")
+        logger.info("API_KEY is set.")
+
         # 1) Init LLM & Embeddings
         llm = ChatGroq(
             model="meta-llama/llama-4-scout-17b-16e-instruct",
             temperature=0,
             max_tokens=1024,
-            api_key=os.getenv("API_KEY"),
+            api_key=api_key,
         )
         embeddings = HuggingFaceEmbeddings(
             model_name="intfloat/multilingual-e5-large",
@@ -109,15 +116,16 @@ Your response:
 
 @app.get("/")
 def root():
-    return {"message": "Arabic Hadith Finder API is up and running!"}
+    return {"message": "API is up and running!"}
 
 @app.post("/query")
 def query(request: QueryRequest):
     try:
-        logger.info(f"Received query: %s", request.question)
+        logger.info("Received query: %s", request.question)
         result = chain.invoke({"query": request.question})
         logger.info("Query processed successfully.")
         return {"answer": result.get("result")}
     except Exception as e:
         logger.error("Error processing query", exc_info=True)
-        raise HTTPException(status_code=500, detail="Internal server error.")
+        # Return detailed error for debugging
+        raise HTTPException(status_code=500, detail=str(e))