Update crew.py

crew.py

@@ -38,18 +38,18 @@ def run_crew(question, file_path):
 
     @tool("Audio Analysis Tool")
     def audio_analysis_tool(question: str, file_path: str) -> str:
-        """Transcribe the audio file and answer the question.
+        """Answer a question about an audio file.
     
            Args:
                question (str): Question about the audio file
-               file_path (str): Path of audio the file
+               file_path (str): Path of the audio file
                 
            Returns:
                str: Answer to the question about the audio file
                 
            Raises:
-               FileNotFoundError: If audio file does not exist
-               RuntimeError: If transcription fails"""
+               FileNotFoundError: If the audio file does not exist
+               RuntimeError: If processing fails"""
         if not os.path.exists(file_path):
             raise FileNotFoundError(f"Audio file not found: {file_path}")
             
@@ -64,11 +64,41 @@ def run_crew(question, file_path):
             
             return transcript.text
         except Exception as e:
-            raise RuntimeError(f"Failed to transcribe audio: {str(e)}")
+            raise RuntimeError(f"Failed to process audio: {str(e)}")
 
+    @tool("Image Analysis Tool")
+    def image_analysis_tool(question: str, file_path: str) -> str:
+        """Answer a question about an image file.
+    
+           Args:
+               question (str): Question about the image file
+               file_path (str): Path of the image file
+                
+           Returns:
+               str: Answer to the question about the image file
+                
+           Raises:
+               FileNotFoundError: If the image file does not exist
+               RuntimeError: If processing fails"""
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"Image file not found: {file_path}")
+            
+        try:
+            client = OpenAI()
+
+            transcript = client.audio.transcriptions.create(
+                file=open(file_path, "rb"),
+                model=STT_MODEL,
+                prompt=question
+            )
+            
+            return transcript.text
+        except Exception as e:
+            raise RuntimeError(f"Failed to process image: {str(e)}")
+    
     # Built-in tools
 
-    image_analysis_tool = VisionTool()
+    #image_analysis_tool = VisionTool()
     python_coding_tool = CodeInterpreterTool()
     video_analysis_tool = YoutubeVideoSearchTool()
     web_search_tool = SerperDevTool()