Update app/main.py

app/main.py

@@ -1,20 +1,59 @@
 from fastapi import FastAPI
 from app.rag import find_top_5_ayahs_qdrant
 from app.models.schemas import QuestionRequest, AnswerListResponse
-import psutil, os, asyncio
+from openai import AzureOpenAI
+import os
 
 app = FastAPI()
 
-
-
+# --- Azure OpenAI setup ---
+client = AzureOpenAI(
+    api_key=os.getenv("AZURE_LLM_KEY"),
+    api_version="2024-05-01-preview",
+    azure_endpoint=os.getenv("AZURE_LLM_ENDPOINT")
+)
+deployment_name = os.getenv("AZURE_LLM_DEPLOYMENT")
 
 @app.post("/ask/", response_model=AnswerListResponse)
-def ask(payload: QuestionRequest):
-    results = find_top_5_ayahs_qdrant(payload.question)
-    return {"results": results}
+async def ask(payload: QuestionRequest):
+    question = payload.question
+
+    # 1️⃣ Find top 5 related ayahs using Qdrant
+    verses = find_top_5_ayahs_qdrant(question)
+
+    # 2️⃣ Generate reflection using Azure OpenAI
+    messages = [
+        {
+            "role": "system",
+            "content": (
+                "You are a compassionate Islamic guide who offers short (≤3 lines) spiritual reflections. "
+                "Speak with empathy, hope, and gentle wisdom. "
+                "Do NOT quote or refer to Qur’an verses, as they are provided separately."
+            ),
+        },
+        {"role": "user", "content": question},
+    ]
+
+    try:
+        completion = client.chat.completions.create(
+            model=deployment_name,
+            messages=messages,
+            max_tokens=120,
+            temperature=0.8
+        )
+        reflection = completion.choices[0].message.content
+    except Exception as e:
+        reflection = f"⚠️ Error generating reflection: {str(e)}"
+
+    # 3️⃣ Combine reflection + verses
+    return {
+        "results": [
+            {"type": "reflection", "answer": reflection},
+            *verses  # Append ayahs from Qdrant
+        ]
+    }
+
 
 @app.get("/")
 def read_root():
     return {"message": "Hello, Mustafa! FastAPI is running 🚀"}
-
-