Update crew.py

crew.py

@@ -23,6 +23,7 @@ IMAGE_MODEL        = "gemini-2.0-flash"
 AUDIO_MODEL        = "gemini-2.0-flash"
 VIDEO_MODEL        = "gemini-2.0-flash"
 YOUTUBE_MODEL      = "gemini-2.0-flash"
+DOCUMENT_MODEL     = "gemini-2.0-flash"
 
 # LLM evaluation
 
@@ -147,6 +148,34 @@ 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:
+           .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="GOOGLE_API_KEY")
+
+            file = client.files.upload(file=file_path)
+            
+            response = client.models.generate_content(
+                model=DOCUMENT_MODEL, 
+                contents=[file, question]
+            )
+          
+            return response.text
+        except Exception as e:
+            raise RuntimeError(f"Processing failed: {str(e)}")
     
     web_search_tool = SerperDevTool()
     web_rag_tool = WebsiteSearchTool()
@@ -165,39 +194,39 @@ def run_crew(question, file_path):
         verbose=False
     )
 
-    audio_analysis_agent = Agent(
-        role="Audio Analysis Agent",
-        goal="Analyze audio to help answer question \"{question}\"",
-        backstory="As an expert audio analysis assistant, you analyze the audio to help answer the question.",
+    image_analysis_agent = Agent(
+        role="Image Analysis Agent",
+        goal="Analyze image file to help answer question \"{question}\"",
+        backstory="As an expert image analysis assistant, you analyze the image file to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
-        tools=[audio_analysis_tool],
+        tools=[image_analysis_tool],
         verbose=False
     )
 
-    image_analysis_agent = Agent(
-        role="Image Analysis Agent",
-        goal="Analyze image to help answer question \"{question}\"",
-        backstory="As an expert image analysis assistant, you analyze the image to help answer the question.",
+    audio_analysis_agent = Agent(
+        role="Audio Analysis Agent",
+        goal="Analyze audio file to help answer question \"{question}\"",
+        backstory="As an expert audio analysis assistant, you analyze the audio file to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
-        tools=[image_analysis_tool],
+        tools=[audio_analysis_tool],
         verbose=False
     )
 
-    youtube_audio_analysis_agent = Agent(
-        role="YouTube Audio Analysis Agent",
-        goal="Analyze YouTube video to help answer audio question \"{question}\"",
-        backstory="As an expert YouTube audio analysis assistant, you analyze the video to help answer the question.",
+    video_analysis_agent = Agent(
+        role="Video Analysis Agent",
+        goal="Analyze video file to help answer question \"{question}\"",
+        backstory="As an expert video analysis assistant, you analyze the video file to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
-        tools=[youtube_audio_analysis_tool],
+        tools=[video_analysis_tool],
         verbose=False
     )
-
+    
     youtube_analysis_agent = Agent(
         role="YouTube Analysis Agent",
         goal="Analyze YouTube video to help answer question \"{question}\"",
@@ -205,7 +234,18 @@ def run_crew(question, file_path):
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
-        tools=[youtube_image_analysis_tool],
+        tools=[youtube_analysis_tool],
+        verbose=False
+    )
+
+    document_analysis_agent = Agent(
+        role="Document Analysis Agent",
+        goal="Analyze document of type .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
     )
     
@@ -248,6 +288,7 @@ def run_crew(question, file_path):
                 audio_analysis_agent, 
                 video_analysis_agent, 
                 youtube_analysis_agent, 
+                document_analysis_agent, 
                 python_coding_agent],
         manager_agent=manager_agent,
         tasks=[manager_task],