Update crew.py

crew.py

@@ -1,8 +1,10 @@
 # References:
+
 # https://docs.crewai.com/introduction
 # https://ai.google.dev/gemini-api/docs
 
 import os
+
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
 from google import genai
@@ -23,7 +25,6 @@ IMAGE_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 AUDIO_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 VIDEO_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
 YOUTUBE_ANALYSIS_MODEL  = "gemini-2.5-flash-preview-04-17"
-DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
 CODE_GENERATION_MODEL   = "gemini-2.5-flash-preview-04-17"
 CODE_EXECUTION_MODEL    = "gemini-2.5-flash-preview-04-17"
 
@@ -178,34 +179,6 @@ def run_crew(question, file_path):
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
-    @tool("Document Analysis Tool")
-    def document_analysis_tool(question: str, file_path: str) -> str:
-        """Answer a question about a document file. Supported document types include:
-           .xlxs, .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js, .py
-    
-           Args:
-               question (str): Question about a document file
-               file_path (str): The document file path
-                
-           Returns:
-               str: Answer to the question about the document file
-                
-           Raises:
-               RuntimeError: If processing fails"""
-        try:
-            client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
-
-            file = client.files.upload(file=file_path)
-            
-            response = client.models.generate_content(
-                model=DOCUMENT_ANALYSIS_MODEL, 
-                contents=[file, question]
-            )
-          
-            return response.text
-        except Exception as e:
-            raise RuntimeError(f"Processing failed: {str(e)}")
-
     @tool("Code Generation Tool")
     def code_generation_tool(question: str) -> str:
         """Generate and execute Python code to answer a question.
@@ -326,17 +299,6 @@ def run_crew(question, file_path):
         verbose=False
     )
 
-    document_analysis_agent = Agent(
-        role="Document Analysis Agent",
-        goal="Analyze document of type .xlxs, .txt, .csv, .xml, .rtf, .pdf, .md, .html, .css, .js, .py to help answer question \"{question}\"",
-        backstory="As an expert document analysis assistant, you analyze the document to help answer the question.",
-        allow_delegation=False,
-        llm=AGENT_MODEL,
-        max_iter=2,
-        tools=[document_analysis_tool],
-        verbose=False
-    )
-
     code_generation_agent = Agent(
         role="Code Generation Agent",
         goal="Generate Python code and execute it to help answer question \"{question}\"",
@@ -387,7 +349,6 @@ def run_crew(question, file_path):
                 audio_analysis_agent, 
                 video_analysis_agent, 
                 youtube_analysis_agent, 
-                document_analysis_agent,
                 code_generation_agent,
                 code_execution_agent],
         manager_agent=manager_agent,