Update crew.py

crew.py

@@ -35,7 +35,7 @@ def run_crew(question, file_name):
     python_coding_tool = CodeInterpreterTool()
     web_search_tool = SerperDevTool()
     web_rag_tool = WebsiteSearchTool()
-    video_analysis_tool = YoutubeVideoSearchTool()
+    youtube_analysis_tool = YoutubeVideoSearchTool()
 
     # Agents
 
@@ -56,37 +56,37 @@ def run_crew(question, file_name):
         backstory="As an expert Python coding assistant, you write and/or execute Python code to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
-        max_iter=3,
+        max_iter=5,
         tools=[python_coding_tool],
         verbose=False
     )
    
-    video_analysis_agent = Agent(
-        role="Video Analysis Agent",
-        goal="Analyze video to help answer question \"{topic}\". ",
-        backstory="As an expert video analysis assistant, you analyze the video to help answer the question. ",
+    web_search_agent = Agent(
+        role="Web Search Agent",
+        goal="Search the web to help answer question \"{topic}\", then scrape the most relevant web page. ",
+        backstory="As an expert web search assistant, you search the web to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=3,
-        tools=[video_analysis_tool],
+        tools=[web_search_tool, web_rag_tool],
         verbose=False
     )
 
-    web_search_agent = Agent(
-        role="Web Search Agent",
-        goal="Search the web to help answer question \"{topic}\", then scrape the most relevant web page. ",
-        backstory="As an expert web search assistant, you search the web to help answer the question. ",
+    youtube_analysis_agent = Agent(
+        role="YouTube Analysis Agent",
+        goal="Analyze YouTube video to help answer question \"{topic}\". ",
+        backstory="As an expert YouTube video analysis assistant, you analyze the video to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=3,
-        tools=[web_search_tool, web_rag_tool],
+        tools=[youtube_analysis_tool],
         verbose=False
     )
 
     manager_agent = Agent(
         role="Manager Agent",
-        goal="Try to answer the following question. If needed, delegate to **one** of your coworkers, image_analysis_agent, python_coding_agent, video_analysis_agent, or web_search_agent for help. "
-             "If there is no good coworker, delegate to the python_coding_agent to implement a tool for the task. "
+        goal="Try to answer the following question. If needed, delegate to **one** of your coworkers, Image Analysis Agent, Python Coding Agent, Web Search Agent, or YouTube Analysis Agent for help. "
+             "If there is no good coworker, delegate to the Python Coding Agent to implement a tool for the task. "
              "Question: \"{topic}\" ",
         backstory="As an expert manager assistant, you answer the question. ",
         allow_delegation=True,
@@ -99,14 +99,14 @@ def run_crew(question, file_name):
   
     manager_task = Task(
         agent=manager_agent,
-        description="Try to answer the following question. If needed, delegate to **one** of your coworkers, Image Analysis Agent, Python Coding Agent, Video Analysis Agent, or Web Search Agent for help. Question: \"{topic}\" ",
+        description="Try to answer the following question. If needed, delegate to **one** of your coworkers, Image Analysis Agent, Python Coding Agent, Web Search Agent, or YouTube Analysis Agent for help. Question: \"{topic}\" ",
         expected_output="The answer to the question. "
     )
     
     # Crew
     
     crew = Crew(
-        agents=[image_analysis_agent, python_coding_agent, video_analysis_agent, web_search_agent],
+        agents=[image_analysis_agent, python_coding_agent, web_search_agent, youtube_analysis_agent],
         manager_agent=manager_agent,
         tasks=[manager_task],
         verbose=True
@@ -114,16 +114,11 @@ def run_crew(question, file_name):
 
     print(crew.agents)
 
-    ###
-    #if file_name:
-    #    question = f"{question} File name: data/{file_name}."
-    ###
-
     if file_name.endswith(".py"):
         with open(f"data/{file_name}", "r") as file:
-            file_content = file.read()
+            file_data = file.read()
 
-        question = f"{question} File content: {file_content}"
+        question = f"{question} File data:\n{file_data}"
     else:
         question = f"{question} File name: data/{file_name}."