Update app.py

app.py

@@ -503,7 +503,7 @@ def chat_with_llm(df,question):
 def bind_llm(llm, tools,prompt_template):
     llm = llm.bind()
     agent = create_tool_calling_agent(llm, tools, ChatPromptTemplate.from_template(prompt_template))
-    agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)
+    agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, max_iterations=3)
     return agent_executor
 
 # Define input and output models using Pydantic
@@ -705,6 +705,13 @@ def answer_question_thread(user_question, chatbot,audio=None):
     while iterations < max_iterations:
 
         response = agent_executor.invoke({"input": user_question}, config={"callbacks": [langfuse_handler]}, early_stopping_method="generate")
+        # After invoking, check if max iterations are reached
+        if agent_executor.iterations >= agent_executor.max_iterations:
+             if agent_executor.history:
+                return agent_executor.history[-1].get("output")  # Extract the last response from the history
+   
+
+        
         #create_file_HF()
         if isinstance(response, dict):
             response_text = response.get("output", "")