Update app/main.py

app/main.py

@@ -424,9 +424,17 @@ async def ask(query: Query):
         if not answer:
             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()
+
+        
         return {
             "status": "success",
-            "response": str(answer.split('Answer:' if 'Answer:' in answer else "الإجابة:")[:-1]),
+            "response": response_text,
             "language": detect(query.question)
         }