Update app.py

app.py

@@ -12,8 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 model.eval()
 
-# Cargar alojamientos en memoria
-
+# Cargar alojamientos en memoria con preprocesamiento
 def cargar_alojamientos():
     with open("alojamientos.txt", "r", encoding="utf-8") as file:
         alojamientos = file.read().split("\n\n")
@@ -21,14 +20,14 @@ def cargar_alojamientos():
 
 alojamientos_info = cargar_alojamientos()
 
-# Índice invertido
+# Índice invertido optimizado
 indice_palabras = defaultdict(set)
 for idx, alojamiento in alojamientos_info.items():
     for palabra in re.split(r'\W+', alojamiento.lower()):
         if len(palabra) > 2:
             indice_palabras[palabra].add(idx)
 
-# Caché LRU
+# Caché LRU con límite de tamaño
 class LRUCache:
     def __init__(self, capacity=100):
         self.cache = OrderedDict()
@@ -53,68 +52,55 @@ class LRUCache:
 cache_respuestas = LRUCache()
 cache_paginas = LRUCache()
 
-# Función de búsqueda
+# Función de búsqueda optimizada
 def buscar_alojamiento(consulta):
     consulta = consulta.lower()
     cached = cache_respuestas.get(consulta)
     if cached is not None:
         return cached
 
-    match_plazas = re.search(r"\b(\d+)\s*(personas|plazas)\b", consulta)
-    plazas = int(match_plazas.group(1)) if match_plazas else None
-
     palabras = set(re.split(r'\W+', consulta))
     indices = set()
     for palabra in palabras:
         if palabra in indice_palabras:
             indices.update(indice_palabras[palabra])
 
-    resultados = [
-        alojamientos_info[idx] for idx in indices
-        if not plazas or f"Plazas: {plazas}" in alojamientos_info[idx]
-    ]
-
+    resultados = [alojamientos_info[idx] for idx in indices]
     cache_respuestas.put(consulta, resultados)
     cache_paginas.put(consulta, 0)
     return resultados
 
-# Función de formateo
+# Formateo visual optimizado
 def formatear_alojamiento(texto):
-    lineas = texto.split("\n")
-    if not lineas:
-        return ""
-
-    nombre_alojamiento = lineas[0].strip()
-    resultado = [
-        f"🏠 **{nombre_alojamiento}**",
-        "----------------------------------------------------------"
-    ]
-
-    iconos = {
-        "Dirección:": "📍", "Teléfono:": "📞", "Email:": "📧",
-        "Plazas:": "👤", "Mascotas": "🐾", "Wifi": "📶", "Directv": "📺",
-        "Ropa blanca": "🛏️", "Habilitación provincial": "📝", "Servicios:": "🛠"
-    }
-
-    for linea in lineas[1:]:
-        for clave, icono in iconos.items():
-            if clave in linea:
-                resultado.append(f"{icono} **{linea}**")
-                break
-        else:
-            if "Descripción:" in linea:
-                resultado.append(f"\n🔹 **{linea.replace('Descripción:', '').strip()}**")
-            elif linea.strip():
-                resultado.append(f"   - {linea.strip()}")
-
-    resultado.append("----------------------------------------------------------")
-    return "\n".join(resultado)
-
-# Mostrar resultados
+    bloques = texto.split("\n\n")
+    resultado = []
+    
+    for bloque in bloques:
+        lineas = bloque.split("\n")
+        alojamiento_info = []
+        tipos_alojamiento = []
+        
+        for linea in lineas:
+            if "Plazas:" in linea:
+                tipos_alojamiento.append(linea)
+            elif "Descripción:" in linea:
+                tipos_alojamiento.append(linea)
+            elif "Servicios:" in linea:
+                tipos_alojamiento.append(linea)
+            else:
+                alojamiento_info.append(linea)
+                
+        resultado.append("\n".join(alojamiento_info))
+        if tipos_alojamiento:
+            resultado.append("\nTipos de alojamiento:\n" + "\n".join(tipos_alojamiento))
+    
+    return "\n\n".join(resultado)
+
+# Paginación y resultados
 def mostrar_resultados(consulta):
     resultados = buscar_alojamiento(consulta)
     if not resultados:
-        return "Lo siento, no encontré información exacta.", ""
+        return "Lo siento, no encontré información exacta. Intenta preguntar de otra manera.", ""
 
     pagina = cache_paginas.get(consulta) or 0
     inicio, fin = pagina * 3, (pagina + 1) * 3
@@ -130,42 +116,15 @@ def mostrar_resultados(consulta):
 
     return respuesta, pregunta_mas
 
-# Generación con DialoGPT
-def generar_respuesta_dialoGPT(message):
-    prompt = f"Eres un asistente de turismo especializado en alojamientos.\nResponde de manera clara y concisa a: '{message}'"
-    with torch.no_grad():
-        inputs = tokenizer.encode(prompt + tokenizer.eos_token, return_tensors="pt").to(device)
-        reply_ids = model.generate(
-            inputs, max_length=100, temperature=0.7, top_p=0.9,
-            pad_token_id=tokenizer.eos_token_id, do_sample=True
-        )
-    return tokenizer.decode(reply_ids[:, inputs.shape[-1]:][0], skip_special_tokens=True)
-
-# Chat
-def chat(message):
-    message = message.strip().lower()
-    if message in ["sí", "atrás"]:
-        consulta = historial_respuestas[-1]["consulta"] if historial_respuestas else ""
-        respuesta, pregunta_mas = mostrar_resultados(consulta)
-    else:
-        resultados = buscar_alojamiento(message)
-        respuesta, pregunta_mas = mostrar_resultados(message) if resultados else (generar_respuesta_dialoGPT(message), "")
-
-    historial_respuestas.append({"consulta": message, "respuesta": respuesta})
-    if len(historial_respuestas) > 10:
-        historial_respuestas.pop(0)
-
-    return "\n\n".join(f"**Pregunta:** {h['consulta']}\n\n{h['respuesta']}" for h in historial_respuestas), pregunta_mas
-
 # Interfaz con Gradio
-historial_respuestas = []
 with gr.Blocks(title="Chat de Turismo") as iface:
     gr.Markdown("### Asistente de Turismo - Alojamientos")
     output_box = gr.Textbox(label="Historial", lines=15, interactive=False)
-    input_box = gr.Textbox(label="Consulta", placeholder="Escribe aquí...")
+    input_box = gr.Textbox(label="Consulta", placeholder="Escribe aquí y presiona Enter...")
     extra_box = gr.Textbox(label="Opciones", interactive=False)
     send_button = gr.Button("Enviar")
-    send_button.click(chat, inputs=input_box, outputs=[output_box, extra_box])
-    input_box.submit(chat, inputs=input_box, outputs=[output_box, extra_box])
+
+    send_button.click(mostrar_resultados, inputs=input_box, outputs=[output_box, extra_box])
+    input_box.submit(mostrar_resultados, inputs=input_box, outputs=[output_box, extra_box])
 
 iface.launch(share=True, inbrowser=True)