Update app.py

app.py

@@ -191,7 +191,7 @@ def buscar_por_genero(genero, k=5):
     # Formatear las recomendaciones
     recommendations = []
     for title in recommended_titles:
-        recommendations.append(f"### {title}")
+        recommendations.append(f"- {title}")
 
     return "\n".join(recommendations) if recommendations else f"No se encontraron K-Dramas del género '{genero}'."
 
@@ -203,7 +203,7 @@ def recomendar_kdramas_chat(entrada_usuario):
     if preferencia_traducida:
         return buscar_por_genero(preferencia_traducida, k=5)
     else:
-        return "Por favor, dime qué tipo de K-Drama te gusta (romántico, acción, misterio, etc.)."
+        return None  # No se encontró una preferencia válida
 
 # Función para generar respuestas del chatbot
 def generar_respuesta(entrada_usuario, historial_chat=""):
@@ -214,20 +214,27 @@ def generar_respuesta(entrada_usuario, historial_chat=""):
 
 # Función para manejar la interacción del chatbot
 def chat(entrada_usuario, historial_chat=""):
-    recomendacion = ""
-    # Check if input contains preference keywords 
-    if any(palabra in entrada_usuario.lower() for palabra in mapeo_generos.keys()):
-        recomendacion = recomendar_kdramas_chat(entrada_usuario)
-        respuesta = recomendacion # Update response to include recommendations (if any)
+    # Agregar el mensaje del usuario al historial
+    historial_chat += f"Usuario: {entrada_usuario}\n"
+
+    # Generar una respuesta natural del chatbot
+    if not historial_chat.strip():  # Si es la primera interacción
+        respuesta = "¡Hola! Soy tu asistente para recomendar K-Dramas. ¿Qué tipo de K-Drama te gustaría ver hoy?"
     else:
         respuesta = generar_respuesta(entrada_usuario, historial_chat)
-        # Pass chatbot's response to preference check (in case it suggests a genre)
-        recomendacion = recomendar_kdramas_chat(respuesta)
-        # Add recommendations to chatbot response only if found
-        if recomendacion != "Por favor, dime qué tipo de K-Drama te gusta (romántico, acción, misterio, etc.).":
-            respuesta = f"{respuesta}\n\n{recomendacion}" 
-    
-    return respuesta, gr.update(value="") # update with empty string
+
+        # Verificar si el usuario mencionó un género
+        recomendacion = recomendar_kdramas_chat(entrada_usuario)
+        if recomendacion:
+            respuesta = f"¡Genial! Aquí tienes algunas recomendaciones de K-Dramas que podrían gustarte:\n{recomendacion}"
+        else:
+            respuesta = "¿Qué tipo de K-Drama te gustaría ver? Puedo recomendarte dramas de acción, romance, misterio y más."
+
+    # Agregar la respuesta del chatbot al historial
+    historial_chat += f"Chatbot: {respuesta}\n"
+
+    # Devolver la respuesta y el historial actualizado
+    return respuesta, historial_chat, ""  # Limpiar el cuadro de texto
 
 # ==================================================
 # Interfaz de Gradio