import json
from smolagents.agents import MultiStepAgent
from smolagents.models import HfApiModel  # <- important, c'est le vrai model object
from Gradio_UI import GradioUI

# Charge la config de l'agent
with open("agent.json") as f:
    agent_config = json.load(f)

# Instancie le modèle
model_cfg = agent_config.get("model", {})
model = HfApiModel(
    model_id=model_cfg["data"]["model_id"],
    max_tokens=model_cfg["data"].get("max_tokens", 2048),
    temperature=model_cfg["data"].get("temperature", 0.5),
)

# Initialise l'agent
agent = MultiStepAgent(
    tools=agent_config.get("tools", []),
    model=model,
    prompt_config_path="prompts.yaml",
    max_steps=agent_config.get("max_steps", 6),
)

# Lance l'interface Gradio
ui = GradioUI(agent, file_upload_folder="uploads")
ui.launch(share=True)