Update crew.py

crew.py

@@ -4,30 +4,23 @@
 # https://ai.google.dev/gemini-api/docs
 
 import os
-import pandas as pd
-from crewai import Agent, Crew, Process, Task
+from crewai import Agent, Crew, Task
 from crewai.agents.agent_builder.base_agent import BaseAgent
 from crewai.project import CrewBase, agent, crew, task
 from google import genai
-from google.genai import types
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from tools.ai_tools import AITools
 from tools.arithmetic_tools import ArithmeticTools
 from typing import List
-from utils import (
-    read_file_json, 
-    read_docx_text, 
-    read_pptx_text, 
-    is_ext
-)
+from utils import read_file_json, is_ext
 
 ## LLMs
 
-MANAGER_MODEL           = "gpt-4.5-preview"
-AGENT_MODEL             = "gpt-4.1-mini"
+MANAGER_MODEL      = "gpt-4.5-preview"
+AGENT_MODEL        = "gpt-4.1-mini"
 
-FINAL_ANSWER_MODEL      = "gemini-2.5-pro-preview-03-25"
+FINAL_ANSWER_MODEL = "gemini-2.5-pro-preview-03-25"
 
 # LLM evaluation
 
@@ -126,24 +119,24 @@ class GAIACrew():
         )
 
     @agent
-    def code_execution_agent(self) -> Agent:
+    def code_generation_agent(self) -> Agent:
         return Agent(
-            config=self.agents_config["code_execution_agent"],
+            config=self.agents_config["code_generation_agent"],
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=3,
-            tools=[AITools.code_execution_tool],
+            tools=[AITools.code_generation_tool],
             verbose=True
         )
 
     @agent
-    def code_generation_agent(self) -> Agent:
+    def code_execution_agent(self) -> Agent:
         return Agent(
-            config=self.agents_config["code_generation_agent"],
+            config=self.agents_config["code_execution_agent"],
             allow_delegation=False,
             llm=AGENT_MODEL,
             max_iter=3,
-            tools=[AITools.code_generation_tool],
+            tools=[AITools.code_execution_tool],
             verbose=True
         )
 
@@ -175,7 +168,10 @@ def run_crew(question, file_path):
     final_question = question
     
     if file_path:
-        if is_ext(file_path, ".csv") or is_ext(file_path, ".xls") or is_ext(file_path, ".xlsx") or is_ext(file_path, ".json") or is_ext(file_path, ".jsonl"):
+        if is_ext(file_path, ".csv") or 
+           is_ext(file_path, ".xls") or 
+           is_ext(file_path, ".xlsx") or 
+           is_ext(file_path, ".json") or is_ext(file_path, ".jsonl"):
             json_data = read_file_json(file_path)
             final_question = f"{question} JSON data:\n{json_data}."
         else: