Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 import yaml
 from tools.final_answer import FinalAnswerTool
 import gradio as gr
+from difflib import get_close_matches
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
@@ -26,7 +27,19 @@ 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}"
+            # Si no se encuentra el personaje, buscar nombres similares
+            all_characters_response = requests.get(base_url, params={"pageSize": 1000})
+            all_characters_response.raise_for_status()
+            all_characters = all_characters_response.json()
+            all_names = [char.get("name", "") for char in all_characters if char.get("name")]
+            
+            # Buscar nombres similares
+            close_matches = get_close_matches(character_name, all_names, n=2, cutoff=0.5)
+            if close_matches:
+                suggestions = "\n- " + "\n- ".join(close_matches)
+                return f"No se encontró información para el personaje: {character_name}.\n\n¿Quizás quisiste decir alguno de estos?\n{suggestions}"
+            else:
+                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]
@@ -61,7 +74,7 @@ def get_game_of_thrones_character_info(character_name: str) -> str:
                 books.append(book_name)
         
         # Formatear la respuesta en español
-        result = f"**Información sobre {name}**:\n\n"
+        result = f"**Información sobre {name}**\n\n"
         result += f"- **Género**: {gender}\n"
         result += f"- **Cultura**: {culture}\n"
         result += f"- **Nacimiento**: {born}\n"
@@ -120,10 +133,10 @@ def display_character_info(character_name):
 interface = gr.Interface(
     fn=display_character_info,
     inputs=gr.Textbox(label="Nombre del personaje", placeholder="Ejemplo: Jon Snow"),
-    outputs=gr.Textbox(label="Información del personaje"),
+    outputs=gr.Textbox(label="Información del personaje", markdown=True),  # Habilitar Markdown
     title="Información de Personajes de Juego de Tronos",
     description="Ingresa el nombre de un personaje de Juego de Tronos para obtener información detallada.",
-    examples=["Jon Snow", "Daenerys Targaryen", "Tyrion Lannister"],
+    examples=["Jon Snow", "Daenerys Targaryen", "Tyrion Lannister", "Viserys Targaryen"],
     theme="soft"
 )