Update app.py

app.py

@@ -1,37 +1,28 @@
-#!/usr/bin/env python
-# coding: utf-8
-
-import os
+import json
 from smolagents.agents import MultiStepAgent
-from smolagents.models import HfApiModel
-from smolagents.utils import load_json, load_yaml
+from smolagents.models import HfApiModel  # <- important, c'est le vrai model object
 from Gradio_UI import GradioUI
 
-# --- Config paths ---
-AGENT_JSON_PATH = "agent.json"
-PROMPTS_YAML_PATH = "prompts.yaml"
-UPLOAD_FOLDER = "uploaded_files"
-
-# --- Load agent configuration ---
-agent_config = load_json(AGENT_JSON_PATH)
-prompts_config = load_yaml(PROMPTS_YAML_PATH)
+# Charge la config de l'agent
+with open("agent.json") as f:
+    agent_config = json.load(f)
 
-# --- Initialize the model ---
-model_cfg = agent_config["model"]["data"]
+# Instancie le modèle
+model_cfg = agent_config.get("model", {})
 model = HfApiModel(
-    model_id=model_cfg.get("model_id", "bigcode/starcoder"),
-    max_tokens=model_cfg.get("max_tokens", 1024),
-    temperature=model_cfg.get("temperature", 0.5),
+    model_id=model_cfg["data"]["model_id"],
+    max_tokens=model_cfg["data"].get("max_tokens", 2048),
+    temperature=model_cfg["data"].get("temperature", 0.5),
 )
 
-# --- Initialize the agent ---
+# Initialise l'agent
 agent = MultiStepAgent(
     tools=agent_config.get("tools", []),
     model=model,
-    prompt_templates=prompts_config,
+    prompt_config_path="prompts.yaml",
     max_steps=agent_config.get("max_steps", 6),
 )
 
-# --- Launch Gradio interface ---
-ui = GradioUI(agent, file_upload_folder=UPLOAD_FOLDER)
-ui.launch(debug=True, share=True)
+# Lance l'interface Gradio
+ui = GradioUI(agent, file_upload_folder="uploads")
+ui.launch(share=True)