Update crew.py

crew.py

@@ -41,8 +41,8 @@ def run_crew(question, file_name):
 
     image_analysis_agent = Agent(
         role="Image Analysis Agent",
-        goal="Analyze image to help answer question \"{topic}\". ",
-        backstory="As an expert image analysis assistant, you analyze the image to help answer the question. ",
+        goal="Analyze image to help answer question \"{topic}\"",
+        backstory="As an expert image analysis assistant, you analyze the image to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
         max_iter=3,
@@ -52,8 +52,8 @@ def run_crew(question, file_name):
     
     python_coding_agent = Agent(
         role="Python Coding Agent",
-        goal="Write and/or execute Python code to help answer question \"{topic}\". ",
-        backstory="As an expert Python coding assistant, you write and/or execute Python code to help answer the question. ",
+        goal="Write and/or execute Python code to help answer question \"{topic}\"",
+        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=5,
@@ -63,8 +63,8 @@ def run_crew(question, file_name):
    
     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. ",
+        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,
@@ -74,8 +74,8 @@ def run_crew(question, file_name):
 
     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. ",
+        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,
@@ -87,8 +87,8 @@ def run_crew(question, file_name):
         role="Manager Agent",
         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. ",
+             "Question: \"{topic}\"",
+        backstory="As an expert manager assistant, you answer the question.",
         allow_delegation=True,
         llm=MANAGER_MODEL,
         max_iter=5,
@@ -99,8 +99,8 @@ 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, Web Search Agent, or YouTube Analysis Agent for help. Question: \"{topic}\" ",
-        expected_output="The answer to the question. "
+        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
@@ -115,8 +115,8 @@ def run_crew(question, file_name):
     # Processing
     
     if file_name:
-        question = f"{question} File name: data/{file_name}. "
-        if file_name.endswith(".py"):
+        question = f"{question} File name: data/{file_name}."
+        if file_name.endswith(".py") or file_name.endswith(".xlsx"):
             with open(f"data/{file_name}", "r") as file:
                 question = f"{question} File data:\n{file.read()}"