Update app.py

app.py

@@ -164,19 +164,39 @@ class DarijaTherapist:
         max_retries = 3
         for attempt in range(max_retries):
             try:
+                # Validate and clean input
+                if not user_input or len(user_input.strip()) == 0:
+                    return "عذراً، ما فهمتش السؤال ديالك. عاود من فضلك."
+
+                # Limit input length to prevent tensor size issues
+                if len(user_input) > 512:
+                    user_input = user_input[:512]
+
                 response = self.conversation_chain({
                     "question": user_input,
-                    "chat_history": self.memory.chat_memory.messages
+                    "chat_history": self.memory.chat_memory.messages[-5:]  # Limit context window
                 })
                 return response['answer']
+
+            except requests.exceptions.HTTPError as e:
+                if e.response.status_code == 424:
+                    if attempt < max_retries - 1:
+                        st.warning("Model error, retrying with simplified input...")
+                        # Try with shorter context
+                        user_input = user_input[:256]
+                        time.sleep(2 ** attempt)
+                        continue
+                    return "عذراً، كاين مشكل مع النموذج. جرب سؤال أقصر."
+                    
             except requests.exceptions.ReadTimeout:
                 if attempt < max_retries - 1:
                     st.warning(f"Attempt {attempt + 1} timed out, retrying...")
-                    time.sleep(2 ** attempt)  # exponential backoff
+                    time.sleep(2 ** attempt)
                     continue
                 return "عذراً، الخادم بطيء حالياً. حاول مرة أخرى."
+                
             except Exception as e:
-                st.error(f"Error getting AI response: {str(e)}")
+                st.error(f"Error: {str(e)}")
                 return "عذراً، كاين شي مشكل. حاول مرة أخرى."
 
     def run(self):