from smolagents import GradioUI, {{ class_name }}, {{ agent_dict['model']['class'] }}

{% for tool in tools.values() %}
from scripts.{{ tool.name }} import {{ tool.__class__.__name__ }}
{% endfor %}

model = {{ agent_dict['model']['class'] }}()

{% for tool in tools.values() %}
{{ tool.name }} = {{ tool.__class__.__name__ }}()
{% endfor %}

agent = {{ class_name }}(
    model=model,
    tools=[{% for tool in tools.keys() %}{{ tool }}{% if not loop.last %}, {% endif %}{% endfor %}],
    {% for attribute_name, value in agent_dict.items() if attribute_name not in ["model", "tools", "prompt_templates"] %}
    {{ attribute_name }}={{ value }},
    {% endfor %}
    prompts_path='./prompts.yaml'
)

GradioUI(agent).launch()