Update app/main.py

app/main.py

@@ -18,16 +18,16 @@ deployment_name = os.getenv("VITE_AZURE_LLM_DEPLOYMENT")
 async def ask(payload: QuestionRequest):
     question = payload.question
 
-    # 1️⃣ Find top 5 related ayahs using Qdrant
-    verses = find_top_5_ayahs_qdrant(question)
+    # 1️⃣ Get ayahs from Qdrant
+    ayah_results = find_top_5_ayahs_qdrant(question)
 
-    # 2️⃣ Generate reflection using Azure OpenAI
+    # 2️⃣ Generate reflection using Azure LLM
     messages = [
         {
             "role": "system",
             "content": (
-                "You are a compassionate Islamic guide who offers short (≤3 lines) spiritual reflections. "
-                "Speak with empathy, hope, and gentle wisdom. "
+                "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."
             ),
         },
@@ -39,19 +39,18 @@ async def ask(payload: QuestionRequest):
             model=deployment_name,
             messages=messages,
             max_tokens=120,
-            temperature=0.8
+            temperature=0.8,
         )
-        reflection = completion.choices[0].message.content
+        reflection_text = 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
-        ]
-    }
+        reflection_text = f"⚠️ Error generating reflection: {str(e)}"
+
+    # 3️⃣ Return structured response
+    return AnswerListResponse(
+        reflection=ReflectionResponse(reflection=reflection_text),
+        ayahs=ayah_results,
+    )
+
 
 
 @app.get("/")