Update crew.py

crew.py

@@ -73,34 +73,6 @@ def run_crew(question, file_path):
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
     
-    @tool("Image Analysis Tool")
-    def image_analysis_tool(question: str, file_path: str) -> str:
-        """Given a question and image file, analyze the image to answer the question.
-    
-           Args:
-               question (str): Question about an image file
-               file_path (str): The image file path
-                
-           Returns:
-               str: Answer to the question about the image 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=IMAGE_ANALYSIS_MODEL,
-                contents=[file, question]
-            )
-          
-            return response.text
-        except Exception as e:
-            raise RuntimeError(f"Processing failed: {str(e)}")
-
-    ###
     @tool("Image Analysis Tool")
     def image_analysis_tool(question: str, file_path: str) -> str:
         """Given a question and image file, analyze the image to answer the question.
@@ -132,7 +104,33 @@ def run_crew(question, file_path):
             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: