Update crew.py

crew.py

@@ -4,7 +4,7 @@
 # https://ai.google.dev/gemini-api/docs
 
 import os
-
+import pandas as pd
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
 from google import genai
@@ -180,18 +180,35 @@ def run_crew(question, file_path):
             raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Code Generation Tool")
-    def code_generation_tool(question: str) -> str:
+    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.
     
            Args:
                question (str): Question to answer
                 
            Returns:
                str: Answer to the question
-                
+               file_path (str): Optional document file path
+               
            Raises:
                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
+                """
+                
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
             file = client.files.upload(file=file_path)