Update app.py

app.py

@@ -199,34 +199,90 @@ def generate_response(
     except Exception as e:
         return [ChatMessage(role="assistant", content=f"Error: {str(e)}")]
 
+def crear_imagen_placeholder():
+    """Crea una imagen placeholder simple usando PIL"""
+    try:
+        from PIL import Image, ImageDraw, ImageFont
+        import io
+        import base64
+        
+        # Crear imagen simple
+        img = Image.new('RGB', (100, 100), color='lightgray')
+        draw = ImageDraw.Draw(img)
+        
+        # Dibujar texto simple
+        draw.text((10, 40), "No Image", fill='black')
+        
+        # Convertir a bytes
+        buffer = io.BytesIO()
+        img.save(buffer, format='PNG')
+        img_str = base64.b64encode(buffer.getvalue()).decode()
+        
+        return f"data:image/png;base64,{img_str}"
+    except:
+        # Fallback: retornar None si no se puede crear la imagen
+        return None
+
 def actualizar_personajes(pais_seleccionado):
     """Actualiza la galería de personajes según el país seleccionado"""
     global current_personajes
+    
+    if not pais_seleccionado:
+        return gr.Gallery(value=None, visible=False), "Selecciona un país para ver sus personajes"
+    
     personajes = PERSONAJES_POR_PAIS.get(pais_seleccionado, [])
     current_personajes = personajes
     
     if not personajes:
-        return [], "Selecciona un país para ver sus personajes"
+        return gr.Gallery(value=None, visible=False), "No hay personajes disponibles para este país"
     
-    # Crear lista de imágenes para la galería
-    imagenes = []
-    for p in personajes:
-        if os.path.exists(p["imagen"]):
-            imagenes.append(p["imagen"])
+    # Crear lista de imágenes válidas para la galería
+    imagenes_validas = []
+    
+    for i, personaje in enumerate(personajes):
+        imagen_path = personaje["imagen"]
+        
+        # Verificar si la imagen existe
+        if os.path.exists(imagen_path) and os.path.isfile(imagen_path):
+            try:
+                # Verificar que es una imagen válida
+                from PIL import Image
+                with Image.open(imagen_path) as img:
+                    img.verify()  # Verificar que la imagen es válida
+                imagenes_validas.append(imagen_path)
+            except Exception as e:
+                print(f"Error al verificar imagen {imagen_path}: {e}")
+                # Skip esta imagen si hay error
+                continue
         else:
-            # Imagen placeholder si no existe
-            imagenes.append("https://via.placeholder.com/100x100.png?text=No+Image")
+            print(f"Imagen no encontrada: {imagen_path}")
+            # Skip imágenes que no existen
+            continue
     
-    return imagenes, f"Personajes de {pais_seleccionado}"
+    # Si no hay imágenes válidas, mostrar mensaje
+    if not imagenes_validas:
+        return gr.Gallery(
+            value=None, 
+            visible=True,
+            label=f"⚠️ Imágenes no encontradas para {pais_seleccionado}"
+        ), f"Las imágenes para {pais_seleccionado} no están disponibles"
+    
+    # Actualizar la galería con las imágenes válidas
+    return gr.Gallery(
+        value=imagenes_validas,
+        visible=True,
+        label=f"Personajes de {pais_seleccionado}"
+    ), f"Mostrando {len(imagenes_validas)} personajes de {pais_seleccionado}"
 
 def crear_prompt_desde_personaje(evt: gr.SelectData):
     """Crea un prompt basado en el personaje seleccionado"""
     global current_personajes
     
     try:
-        if evt.index is not None and evt.index < len(current_personajes):
+        if evt.index is not None and 0 <= evt.index < len(current_personajes):
             personaje = current_personajes[evt.index]
-            return f"Crea una historia sobre {personaje['nombre']}, {personaje['descripcion']}"
+            prompt = f"Crea una historia sobre {personaje['nombre']}: {personaje['descripcion']}"
+            return prompt
         else:
             return "Crea una historia sobre un personaje mítico"
     except Exception as e:
@@ -278,14 +334,24 @@ with gr.Blocks(title="Iberotales", css=custom_css) as demo:
                 container=False
             )
             
+            # Inicializar la galería vacía
             galeria_personajes = gr.Gallery(
-                value=[],
-                label="Personajes",
-                show_label=False,
+                value=None,
+                label="Selecciona un país",
+                show_label=True,
                 elem_id="galeria",
                 columns=1,
                 rows=4,
-                height=350
+                height=350,
+                type="filepath"  # Especificar el tipo explícitamente
+            )
+            
+            # Mensaje de estado
+            status_msg = gr.Textbox(
+                value="Selecciona un país para ver sus personajes",
+                show_label=False,
+                interactive=False,
+                container=False
             )
         
         # Panel derecho - Chat
@@ -320,14 +386,14 @@ with gr.Blocks(title="Iberotales", css=custom_css) as demo:
     pais_dropdown.change(
         fn=actualizar_personajes,
         inputs=[pais_dropdown],
-        outputs=[galeria_personajes]
+        outputs=[galeria_personajes, status_msg]
     )
     
     # Cargar personajes iniciales
     demo.load(
         fn=actualizar_personajes,
         inputs=[pais_dropdown],
-        outputs=[galeria_personajes]
+        outputs=[galeria_personajes, status_msg]
     )
     
     # Crear prompt desde galería