Update app.py

app.py

@@ -91,12 +91,19 @@ agent_executor = AgentExecutor(
     agent=agent,
     tools=tools_add,
     verbose=True,
+    max_iterations = 10,
+    
 )
 
 def data_gov_ma(message, history):
   try:
     response = agent_executor.invoke({"input": message})
-    return response['output']
+    final_response = response['output']
+    timeout_iteration_error = 'Agent stopped due to iteration limit or time limit.'
+    if final_response == timeout_iteration_error:
+        return "Je suis désolé, je n'ai pas compris votre question.Pourriez-vous la reformuler s'il vous plaît ?"
+    else:
+        return final_response
   except ValueError as e:
     return "Je suis désolé, je n'ai pas compris votre question.Pourriez-vous la reformuler s'il vous plaît ?"