Update app.py

app.py

@@ -9,16 +9,20 @@ 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) -> str:
+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 containing the character info and the base64-encoded image.
     """
     base_url = "https://anapioficeandfire.com/api/characters"
     
@@ -30,7 +34,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 {"error": 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]
@@ -65,10 +69,10 @@ def get_game_of_thrones_character_info(character_name: str) -> str:
         # Generar la imagen usando la herramienta de generación de imágenes
         image = image_generation_tool(image_prompt)
         
-        # Guardar la imagen en un archivo temporal
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_file:
-            image.save(temp_file, format="JPEG")
-            temp_file_path = temp_file.name
+        # 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"
@@ -82,12 +86,11 @@ def get_game_of_thrones_character_info(character_name: str) -> str:
             result += f"- House(s): {', '.join(houses)}\n"
         else:
             result += "- Casa(s): Ninguna o desconocida\n"
-        result += f"- Imagen generada: {temp_file_path}\n"
         
-        return result
+        return {"info": result, "image": f"data:image/jpeg;base64,{img_base64}"}
     
     except Exception as e:
-        return f"Error al obtener información del personaje: {str(e)}"
+        return {"error": f"Error al obtener información del personaje: {str(e)}"}
 
 ###############
 
@@ -119,4 +122,18 @@ agent = CodeAgent(
 )
 
 ##############
-GradioUI(agent).launch()
+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"]
+
+interface = gr.Interface(
+    fn=display_character_info,
+    inputs="text",
+    outputs=["text", "image"],
+    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."
+)
+
+interface.launch()