Update app.py

app.py

@@ -1,146 +1,109 @@
-import re
 import os
-import gradio as gr
+import re
 import random
+import gradio as gr
 from fuzzywuzzy import fuzz
 
-# Función para cargar el dataset desde el archivo local
+# Cargar dataset desde archivo
 def cargar_dataset(filename="datos_alojamientos.txt"):
-    try:
-        if not os.path.exists(filename):
-            raise FileNotFoundError(f"Archivo no encontrado: {filename}")
+    if not os.path.exists(filename):
+        raise FileNotFoundError(f"Archivo no encontrado: {filename}")
+    with open(filename, "r", encoding="utf-8") as f:
+        contenido = f.read().strip()
+    return [bloque.strip() for bloque in contenido.split("---") if bloque.strip()]
 
-        with open(filename, "r", encoding="utf-8") as f:
-            contenido = f.read().strip()
-            alojamientos = [bloque.strip() for bloque in contenido.split("\n\n") if bloque.strip()]
-        return alojamientos
-    except Exception as e:
-        print(f"Error al cargar el dataset: {e}")
-        return []
+dataset = cargar_dataset()
 
-# Función para limpiar y normalizar texto
+# Normalización de texto
 def limpiar_texto(texto):
     return re.sub(r'\s+', ' ', texto.strip().lower())
 
-# Función para extraer valores correctamente sin dejar "No especificado"
 def obtener_valor(texto, patron):
     match = re.search(patron, texto, re.IGNORECASE | re.MULTILINE)
-    if match and match.group(1).strip():
-        return match.group(1).strip()
-    return "No disponible"
+    return match.group(1).strip() if match else "No disponible"
 
-# Función para extraer la lista de servicios correctamente
 def obtener_servicios(texto):
     match = re.findall(r"servicios\s*:\s*(.+)", texto, re.IGNORECASE | re.MULTILINE)
-    if match:
-        servicios_lista = [s.strip() for s in match[0].split(",")]
-        return "\n".join([f"🛠 {servicio}" for servicio in servicios_lista])
-    return "No especificado"
-
-# Función para buscar coincidencias difusas
-def buscar_coincidencia(pregunta, texto):
-    return fuzz.partial_ratio(limpiar_texto(pregunta), limpiar_texto(texto)) > 70
+    return "\n".join([f"🛠 {s.strip()}" for s in match[0].split(",")]) if match else "No especificado"
 
-# Variable global para almacenar resultados previos
-resultados_previos = []
-
-# Función principal para responder preguntas
-def responder_pregunta(pregunta, historial):
-    global resultados_previos
+# Filtrar alojamientos
+def filtrar_alojamientos(pregunta):
     pregunta = limpiar_texto(pregunta)
     resultados = []
     cantidad_solicitada = next((int(p) for p in pregunta.split() if p.isdigit()), None)
-
-    # Lista de palabras clave
-    palabras_clave_servicios = [
-        "wi fi", "directv", "ropa blanca", "amueblado", "equipado", "habitación", 
-        "cocina", "comedor", "baño", "sommier", "cama", "vajilla completa", "heladera con freezer", 
-        "tv led", "aire acondicionado", "parrilla", "pileta", "lavarropas", "garage", "quincho", "balcón"
-    ]
-
-    busca_servicio = next((s for s in palabras_clave_servicios if s in pregunta), None)
-
+    busca_mascotas = "mascota" in pregunta or "pet friendly" in pregunta
+    busca_wifi = "wifi" in pregunta or "internet" in pregunta
+    
     for alojamiento in dataset:
-        if buscar_coincidencia(pregunta, alojamiento) or \
-           (cantidad_solicitada and (f"{cantidad_solicitada} plazas" in alojamiento or f"para {cantidad_solicitada} personas" in alojamiento)) or \
-           (busca_servicio and busca_servicio in alojamiento):
+        if (cantidad_solicitada and f"plazas: {cantidad_solicitada}" in alojamiento.lower()) or \
+           (busca_mascotas and "mascotas: sí" in alojamiento.lower()) or \
+           (busca_wifi and "wifi: sí" in alojamiento.lower()) or \
+           fuzz.partial_ratio(pregunta, alojamiento) > 70:
             resultados.append(alojamiento)
+    
+    random.shuffle(resultados)
+    return resultados
 
-    if resultados:
-        random.shuffle(resultados)  # 🔥 Ordena aleatoriamente los resultados
-        mostrar_resultados = resultados[:2]  # Muestra al menos 2 alojamientos
-        resultados_previos = resultados[2:]  # Guarda los resultados restantes para "ver más"
-
-        historial += "\n\n" + "\n\n---\n\n".join(formatear_alojamiento(aloj) for aloj in mostrar_resultados)
+# Formatear la respuesta
+def formatear_alojamiento(texto):
+    nombre = obtener_valor(texto, r"alojamiento\s*:\s*(.+)")
+    direccion = obtener_valor(texto, r"dirección\s*:\s*(.+)")
+    telefono = obtener_valor(texto, r"tel[eé]fono[s]*\s*:\s*(.+)")
+    email = obtener_valor(texto, r"email\s*:\s*(.+)")
+    plazas = obtener_valor(texto, r"plazas\s*:\s*(\d+)")
+    mascotas = "Sí" if "mascotas: sí" in texto.lower() else "No"
+    wifi = "Sí" if "wifi: sí" in texto.lower() else "No"
+    servicios = obtener_servicios(texto)
+    return f"""🏠 **{nombre}**\n📍 Dirección: {direccion}\n📞 Teléfono: {telefono}\n📧 Email: {email}\n🛏 Plazas: {plazas}\n🐶 Mascotas: {mascotas}\n📶 Wi-Fi: {wifi}\n{servicios}"""
 
-        if len(resultados_previos) > 0:
-            historial += "\n\n🔎 ¿Quieres ver más resultados? Escribe 'ver más'."
+# Manejo de paginación
+resultados_previos = []
 
-        return historial
-    return "No encontré alojamientos que coincidan con tu búsqueda."
+def responder_pregunta(pregunta, historial):
+    global resultados_previos
+    resultados = filtrar_alojamientos(pregunta)
+    
+    if not resultados:
+        return "No se encontraron alojamientos que coincidan con su búsqueda."
+    
+    mostrar = resultados[:3]
+    resultados_previos = resultados[3:]
+    historial += "\n\n".join(formatear_alojamiento(aloj) for aloj in mostrar)
+    
+    if resultados_previos:
+        historial += "\n\n¿Desea ver más opciones?"
+    
+    return historial
 
-# Función para mostrar más resultados cuando el usuario escribe "ver más"
 def ver_mas(historial):
     global resultados_previos
     if not resultados_previos:
         return historial + "\n\nNo hay más alojamientos para mostrar."
     
-    mostrar_resultados = resultados_previos[:2]  # Muestra los siguientes 2 alojamientos
-    resultados_previos = resultados_previos[2:]  # Actualiza la lista eliminando los ya mostrados
+    mostrar = resultados_previos[:3]
+    resultados_previos = resultados_previos[3:]
+    historial += "\n\n".join(formatear_alojamiento(aloj) for aloj in mostrar)
+    
+    if resultados_previos:
+        historial += "\n\n¿Desea ver más opciones?"
     
-    historial += "\n\n" + "\n\n---\n\n".join(formatear_alojamiento(aloj) for aloj in mostrar_resultados)
-
-    if len(resultados_previos) > 0:
-        historial += "\n\n🔎 ¿Quieres ver más resultados? Escribe 'ver más'."
-
     return historial
 
-# Función para formatear la información de los alojamientos correctamente
-def formatear_alojamiento(texto):
-    nombre = obtener_valor(texto, r"(?i)nombre\s*:\s*(.+)")
-    direccion = obtener_valor(texto, r"(?i)dirección\s*:\s*(.+)")
-    telefono = obtener_valor(texto, r"(?i)tel[eé]fono\s*:\s*(.+)")
-    plazas = obtener_valor(texto, r"(?i)plazas\s*:\s*(.+)")
-    mascotas = "Sí" if re.search(r"mascotas\s*:\s*sí", texto, re.IGNORECASE) else "No"
-    wifi = "Sí" if re.search(r"wifi\s*:\s*sí", texto, re.IGNORECASE) else "No"
-    servicios = obtener_servicios(texto)
-
-    return f"""🏠 Nombre: {nombre}
-📍 Dirección: {direccion}
-📞 Teléfono: {telefono}
-🛏 Plazas: {plazas}
-🐶 Política de mascotas: {mascotas}
-📶 Wi-Fi: {wifi}
-{servicios}"""
-
-# Cargar el dataset
-dataset = cargar_dataset()
-
-# Interfaz con Gradio - Mantiene historial del chat
-with gr.Blocks(theme=gr.themes.Default()) as iface:
+# Interfaz con Gradio
+with gr.Blocks() as iface:
     gr.Markdown("### Bienvenido a Valer-IA, tu informante turístico virtual 🏡")
-    gr.Markdown("Escribe una consulta sobre alojamientos, por ejemplo: '¿Qué alojamientos aceptan mascotas?' o 'Alojamiento con pileta y aire acondicionado'")
-
     chat_historial = gr.Textbox(label="Historial de chat", lines=15, interactive=False)
     pregunta_input = gr.Textbox(label="Escribe tu consulta:")
-
+    
     def actualizar_chat(historial, nueva_pregunta):
         nueva_pregunta = nueva_pregunta.strip().capitalize()
-        saludos = ["hola", "buenos días", "buenas tardes", "buenas noches", "qué tal", "hey", "saludos"]
-
-        if nueva_pregunta.lower() in saludos:
-            respuesta = "¡Hola! 😊 Soy Valer-IA, tu asistente turístico virtual. ¿En qué puedo ayudarte hoy?"
-        elif nueva_pregunta.lower() == "ver más":
-            return ver_mas(historial), ""  # Muestra más resultados sin borrar el historial
-        else:
-            respuesta = responder_pregunta(nueva_pregunta, historial)
-
-        return respuesta, ""
-
-    preguntar_btn = gr.Button("Preguntar")
+        if nueva_pregunta.lower() == "más" or nueva_pregunta.lower() == "mas":
+            return ver_mas(historial), ""
+        return responder_pregunta(nueva_pregunta, historial), ""
     
-    # 🔥 Se borra la pregunta después de ser contestada
+    preguntar_btn = gr.Button("Preguntar")
     preguntar_btn.click(actualizar_chat, inputs=[chat_historial, pregunta_input], outputs=[chat_historial, pregunta_input])
 
 if __name__ == "__main__":
     iface.launch()
+