Update crew.py

crew.py

@@ -99,6 +99,35 @@ 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_2(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("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.