import gradio as gr
import random
from smolagents import GradioUI, CodeAgent, HfApiModel


from retriever import load_guest_dataset


guest_info_tool = load_guest_dataset()

alfred = CodeAgent(
        tools=[guest_info_tool],
        model=model,
        add_base_tools=True,
        planning_interval=3   # Enable planning every 3 steps
)
 
if __name__ == "__main__":
    GradioUI(alfred).launch()