Update crew.py

crew.py

@@ -102,7 +102,7 @@ def run_crew(question, file_path):
 
     ###
     @tool("Image Analysis Tool")
-    def image_analysis_tool_2(question: str, file_path: str) -> str:
+    def image_analysis_tool(question: str, file_path: str) -> str:
         """Given a question and image file, analyze the image to answer the question.
     
            Args:
@@ -125,43 +125,14 @@ def run_crew(question, file_path):
             completion = client.chat.completions.create(
                 model = "gpt-4.5-preview",
                 messages = [{"role": "user",
-                             "content": [{"type": "text", 
-                                          "text": question},
-                                         {"type": "image_url",
-                                          "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}}]}]
+                             "content": [{"type": "text", "text": question},
+                                         {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}}]}]
             )
         
             content = completion.choices[0].message.content
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
     ###
-    
-    @tool("Audio Analysis Tool")
-    def audio_analysis_tool(question: str, file_path: str) -> str:
-        """Given a question and audio file, analyze the audio to answer the question.
-    
-           Args:
-               question (str): Question about an audio file
-               file_path (str): The audio file path
-                
-           Returns:
-               str: Answer to the question about the audio 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=AUDIO_ANALYSIS_MODEL, 
-                contents=[file, question]
-            )
-          
-            return response.text
-        except Exception as e:
-            raise RuntimeError(f"Processing failed: {str(e)}")
 
     @tool("Video Analysis Tool")
     def video_analysis_tool(question: str, file_path: str) -> str:
@@ -298,7 +269,7 @@ def run_crew(question, file_path):
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=2,
-        tools=[image_analysis_tool_2],
+        tools=[image_analysis_tool],
         verbose=True
     )
 
@@ -395,7 +366,7 @@ def run_crew(question, file_path):
                 code_execution_agent],
         manager_agent=manager_agent,
         tasks=[manager_task],
-        verbose=False
+        verbose=True
     )
 
     # Process