Update app.py

app.py

@@ -6,13 +6,21 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+import gradio as gr
 
+
+
+
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 ##############
 @tool
-def get_game_of_thrones_character_info(character_name: str) -> str:
-    """A tool that fetches information about a character from Game of Thrones, including their house.
+def get_game_of_thrones_character_info(character_name: str) -> dict:
+    """A tool that fetches information about a character from Game of Thrones, including their house and a generated image.
     Args:
         character_name: The name of the character (e.g., 'Jon Snow').
+    Returns:
+        A dictionary with 'text' (character info) and 'image' (generated image URL).
     """
     base_url = "https://anapioficeandfire.com/api/characters"
     
@@ -24,7 +32,7 @@ def get_game_of_thrones_character_info(character_name: str) -> str:
         characters = response.json()
         
         if not characters:
-            return f"No se encontró información para el personaje: {character_name}"
+            return {"text": f"No se encontró información para el personaje: {character_name}", "image": None}
         
         # Tomar el primer resultado (puede haber varios con el mismo nombre)
         character = characters[0]
@@ -48,106 +56,42 @@ def get_game_of_thrones_character_info(character_name: str) -> str:
                 house_name = house_data.get("name", "Desconocido")
                 houses.append(house_name)
         
-        # Formatear la respuesta
-        result = f"Información sobre {name}:\n"
-        result += f"- Gender: {gender}\n"
-        result += f"- Culture: {culture}\n"
-        result += f"- Born: {born}\n"
-        result += f"- Died: {died}\n"
-        result += f"- Titles: {titles}\n"
-        result += f"- Aliases: {aliases}\n"
+        # Generar una descripción para la imagen
+        image_prompt = (
+            f"Personaje de Juego de Tronos: {name}, "
+            f"Género: {gender}, Cultura: {culture}, "
+            f"Títulos: {titles}, Alias: {aliases}, "
+            f"Casa: {', '.join(houses) if houses else 'Ninguna'}"
+        )
+        
+        # Generar la imagen usando la herramienta de generación de imágenes
+        image_url = image_generation_tool(image_prompt)
+        
+        # Formatear la respuesta de texto
+        text_result = f"Información sobre {name}:\n"
+        text_result += f"- Género: {gender}\n"
+        text_result += f"- Cultura: {culture}\n"
+        text_result += f"- Nacimiento: {born}\n"
+        text_result += f"- Muerte: {died}\n"
+        text_result += f"- Títulos: {titles}\n"
+        text_result += f"- Alias: {aliases}\n"
         if houses:
-            result += f"- House(s): {', '.join(houses)}\n"
+            text_result += f"- Casa(s): {', '.join(houses)}\n"
         else:
-            result += "- Casa(s): Ninguna o desconocida\n"
+            text_result += "- Casa(s): Ninguna o desconocida\n"
         
-        return result
+        # Devolver el texto y la URL de la imagen
+        return {"text": text_result, "image": image_url}
     
     except Exception as e:
-        return f"Error al obtener información del personaje: {str(e)}"
+        return {"text": f"Error al obtener información del personaje: {str(e)}", "image": None}
 
 
 ###############
 
 
 
-@tool
-def recommend_books(preferences: str) -> str:
-    """A tool that recommends books based on a user's preferences, including cover images and purchase links for Colombia.
-    Args:
-        preferences: A string describing the user's preferences (e.g., 'Me gustan los libros de ciencia ficción con personajes fuertes').
-    """
-    base_url = "https://openlibrary.org/search.json"
-    
-    try:
-        # Convertir las preferencias en una consulta de búsqueda
-        query = preferences.replace(" ", "+")  # Formatear para la URL
-        params = {"q": query, "fields": "title,author_name,first_publish_year,subject,cover_i,key", "limit": 5}
-        
-        # Hacer la solicitud a la API
-        response = requests.get(base_url, params=params)
-        response.raise_for_status()
-        data = response.json()
-        
-        # Obtener los libros recomendados
-        books = data.get("docs", [])
-        
-        if not books:
-            return "No se encontraron libros que coincidan con tus preferencias."
-        
-        # Formatear la respuesta
-        result = "Aquí tienes algunas recomendaciones de libros:\n"
-        for i, book in enumerate(books, start=1):
-            title = book.get("title", "Título desconocido")
-            author = ", ".join(book.get("author_name", ["Autor desconocido"]))
-            year = book.get("first_publish_year", "Año desconocido")
-            subjects = ", ".join(book.get("subject", ["Género no especificado"]))[:50]  # Limitar la longitud de los géneros
-            cover_id = book.get("cover_i", "")
-            book_key = book.get("key", "")  # Clave única del libro para obtener más detalles
-            
-            # Obtener descripción del libro (si está disponible)
-            description = "Descripción no disponible."
-            if book_key:
-                book_details_url = f"https://openlibrary.org{book_key}.json"
-                book_details_response = requests.get(book_details_url)
-                if book_details_response.status_code == 200:
-                    book_details = book_details_response.json()
-                    description = book_details.get("description", "Descripción no disponible.")
-                    if isinstance(description, dict):  # Algunas descripciones son diccionarios
-                        description = description.get("value", "Descripción no disponible.")
-                    description = description[:200] + "..."  # Limitar la longitud del resumen
-            
-            # Generar enlace de portada (si está disponible)
-            cover_url = f"https://covers.openlibrary.org/b/id/{cover_id}-L.jpg" if cover_id else "Portada no disponible"
-            
-            # Generar enlaces de compra para Colombia
-            search_query = f"{title} {author}".replace(" ", "+")
-            amazon_link = f"https://www.amazon.com/s?k={search_query}"
-            mercadolibre_link = f"https://listado.mercadolibre.com.co/{search_query}"
-            
-            # Formatear el resumen útil (sin spoilers)
-            useful_summary = (
-                f"Este libro es útil porque explora temas como {subjects}. "
-                f"Es una excelente opción si te gustan los libros que {preferences.lower()}. "
-                f"Publicado en {year}, ha sido aclamado por su narrativa y personajes memorables."
-            )
-            
-            # Añadir al resultado
-            result += f"\n{i}. **{title}**\n"
-            result += f"   - Autor: {author}\n"
-            result += f"   - Año de publicación: {year}\n"
-            result += f"   - Géneros: {subjects}...\n"
-            result += f"   - Resumen útil: {useful_summary}\n"
-            if cover_url != "Portada no disponible":
-                result += f"   - Portada: {cover_url}\n"
-            result += f"   - Comprar en Amazon: {amazon_link}\n"
-            result += f"   - Comprar en MercadoLibre Colombia: {mercadolibre_link}\n"
-        
-        return result
-    
-    except Exception as e:
-        return f"Error al buscar recomendaciones de libros: {str(e)}"
-####################
+
 
 
 final_answer = FinalAnswerTool()
@@ -163,8 +107,7 @@ custom_role_conversions=None,
 )
 
 
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
@@ -182,4 +125,31 @@ agent = CodeAgent(
 )
 
 
-GradioUI(agent).launch()
+
+##############
+
+
+class EnhancedGradioUI(GradioUI):
+    def launch(self):
+        def run_agent(query):
+            response = self.agent.run(query)
+            if isinstance(response, dict) and "text" in response and "image" in response:
+                return response["text"], response["image"]
+            else:
+                return response, None
+        
+        # Crear la interfaz de Gradio
+        iface = gr.Interface(
+            fn=run_agent,
+            inputs="text",
+            outputs=["text", "image"],
+            title="Game of Thrones Character Info",
+            description="Obtén información sobre personajes de Juego de Tronos, incluyendo una imagen generada."
+        )
+        iface.launch()
+
+# Lanzar la interfaz mejorada
+EnhancedGradioUI(agent).launch()
+
+
+#GradioUI(agent).launch()