Update app.py

app.py

@@ -152,27 +152,21 @@ async def ask(query: Query):
 
         loop = asyncio.get_event_loop()
         
-        answer = await asyncio.wait_for(
+        response = await asyncio.wait_for(
             # qa_chain.run(prompt, callbacks=[timeout_callback]),
             loop.run_in_executor(None, qa_chain.run, prompt),
             timeout=360
         )
 
-        if not answer:
+        if not response:
             raise ValueError("Empty answer returned from model")
 
-        if 'Answer:' in answer:
-            response_text = answer.split('Answer:')[-1].strip()
-        elif 'الإجابة:' in answer:
-            response_text = answer.split('الإجابة:')[-1].strip()
-        else:
-            response_text = answer.strip()
-
+        answer = response.split("Answer:")[-1].strip() if "Answer:" in response else response.split("الإجابة:")[-1].strip()
         
         return {
             "status": "success",
+            "response": response,
             "answer": answer,
-            "response": response_text,
             "language": detect(query.question)
         }
 
@@ -195,10 +189,13 @@ def chat(query: Query):
 
     prompt = generate_prompt(query.question)
 
-    answer = qa_chain.run(prompt)
+    response = qa_chain.run(prompt)
 
-    return {
+    answer = response.split("Answer:")[-1].strip() if "Answer:" in response else response.split("الإجابة:")[-1].strip()
+    
 
+    return {
+        "response": response,
         "answer": answer
     }