Update crew.py

crew.py

@@ -61,16 +61,49 @@ def run_crew(question, file_path):
             transcript = client.audio.transcriptions.create(
                 file=open(file_path, "rb"),
                 model=AUDIO_MODEL,
-                prompt=question
+                prompt=f"{question} Think step by step."
             )
             
             return transcript.text
         except Exception as 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:  
+            img_b64 = get_img_b64(file_path)
+
+            client = OpenAI()
+
+            completion = client.chat.completions.create(
+                messages=[{"role": "user",
+                           "content": [{"type": "text", "text": f"{question} Think step by step."},
+                                       {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}}]}],
+                model=IMAGE_MODEL
+            )
+        
+            return completion.choices[0].message.content
+        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()