Update app.py

app.py

@@ -584,36 +584,39 @@ async def health_check():
 @app.post("/ask")
 async def chat(query: ChatQuery):
     try:
+        # Define constraints
+        constraints = "Provide a medically reliable answer in no more than 250 words."
+
         # Handle Arabic input
         if query.language_code == 0:
             # Translate question from Arabic to English
             english_query = translate_ar_to_en(query.query)
             if not english_query:
                 raise HTTPException(status_code=500, detail="Failed to translate question from Arabic to English")
-            
-            # Get completion in English
-            english_response = get_completion(english_query)
-            
+
+            # Modify the prompt with constraints
+            english_response = get_completion(f"{english_query} {constraints}")
+
             # Translate response back to Arabic
             arabic_response = translate_en_to_ar(english_response)
             if not arabic_response:
                 raise HTTPException(status_code=500, detail="Failed to translate response to Arabic")
-                
+
             return {
                 "original_query": query.query,
                 "translated_query": english_query,
                 "response": arabic_response,
                 "response_in_english": english_response
             }
-            
+
         # Handle English input
         else:
-            response = get_completion(query.query)
+            response = get_completion(f"{query.query} {constraints}")
             return {
                 "query": query.query,
                 "response": response
             }
-            
+
     except HTTPException as e:
         raise e
     except Exception as e: