Update crew.py

crew.py

@@ -11,7 +11,7 @@ from google import genai
 from google.genai import types
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
-from util import get_final_answer
+from util import read_file, get_final_answer
 
 ## LLMs
 
@@ -182,7 +182,7 @@ def run_crew(question, file_path):
     @tool("Code Generation Tool")
     def code_generation_tool(question: str, file_path: str) -> str:
         """Generate and execute Python code to answer a question.
-           The question can be about an optional document.
+           The question can be about an optional csv, xls, xlsx, json, or jsonl document.
     
            Args:
                question (str): Question to answer
@@ -195,20 +195,9 @@ def run_crew(question, file_path):
                RuntimeError: If processing fails"""
         try:
             if file_path:
-                df = pd.read_excel("data.xlsx")
-                data = f"data: {df.head().to_json()}"
-                """
-                pd.read_csv() for CSV files
-
-                pd.read_excel() for Excel files (.xls, .xlsx)
-                
-                pd.read_json() for JSON files
-                
-                pd.read_parquet() for Parquet files
-                
-                pd.read_sql() for SQL queries
-                """
-                
+                df = read_file(file_path)
+                question = f"Question:\n{question}\nData:\n{df.to_string()}"
+            
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
             file = client.files.upload(file=file_path)