Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import gradio as gr
-from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel, FinalAnswerTool 
+from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel, FinalAnswerTool, MultiStepAgent 
 import requests
 import inspect
 import pandas as pd
@@ -23,9 +23,20 @@ class BasicAgent:
             api_base="https://api.openai.com/v1",
             api_key=os.environ["OPENAI_API_KEY"],
         )
-        self.my_agent = CodeAgent(
+        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]'"
+
+* 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(
             tools=[search_tool, google_search_tool, wiki_search_tool, 
                    visit_web_tool, FinalAnswerTool()], 
+            instructions=instruction,
             model=model,
             add_base_tools=True,  # Add any additional base tools
             planning_interval=3   # Enable planning every 3 steps