Update crew.py

crew.py

@@ -5,6 +5,7 @@ from crewai.tools import tool
 from crewai_tools import (
     CodeInterpreterTool,
     SerperDevTool,
+    VisionTool,
     WebsiteSearchTool,
     YoutubeVideoSearchTool
 )
@@ -29,13 +30,34 @@ CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
 def run_crew(question):
     # Tools
     
-    code_interpreter_tool = CodeInterpreterTool()
+    python_coding_tool = CodeInterpreterTool()
+    image_search_tool = VisionTool()
     web_search_tool = SerperDevTool()
     web_rag_tool = WebsiteSearchTool()
     youtube_search_tool = YoutubeVideoSearchTool()
 
     # Agents
+
+    image_search_agent = Agent(
+        role="Image Search Agent",
+        goal="Search an image for question \"{topic}\".",
+        backstory="As an expert image search assistant, you search an image for the question.",
+        allow_delegation=False,
+        max_iter=1,
+        tools=[image_search_tool],
+        verbose=True
+    )
     
+    python_coding_agent = Agent(
+        role="Python Coding Agent",
+        goal="Write and execute Python code to solve question \"{topic}\".",
+        backstory="As an expert Python coding assistant, you write and execute Python for the question.",
+        allow_delegation=False,
+        max_iter=1,
+        tools=[python_coding],
+        verbose=True,
+    )
+   
     web_search_agent = Agent(
         role="Web Search Agent",
         goal="Search the web for question \"{topic}\" and scrape the most relevant web page.",
@@ -56,44 +78,40 @@ def run_crew(question):
         verbose=True
     )
 
-    python_coding_agent = Agent(
-        role="Python Coding Agent",
-        goal="Write and execute Python code to solve question \"{topic}\".",
-        backstory="As an expert Python coding assistant, you write and execute Python for the question.",
-        allow_delegation=False,
-        max_iter=1,
-        tools=[code_interpreter_tool],
-        verbose=True,
-    )
-
     # Tasks
+
+    image_search_task = Task(
+        agent=image_search_agent,
+        description="Search an image for question \"{topic}\"",
+        expected_output="Content to help answer the image related question."
+    )
     
-    web_search = Task(
+    python_coding_task = Task(
+        agent=python_coding_agent,
+        description="Write and execute Python code to solve question \"{topic}\".",
+        expected_output="Content to help answer the Phyton coding related question."
+    )
+
+    web_search_task = Task(
         agent=web_search_agent,
         description="Search the web for question \"{topic}\" and scrape the most relevant web page.",
         expected_output="Content to help answer the web related question."
     )
-
-    youtube_search = Task(
+    
+    youtube_search_task = Task(
         agent=youtube_search_agent,
         description="Search a YouTube video for question \"{topic}\"",
         expected_output="Content to help answer the YouTube related question."
     )
 
-    coding_task = Task(
-        agent=python_coding_agent,
-        description="Write and execute Python code to solve question \"{topic}\".",
-        expected_output="Content to help answer the Phyton coding related question."
-    )
-
     # Crew
     
     crew = Crew(
-        agents=[python_coding_agent, web_search_agent, youtube_search_agent],
+        agents=[image_search_agent, python_coding_agent, web_search_agent, youtube_search_agent],
         #planning=True,
         #planning_llm=PLANNING_MODEL,
         process=Process.sequential,
-        tasks=[coding_task, web_search, youtube_search],
+        tasks=[image_search_task, python_coding_task, web_search_task, youtube_search_task],
         verbose=True
     )