Update crew.py

crew.py

@@ -303,7 +303,7 @@ def run_crew(question, file_path):
         backstory="As an expert Python code generation assistant, you generate code to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
-        max_iter=2,
+        max_iter=5,
         tools=[code_generation_tool],
         verbose=False
     )
@@ -314,7 +314,7 @@ def run_crew(question, file_path):
         backstory="As an expert Python code execution assistant, you execute the code file to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
-        max_iter=2,
+        max_iter=5,
         tools=[code_execution_tool],
         verbose=False
     )
@@ -362,12 +362,8 @@ def run_crew(question, file_path):
         if df.empty:
             question = f"{question} File path: {file_path}."
         else:
-            question = f"{question} File data: {df.to_string()}" # load data due to sandbox contraints
+            question = f"{question} File data: {df.to_string()}" # sandbox contraints
 
-    print("###")
-    print(question)
-    print("###")
-    
     initial_answer = crew.kickoff(inputs={"question": question})
     final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(initial_answer))