Update app.py

app.py

@@ -55,7 +55,7 @@ PERSONAJES_POR_PAIS = {
         {"nombre": "El Ekeko", "imagen": "images/bo2.jpg", "descripcion": "Dios aymara de la abundancia y la fortuna con jorobas"},
         {"nombre": "El Jichi", "imagen": "images/bo3.webp", "descripcion": "Serpiente protectora de ríos y lagunas en la cultura andina"}
     ],
-    "🇧🇷 Brasil": [
+        "🇧🇷 Brasil": [
         {"nombre": "Curupira", "imagen": "images/br1.jpeg", "descripcion": "Protector del bosque amazónico con pies al revés"},
         {"nombre": "Saci-Pererê", "imagen": "images/br2.jpg", "descripcion": "Duende travieso de una pierna que fuma pipa"},
         {"nombre": "Yebá Bëló", "imagen": "images/br3.jpg", "descripcion": "Abuela del mundo en mitología desana, creadora del universo"}
@@ -204,35 +204,8 @@ def load_model():
 
 model_loaded = load_model()
 
-def format_chat_history(messages: list, exclude_last_user: bool = True) -> list:
-    """Formatea el historial de chat para el modelo"""
-    formatted_history = []
-    messages_to_process = messages[:]
-    if exclude_last_user and messages_to_process and messages_to_process[-1].get("role") == "user":
-        messages_to_process = messages_to_process[:-1]
-    
-    for message in messages_to_process:
-        current_role = message.get("role")
-        current_content = message.get("content", "").strip()
-        
-        if current_role == "assistant" and message.get("metadata"):
-            continue
-        if not current_content:
-            continue
-        
-        if formatted_history and formatted_history[-1]["role"] == current_role:
-            formatted_history[-1]["content"] += "\n\n" + current_content
-        else:
-            formatted_history.append({
-                "role": current_role,
-                "content": current_content
-            })
-    
-    return formatted_history
-
 def stream_iberotales_response(
-    user_message: str, 
-    messages: list,
+    user_message: str,
     system_message: str = DEFAULT_SYSTEM_MESSAGE,
     max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
     temperature: float = 0.7,
@@ -240,29 +213,20 @@ def stream_iberotales_response(
     top_k: int = 50,
     repetition_penalty: float = 1.2,
 ) -> Iterator[list]:
-    """Genera respuesta con streaming"""
+    """Genera respuesta con streaming - sin historial"""
     global model, tokenizer
 
     if model is None or tokenizer is None:
-        messages.append(ChatMessage(role="assistant", content="Error: Modelo no disponible."))
-        yield messages
+        yield [ChatMessage(role="assistant", content="Error: Modelo no disponible.")]
         return
 
     try:
-        chat_history = format_chat_history(messages, exclude_last_user=True)
+        # Crear conversación simple sin historial
         conversation = []
         if system_message.strip():
             conversation.append({"role": "system", "content": system_message.strip()})
-        conversation.extend(chat_history)
         conversation.append({"role": "user", "content": user_message})
 
-        # Validar alternancia
-        for i in range(1, len(conversation)):
-            if conversation[i]["role"] == conversation[i-1]["role"] and conversation[i-1]["role"] != "system":
-                messages.append(ChatMessage(role="assistant", content="Error: Reinicia la conversación."))
-                yield messages
-                return
-
         prompt = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
         response = model(
             prompt,
@@ -275,6 +239,7 @@ def stream_iberotales_response(
         )
 
         full_response = ""
+        messages = []
         thinking_message_index = None
         solution_message_index = None
         in_think_block = False
@@ -352,15 +317,8 @@ def stream_iberotales_response(
                 yield messages
                 
     except Exception as e:
-        messages.append(ChatMessage(role="assistant", content=f"Error: {str(e)}"))
-        yield messages
-
-def user_message(msg: str, history: list) -> tuple[str, list]:
-    """Añade mensaje del usuario al historial"""
-    history.append(ChatMessage(role="user", content=msg))
-    return "", history
+        yield [ChatMessage(role="assistant", content=f"Error: {str(e)}")]
 
-# 1. CAMBIO EN LA FUNCIÓN actualizar_personajes:
 def actualizar_personajes(pais_seleccionado):
     """Actualiza la galería de personajes según el país seleccionado"""
     global current_personajes
@@ -370,11 +328,11 @@ def actualizar_personajes(pais_seleccionado):
     if not personajes:
         return [], "Selecciona un país para ver sus personajes"
     
-    # Crear lista de imágenes y etiquetas para la galería - CAMBIO AQUÍ
+    # Crear lista de imágenes para la galería
     imagenes = []
     for p in personajes:
         if os.path.exists(p["imagen"]):
-            imagenes.append(p["imagen"])  # Solo la ruta de la imagen
+            imagenes.append(p["imagen"])
         else:
             # Imagen placeholder si no existe
             imagenes.append("https://via.placeholder.com/100x100.png?text=No+Image")
@@ -388,17 +346,13 @@ def crear_prompt_desde_personaje(evt: gr.SelectData):
     try:
         if evt.index is not None and evt.index < len(current_personajes):
             personaje = current_personajes[evt.index]
-            return f"Crea una historia sobre {personaje['nombre']}, {personaje['descripcion']}"  #si alguien lee esto, cambiar el dataste a cuenta en lugar de crea
+            return f"Crea una historia sobre {personaje['nombre']}, {personaje['descripcion']}"
         else:
             return "Crea una historia sobre un personaje mítico"
     except Exception as e:
         print(f"Error al crear prompt: {e}")
         return "Crea una historia sobre un personaje mítico"
 
-# Aplicar decorador @spaces.GPU si es necesario
-# if IS_HF_SPACE and SPACES_AVAILABLE and torch.cuda.is_available():
-#    stream_iberotales_response = spaces.GPU(stream_iberotales_response)
-
 # CSS personalizado para mejorar la apariencia
 custom_css = """
 .gradio-container {
@@ -472,12 +426,8 @@ with gr.Blocks(fill_height=True, title="Iberotales", css=custom_css) as demo:
                 elem_id="galeria",
                 columns=1,
                 rows=4,
-                height=350,
-                # REMOVER ESTAS LÍNEAS QUE CAUSAN EL ERROR:
-                # object_fit="cover",
-                # preview=False
+                height=350
             )
-
         
         # Panel derecho - Chat
         with gr.Column(scale=2):
@@ -505,30 +455,32 @@ with gr.Blocks(fill_height=True, title="Iberotales", css=custom_css) as demo:
                         max_tokens = gr.Slider(100, MAX_MAX_NEW_TOKENS, DEFAULT_MAX_NEW_TOKENS, label="Tokens", container=False)
                         temperature = gr.Slider(0.1, 2.0, 0.7, label="Temp", container=False)
 
-    # Variables de estado
-    msg_store = gr.State("")
-
     # Eventos
-    def submit_message(msg, history):
+    def generate_and_display(msg, max_tok, temp):
+        """Genera respuesta y la muestra en el chatbot"""
         if not msg.strip():
-            return msg, history
-        return "", user_message(msg, history)[1]
-    
-    def generate_response(msg, history, max_tok, temp):
-        yield from stream_iberotales_response(msg, history, DEFAULT_SYSTEM_MESSAGE, max_tok, temp)
+            return []
+        
+        # Agregar mensaje del usuario
+        messages = [ChatMessage(role="user", content=msg)]
+        yield messages
+        
+        # Generar respuesta
+        for response_messages in stream_iberotales_response(msg, DEFAULT_SYSTEM_MESSAGE, max_tok, temp):
+            yield [ChatMessage(role="user", content=msg)] + response_messages
 
     # Actualizar personajes cuando cambia el país
     pais_dropdown.change(
         fn=actualizar_personajes,
         inputs=[pais_dropdown],
-        outputs=[galeria_personajes]  # REMOVER el segundo output
+        outputs=[galeria_personajes]
     )
     
     # Cargar personajes iniciales
     demo.load(
         fn=actualizar_personajes,
         inputs=[pais_dropdown],
-        outputs=[galeria_personajes]  # REMOVER el segundo output
+        outputs=[galeria_personajes]
     )
     
     # Crear prompt desde galería
@@ -539,30 +491,26 @@ with gr.Blocks(fill_height=True, title="Iberotales", css=custom_css) as demo:
 
     # Envío de mensajes
     input_box.submit(
-        lambda msg, hist: (msg, submit_message(msg, hist)[1]),
-        inputs=[input_box, chatbot],
-        outputs=[msg_store, chatbot],
-        queue=False
-    ).then(
-        generate_response,
-        inputs=[msg_store, chatbot, max_tokens, temperature],
+        generate_and_display,
+        inputs=[input_box, max_tokens, temperature],
         outputs=chatbot
+    ).then(
+        lambda: "",
+        outputs=input_box
     )
 
     send_button.click(
-        lambda msg, hist: (msg, submit_message(msg, hist)[1]),
-        inputs=[input_box, chatbot],
-        outputs=[msg_store, chatbot],
-        queue=False
-    ).then(
-        generate_response,
-        inputs=[msg_store, chatbot, max_tokens, temperature],
+        generate_and_display,
+        inputs=[input_box, max_tokens, temperature],
         outputs=chatbot
+    ).then(
+        lambda: "",
+        outputs=input_box
     )
 
     clear_button.click(
-        lambda: ([], "", ""),
-        outputs=[chatbot, input_box, msg_store],
+        lambda: ([], ""),
+        outputs=[chatbot, input_box],
         queue=False
     )