Update app.py

app.py

@@ -1,7 +1,9 @@
 import os
 import re
+import yaml
 import openai
 import pathlib
+import inspect
 import requests
 import tempfile
 import gradio as gr
@@ -11,10 +13,17 @@ from pathlib import Path
 from tabulate import tabulate
 from typing import Union, Optional
 from smolagents.tools import PipelineTool, Tool
-from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, PythonInterpreterTool
+from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, PythonInterpreterTool, VisitWebpageTool
 
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+SYSTEM_PROMPT = """
+You are a helpful assistant tasked with answering questions using a set of tools. 
+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 comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer. 
+"""
 
 #-------------------------------------------------
 #                      TOOLS
@@ -132,17 +141,45 @@ class ExcelToTextTool(Tool):
 #-------------------------------------------------
 class BasicAgent:
     def __init__(self):
+
+        #Define model
+        self.model = OpenAIServerModel(
+            model_id="gpt-4o-mini",
+            api_base="https://api.openai.com/v1",
+            api_key=os.getenv("OPENAI_API_KEY")
+        ),
+
+        #Define prompts
+        self.prompt_templates = {}
+        with open("prompt_templates.yaml", 'r') as stream:
+            self.prompt_templates = yaml.safe_load(stream)
+
+        #Define agent
         self.agent = CodeAgent(
-            model=OpenAIServerModel(model_id="gpt-4o"),
+            model = self.model,
             tools=[
-                DuckDuckGoSearchTool(), 
+                DuckDuckGoSearchTool(),
+                VisitWebpageTool(),
+                PythonInterpreterTool(
+                    authorized_imports=[
+                        'pandas',
+                        'numpy',
+                        're'
+                    ]
+                ),
                 WikipediaSearchTool(), 
-                PythonInterpreterTool(),
                 #SpeechToTextTool(), 
                 #ExcelToTextTool()
             ],
-            add_base_tools=True,
-            additional_authorized_imports=['pandas','numpy','csv','subprocess']
+            max_steps = 6,
+            verbosity_level = 1,
+            grammar = None,
+            planning_interval = None,
+            name = None,
+            description = None,
+            #add_base_tools = True,
+            #prompt_templates=self.prompt_templates,
+            additional_authorized_imports=['pandas','numpy','csv','subprocess', 're']
         )
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str: