app.py

#from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
from smolagents import CodeAgent,HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
from tools.visit_webpage import VisitWebpageTool
from tools.web_search import DuckDuckGoSearchTool

from Gradio_UI import GradioUI

# Tool to suggest a menu based on the occasion
@tool
def suggest_menu(occasion: str) -> str:
    """
    Suggests a menu based on the occasion.  Elaborate your suggestion, if possible giving serving time.
    Args:
        occasion: The type of occasion for the party. Valid values are casual, formal and superhero
    Output: 
        Your sugestion for the menu, that you will use to give detail about each dish.        
    """
    if occasion == "casual":
        return "Pizza, snacks, and drinks. Please, elaborate this menu."
    elif occasion == "formal":
        return "3-course dinner with wine and dessert. Please, elaborate this menu, if possible giving serving time."
    elif occasion == "superhero":
        return "Buffet with high-energy and healthy food. Please, elaborate this menu."
    else:
        return "Custom menu for the butler."

final_answer = FinalAnswerTool()
visit_webpage = VisitWebpageTool()
web_search = DuckDuckGoSearchTool()

model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)

agent = CodeAgent(
    model=model,
    tools=[suggest_menu,visit_webpage,web_search,final_answer], ## add your tools here (don't remove final answer)
    max_steps=10,
    verbosity_level=2,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)


GradioUI(agent).launch()