Update app.py

app.py

@@ -1,10 +1,10 @@
 import os
 import gradio as gr
-from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel, FinalAnswerTool, MultiStepAgent 
+from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel, FinalAnswerTool 
 import requests
 import inspect
 import pandas as pd
-from tools import search_tool, google_search_tool, wiki_search_tool, visit_web_tool
+from tools import search_tool, google_search_tool, wiki_search_tool, visit_web_tool, file_download_tool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -26,16 +26,17 @@ class BasicAgent:
         instruction = """
 You are an expert in question-and-answer tasks.
 You will receive questions in the following format:
-"[Question content] | Task Id: [task_id] - file: '[file_name]'"
+"[Question content] | Task ID: [task_id] - file: '[file_name]'"
 
 * Use the available tools to answer each question.
 * If a question requires accessing an additional file, use the download tool with the provided task ID and file name.
 * For questions involving mathematics or counting, write code to calculate the answer.
 * Return your response in the exact format specified by the question.
         """
-        self.my_agent = MultiStepAgent(
+        self.my_agent = CodeAgent(
             tools=[search_tool, google_search_tool, wiki_search_tool, 
-                   visit_web_tool, FinalAnswerTool()], 
+                   visit_web_tool, file_download_tool,
+                   FinalAnswerTool()], 
             instructions=instruction,
             model=model,
             add_base_tools=True,  # Add any additional base tools
@@ -46,7 +47,8 @@ You will receive questions in the following format:
         print(f"Task {task_id} - Agent received question (first 50 chars): {question[:50]}...")
         # fixed_answer = "This is a default answer."
         # print(f"Agent returning fixed answer: {fixed_answer}")
-        response = self.my_agent.run(question)
+        submitted_question = f"{question} | Task ID: {task_id} - file: '{file_name}'"
+        response = self.my_agent.run(submitted_question)
         return response
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):