Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, WikipediaSearchTool, MultiStepAgent
+from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, WikipediaSearchTool
 from smolagents.agents import PromptTemplates
 
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
@@ -14,13 +14,52 @@ model = OpenAIServerModel(
     api_base="https://generativelanguage.googleapis.com/v1beta/openai/"
 )
 
-class CustomMultiStepAgent(MultiStepAgent):
-    def initialize_system_prompt(self):
-        # Implement the method as required
-        pass
+# prompt = PromptTemplates(
+#     system_prompt="""
+# You are a helpful assistant tasked with answering questions using a set of tools.
+# {{tool_descriptions}}
 
-prompt = PromptTemplates(
-    system_prompt="""
+# Now, I will ask you a question. Report your thoughts, and finish your answer with the following template:
+
+# FINAL ANSWER: [YOUR FINAL ANSWER].
+
+# YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+# - If you are asked for a number, don't use commas or units like $ or %, unless explicitly required.
+# - If you are asked for a string, do not include articles or abbreviations, and write digits in full unless otherwise specified.
+# - If asked for a comma-separated list, apply the above rules depending on whether the list elements are numbers or strings.
+
+# Your answer should **only** start with `FINAL ANSWER: ` followed by the answer.
+# """,
+
+#     planning="""
+# You have received the following task:
+# {{task_description}}
+
+# Think step-by-step about what tools or steps are needed to solve it. Plan clearly before proceeding.
+# """,
+
+#     managed_agent="""
+# You are executing the following step:
+# {{step_description}}
+
+# Use available tools wisely. Think before acting, and ensure your actions follow the original instruction format.
+
+# Remember to ultimately return the answer in the format:
+# FINAL ANSWER: [YOUR FINAL ANSWER]
+# """,
+
+#     final_answer="""
+# You have completed all steps.
+
+# Now, return your final response using the following format:
+# FINAL ANSWER: [YOUR FINAL ANSWER]
+
+# YOUR FINAL ANSWER should be a number, a few words, or a comma-separated list, following the formatting rules provided earlier.
+# """
+# )
+
+prompt = {
+    'system_prompt':"""
 You are a helpful assistant tasked with answering questions using a set of tools.
 {{tool_descriptions}}
 
@@ -36,14 +75,14 @@ YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma sepa
 Your answer should **only** start with `FINAL ANSWER: ` followed by the answer.
 """,
 
-    planning="""
+    'planning':"""
 You have received the following task:
 {{task_description}}
 
 Think step-by-step about what tools or steps are needed to solve it. Plan clearly before proceeding.
 """,
 
-    managed_agent="""
+    'managed_agent':"""
 You are executing the following step:
 {{step_description}}
 
@@ -53,7 +92,7 @@ Remember to ultimately return the answer in the format:
 FINAL ANSWER: [YOUR FINAL ANSWER]
 """,
 
-    final_answer="""
+    'final_answer':"""
 You have completed all steps.
 
 Now, return your final response using the following format:
@@ -61,7 +100,8 @@ FINAL ANSWER: [YOUR FINAL ANSWER]
 
 YOUR FINAL ANSWER should be a number, a few words, or a comma-separated list, following the formatting rules provided earlier.
 """
-)
+}
+
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -71,7 +111,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        self.agent = CustomMultiStepAgent(
+        self.agent = CodeAgent(
             tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
             model = model,
             add_base_tools=True,