Update app.py

app.py

@@ -1,20 +1,21 @@
 # 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
+# 1. Instal·lació automàtica de llibreries (al principi del fitxer)
 import subprocess
 import sys
 
 def install(package):
     subprocess.check_call([sys.executable, "-m", "pip", "install", package])
 
-# Instalar todas las dependencias primero
-install("deep-translator")
+# Instalar primero las dependencias pesadas
+install("torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu")
 install("transformers")
+install("deep-translator")
 install("langchain")
-install("sentencepiece")  # Necesario para algunos modelos de transformers
+install("sentencepiece")
 
-# Ahora importar el resto de módulos
+# 2. Importaciones después de las instalaciones
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
@@ -22,13 +23,17 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
-from langchain.tools import Tool  # Ahora debería funcionar
+from langchain.tools import Tool
 
-# 🔤 Eina 1: Traducció de text
+# 3. Configuración del traductor (no requiere PyTorch/TensorFlow)
 from deep_translator import GoogleTranslator
 
 def translate_text(text: str) -> str:
-    return GoogleTranslator(source='auto', target='en').translate(text)
+    """Traduce texto a inglés usando Google Translate"""
+    try:
+        return GoogleTranslator(source='auto', target='en').translate(text)
+    except Exception as e:
+        return f"Error en traducción: {str(e)}"
 
 translate_tool = Tool(
     name="TextTranslator",
@@ -36,14 +41,29 @@ translate_tool = Tool(
     description="Translate input text to English. Input can be in any language."
 )
 
-# 🧠 Eina 2: Resumidor de text
+# 4. Configuración alternativa del resumidor (sin PyTorch/TensorFlow)
 from transformers import pipeline
 
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+try:
+    # Intentamos cargar un modelo pequeño que pueda funcionar sin GPU
+    summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
+except:
+    # Fallback a una implementación más simple si falla
+    def simple_summarizer(text: str) -> str:
+        """Resumen básico tomando las primeras oraciones"""
+        sentences = text.split('.')
+        return '. '.join(sentences[:3]) + '.' if len(sentences) > 3 else text
+    
+    summarizer = simple_summarizer
 
 def summarize_text(text: str) -> str:
-    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
-    return summary[0]['summary_text']
+    try:
+        if callable(summarizer):
+            summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
+            return summary[0]['summary_text'] if isinstance(summary, list) else summary
+        return summarizer(text)
+    except Exception as e:
+        return f"Error en resumen: {str(e)}"
 
 summarize_tool = Tool(
     name="TextSummarizer",
@@ -51,34 +71,20 @@ summarize_tool = Tool(
     description="Summarize input text. Use it for long paragraphs or documents."
 )
 
-# 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
-        arg2: the second argument
-    """
-    return "What magic will you build?"
-
-# Eina 4: Consultar l'hora en una zona horària
+# 5. Herramientas adicionales
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
+    """Obtiene la hora actual en una zona horaria específica"""
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return f"Error fetching time: {str(e)}"
 
-# Final answer tool (obligatori)
+# 6. Configuración del agente
 final_answer = FinalAnswerTool()
 
-# Model LLM
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -86,20 +92,22 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# 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)
+try:
+    image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+except:
+    @tool
+    def image_generation_tool(prompt: str) -> str:
+        return "Image generation disabled in this environment"
 
-# Carrega plantilles de prompts
+# 7. Carga de plantillas y creación del agente
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Agent amb totes les eines
 agent = CodeAgent(
     model=model,
     tools=[
         translate_tool,
         summarize_tool,
-        my_custom_tool,
         get_current_time_in_timezone,
         image_generation_tool,
         final_answer
@@ -107,5 +115,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Llançar la interfície
+# 8. Iniciar la interfaz
 GradioUI(agent).launch()