Update app.py

app.py

@@ -1,36 +1,37 @@
-import yaml
-from smolagents import CodeAgent, tool
+#!/usr/bin/env python
+# coding: utf-8
 
-# Définition d’un outil simple
-@tool
-def visit_webpage(url: str) -> str:
-    """
-    Visit a webpage and return its content.
+import os
+from smolagents.agents import MultiStepAgent
+from smolagents.models import HfApiModel
+from smolagents.utils import load_json, load_yaml
+from Gradio_UI import GradioUI
 
-    Args:
-        url (str): The URL to visit.
+# --- Config paths ---
+AGENT_JSON_PATH = "agent.json"
+PROMPTS_YAML_PATH = "prompts.yaml"
+UPLOAD_FOLDER = "uploaded_files"
 
-    Returns:
-        str: The full HTML content of the page.
-    """
-    import requests
-    response = requests.get(url)
-    response.raise_for_status()
-    return response.text
+# --- Load agent configuration ---
+agent_config = load_json(AGENT_JSON_PATH)
+prompts_config = load_yaml(PROMPTS_YAML_PATH)
 
-# Charger les prompts depuis prompts.yaml
-with open("prompts.yaml", "r", encoding="utf-8") as f:
-    prompts = yaml.safe_load(f)
+# --- Initialize the model ---
+model_cfg = agent_config["model"]["data"]
+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),
+)
 
-# Créer l’agent avec le system_prompt depuis le yaml
-agent = CodeAgent(
-    tools=[visit_webpage],
-    system_prompt=prompts["system_prompt"]
+# --- Initialize the agent ---
+agent = MultiStepAgent(
+    tools=agent_config.get("tools", []),
+    model=model,
+    prompt_templates=prompts_config,
+    max_steps=agent_config.get("max_steps", 6),
 )
 
-if __name__ == "__main__":
-    print("===== Application Startup =====")
-    task = "Visit https://example.com and summarize the page."
-    result = agent.run(task)
-    print("===== Final Answer =====")
-    print(result)
+# --- Launch Gradio interface ---
+ui = GradioUI(agent, file_upload_folder=UPLOAD_FOLDER)
+ui.launch(debug=True, share=True)