Cambio de Agent por CodeAgent es app.py

app.py

@@ -1,35 +1,14 @@
-from smolagents import Agent
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
 import gradio as gr
-from transformers import pipeline
 import emoji
 import random
 
-# Definimos el prompt base para el agente
-AGENT_PROMPT = """
-Eres un experto convertidor de texto a memes. Tu trabajo es tomar el texto del usuario y convertirlo 
-en un estilo de meme divertido y apropiado. Tienes los siguientes estilos disponibles:
-
-- Sarcástico: Alternando mayúsculas y minúsculas
-- Vibe: Añadiendo ✨ entre palabras
-- Dramático: Usando emojis dramáticos
-- UwU: Estilo kawaii
-- Gamer: Usando lenguaje de gaming
-- Deep: Estilo filosófico
-- Cool: Estilo relajado con emojis cool
-
-Debes:
-1. Analizar el texto del usuario
-2. Elegir el mejor estilo si no se especifica uno
-3. Convertir el texto al estilo seleccionado
-4. Añadir emojis apropiados
-5. Explicar brevemente por qué elegiste ese estilo
-
-Mantén un tono divertido y amigable en todo momento.
-"""
-
-class MemeAgent(Agent):
+class MemeAgent:
     def __init__(self):
-        super().__init__(prompt=AGENT_PROMPT)
+        self.agent = CodeAgent(
+            tools=[DuckDuckGoSearchTool()],
+            llm=HfApiModel('mistralai/Mistral-7B-Instruct-v0.1')
+        )
         self.styles = {
             "sarcastic": self._sarcastic_style,
             "vibe": self._vibe_style,
@@ -39,7 +18,7 @@ class MemeAgent(Agent):
             "deep": self._deep_style,
             "cool": self._cool_style
         }
-        
+
     def _sarcastic_style(self, text):
         return "".join(c.upper() if i % 2 else c.lower() for i, c in enumerate(text)) + " 🥴"
     
@@ -62,41 +41,32 @@ class MemeAgent(Agent):
     def _cool_style(self, text):
         return f"😎 {text} 🔥"
 
-    async def process_text(self, text, style=None):
-        # Crear el contexto para el agente
-        context = f"Texto del usuario: '{text}'\n"
-        if style:
-            context += f"Estilo solicitado: {style}\n"
-        else:
-            context += "No se especificó estilo, elige el más apropiado.\n"
-
-        # Obtener la respuesta del agente
-        response = await self.run(context)
+    def process_text(self, text, style=None):
+        if not text.strip():
+            return "Por favor, ingresa algún texto", "No hay texto para procesar"
         
-        # Procesar el texto según el estilo
-        if style and style in self.styles:
-            converted_text = self.styles[style](text)
+        # Usar el agente para analizar el texto y sugerir un estilo si no se proporciona uno
+        if not style:
+            prompt = f"""
+            Analiza este texto: '{text}'
+            Selecciona el mejor estilo de meme entre estas opciones: {list(self.styles.keys())}
+            Solo responde con el nombre del estilo, nada más.
+            """
+            style = self.agent.run(prompt).strip().lower()
+            if style not in self.styles:
+                style = random.choice(list(self.styles.keys()))
+            explanation = f"El agente eligió el estilo '{style}' basado en el análisis del texto."
         else:
-            # Si no hay estilo especificado, usar el que el agente sugiera
-            for style_name in self.styles:
-                if style_name.lower() in response.lower():
-                    converted_text = self.styles[style_name](text)
-                    break
-            else:
-                converted_text = self._vibe_style(text)  # Estilo por defecto
+            explanation = f"Usando el estilo '{style}' como solicitaste."
 
-        return {
-            "converted_text": converted_text,
-            "explanation": response
-        }
+        converted_text = self.styles[style](text)
+        return converted_text, explanation
 
-# Interfaz de Gradio
 def create_interface():
     agent = MemeAgent()
     
-    async def process_input(text, style=None):
-        result = await agent.process_text(text, style)
-        return result["converted_text"], result["explanation"]
+    def process_input(text, style=None):
+        return agent.process_text(text, style)
     
     iface = gr.Interface(
         fn=process_input,
@@ -113,9 +83,9 @@ def create_interface():
         ],
         outputs=[
             gr.Textbox(label="Texto convertido"),
-            gr.Textbox(label="Explicación del agente")
+            gr.Textbox(label="Explicación")
         ],
-        title="🎭 Agente Convertidor de Memes",
+        title="🎭 Convertidor de Memes con IA",
         description="Convierte texto normal en diferentes estilos de memes usando IA",
         examples=[
             ["la vida es bella", "dramatic"],
@@ -128,7 +98,6 @@ def create_interface():
     )
     return iface
 
-
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch()