Update app.py

app.py

@@ -1,6 +1,20 @@
 # Autors: [Escriu aquí el teu nom] i [nom del company si en teniu]
 # Pràctica 26/05/2025 - Agent amb eines noves: traductor i resumidor
 
+# Instal·lació automàtica de llibreries al principi
+import subprocess
+import sys
+
+def install(package):
+    subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+
+# Instalar todas las dependencias primero
+install("deep-translator")
+install("transformers")
+install("langchain")
+install("sentencepiece")  # Necesario para algunos modelos de transformers
+
+# Ahora importar el resto de módulos
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
@@ -8,19 +22,10 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
-
-# Instal·lació automàtica de llibreries
-import subprocess, sys
-
-def install(package):
-    subprocess.check_call([sys.executable, "-m", "pip", "install", package])
-
-install("deep-translator")
-install("transformers")
+from langchain.tools import Tool  # Ahora debería funcionar
 
 # 🔤 Eina 1: Traducció de text
 from deep_translator import GoogleTranslator
-from langchain.tools import Tool
 
 def translate_text(text: str) -> str:
     return GoogleTranslator(source='auto', target='en').translate(text)
@@ -34,10 +39,10 @@ translate_tool = Tool(
 # 🧠 Eina 2: Resumidor de text
 from transformers import pipeline
 
-summarizer = pipeline("summarization")
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 def summarize_text(text: str) -> str:
-    summary = summarizer(text, max_length=60, min_length=20, do_sample=False)
+    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
     return summary[0]['summary_text']
 
 summarize_tool = Tool(
@@ -46,8 +51,9 @@ summarize_tool = Tool(
     description="Summarize input text. Use it for long paragraphs or documents."
 )
 
-# Eina 3: Exemple de tool personalitzada (ja estava al template)
-def my_custom_tool(arg1:str, arg2:int)-> str:
+# Eina 3: Exemple de tool personalitzada
+@tool
+def my_custom_tool(arg1: str, arg2: int) -> str:
     """A tool that does nothing yet 
     Args:
         arg1: the first argument
@@ -55,7 +61,7 @@ def my_custom_tool(arg1:str, arg2:int)-> str:
     """
     return "What magic will you build?"
 
-# Eina 4: Consultar l’hora en una zona horària
+# Eina 4: Consultar l'hora en una zona horària
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -76,11 +82,11 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id = "mistralai/Mistral-Small-3.1-24B-Instruct-2503",
+    model_id="mistralai/Mistral-Small-3.1-24B-Instruct-2503",
     custom_role_conversions=None,
 )
 
-# Carrega eina de generació d’imatges des de Hugging Face Hub
+# Carrega eina de generació d'imatges des de Hugging Face Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 # Carrega plantilles de prompts
@@ -88,8 +94,18 @@ with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
 # Agent amb totes les eines
-agent = Cod
-
-
+agent = CodeAgent(
+    model=model,
+    tools=[
+        translate_tool,
+        summarize_tool,
+        my_custom_tool,
+        get_current_time_in_timezone,
+        image_generation_tool,
+        final_answer
+    ],
+    prompt_templates=prompt_templates
+)
 
+# Llançar la interfície
 GradioUI(agent).launch()