Update app.py

app.py

@@ -1,28 +1,20 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-import datetime
 import requests
-import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
 import gradio as gr
-from PIL import Image
-import tempfile
-import os
-import base64
-from io import BytesIO
 
 # 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) -> dict:
-    """A tool that fetches information about a character from Game of Thrones, including their house and a generated image.
+def get_game_of_thrones_character_info(character_name: str) -> str:
+    """Una herramienta que obtiene información detallada sobre un personaje de Juego de Tronos.
     Args:
-        character_name: The name of the character (e.g., 'Jon Snow').
+        character_name: El nombre del personaje (por ejemplo, 'Jon Snow').
     Returns:
-        A dictionary containing the character info and the base64-encoded image.
+        Un string con la información detallada del personaje.
     """
     base_url = "https://anapioficeandfire.com/api/characters"
     
@@ -34,14 +26,14 @@ def get_game_of_thrones_character_info(character_name: str) -> dict:
         characters = response.json()
         
         if not characters:
-            return {"error": f"No se encontró información para el personaje: {character_name}"}
+            return f"No se encontró información para el personaje: {character_name}"
         
         # Tomar el primer resultado (puede haber varios con el mismo nombre)
         character = characters[0]
         
         # Extraer información relevante del personaje
         name = character.get("name", "Desconocido")
-        gender = character.get("gender", "Desconocido")
+        gender = "Masculino" if character.get("gender", "").lower() == "male" else "Femenino"
         culture = character.get("culture", "Desconocido")
         born = character.get("born", "Desconocido")
         died = character.get("died", "Aún vivo")
@@ -58,39 +50,37 @@ def get_game_of_thrones_character_info(character_name: str) -> dict:
                 house_name = house_data.get("name", "Desconocido")
                 houses.append(house_name)
         
-        # 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'}"
-        )
+        # Obtener información de los libros en los que aparece
+        books = []
+        if "books" in character:
+            for book_url in character["books"]:
+                book_response = requests.get(book_url)
+                book_response.raise_for_status()
+                book_data = book_response.json()
+                book_name = book_data.get("name", "Desconocido")
+                books.append(book_name)
         
-        # Generar la imagen usando la herramienta de generación de imágenes
-        image = image_generation_tool(image_prompt)
-        
-        # Convertir la imagen a base64
-        buffered = BytesIO()
-        image.save(buffered, format="JPEG")
-        img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
-        
-        # 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"
+        # Formatear la respuesta en español
+        result = f"**Información sobre {name}**:\n\n"
+        result += f"- **Género**: {gender}\n"
+        result += f"- **Cultura**: {culture}\n"
+        result += f"- **Nacimiento**: {born}\n"
+        result += f"- **Muerte**: {died}\n"
+        result += f"- **Títulos**: {titles}\n"
+        result += f"- **Alias**: {aliases}\n"
         if houses:
-            result += f"- House(s): {', '.join(houses)}\n"
+            result += f"- **Casa(s)**: {', '.join(houses)}\n"
+        else:
+            result += "- **Casa(s)**: Ninguna o desconocida\n"
+        if books:
+            result += f"- **Libros en los que aparece**: {', '.join(books)}\n"
         else:
-            result += "- Casa(s): Ninguna o desconocida\n"
+            result += "- **Libros en los que aparece**: Ninguno o desconocido\n"
         
-        return {"info": result, "image": f"data:image/jpeg;base64,{img_base64}"}
+        return result
     
     except Exception as e:
-        return {"error": f"Error al obtener información del personaje: {str(e)}"}
+        return f"Error al obtener información del personaje: {str(e)}"
 
 ###############
 
@@ -124,16 +114,17 @@ agent = CodeAgent(
 ##############
 def display_character_info(character_name):
     result = get_game_of_thrones_character_info(character_name)
-    if "error" in result:
-        return result["error"], None
-    return result["info"], result["image"]
+    return result
 
+# Crear una interfaz de Gradio
 interface = gr.Interface(
     fn=display_character_info,
-    inputs="text",
-    outputs=["text", "image"],
+    inputs=gr.Textbox(label="Nombre del personaje", placeholder="Ejemplo: Jon Snow"),
+    outputs=gr.Textbox(label="Información del personaje"),
     title="Información de Personajes de Juego de Tronos",
-    description="Ingresa el nombre de un personaje de Juego de Tronos para obtener información y una imagen generada."
+    description="Ingresa el nombre de un personaje de Juego de Tronos para obtener información detallada.",
+    examples=["Jon Snow", "Daenerys Targaryen", "Tyrion Lannister"],
+    theme="soft"
 )
 
 interface.launch()