Update crew.py

crew.py

@@ -41,7 +41,7 @@ def run_crew(question, file_name):
 
     image_analysis_agent = Agent(
         role="Image Analysis Agent",
-        goal="Analyze image to help answer question \"{topic}\"",
+        goal="Analyze image to help answer question \"{question}\"",
         backstory="As an expert image analysis assistant, you analyze the image to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
@@ -52,7 +52,7 @@ 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}\"",
+        goal="Write and/or execute Python code to help answer question \"{question}\"",
         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,
@@ -63,7 +63,7 @@ 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.",
+        goal="Search the web to help answer question \"{question}\", 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,
@@ -74,7 +74,7 @@ def run_crew(question, file_name):
 
     youtube_analysis_agent = Agent(
         role="YouTube Analysis Agent",
-        goal="Analyze YouTube video to help answer question \"{topic}\"",
+        goal="Analyze YouTube video to help answer question \"{question}\"",
         backstory="As an expert YouTube video analysis assistant, you analyze the video to help answer the question.",
         allow_delegation=False,
         llm=AGENT_MODEL,
@@ -87,7 +87,7 @@ 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}\"",
+             "Question: \"{question}\"",
         backstory="As an expert manager assistant, you answer the question.",
         allow_delegation=True,
         llm=MANAGER_MODEL,
@@ -99,7 +99,7 @@ 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}\"",
+        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: \"{question}\"",
         expected_output="The answer to the question."
     )
     
@@ -116,12 +116,12 @@ def run_crew(question, file_name):
     
     if file_name:
         question = f"{question} File name: data/{file_name}."
-        if file_name.endswith(".py") or file_name.endswith(".xlsx"):
+        if file_name.endswith(".py"):
             with open(f"data/{file_name}", "r") as file:
                 question = f"{question} File data:\n{file.read()}"
     
     print(f"Question: {question}")
-    initial_answer = crew.kickoff(inputs={"topic": question})
+    initial_answer = crew.kickoff(inputs={"question": question})
     print(f"Initial answer: {initial_answer}")
     final_answer = get_final_answer(question, str(initial_answer))
     print(f"Final answer: {final_answer}")