Update app.py

app.py

@@ -75,28 +75,25 @@ def formatear_alojamiento(texto):
     bloques = texto.split("\n\n")
     resultado = []
     nombre_alojamiento = ""
-    
+    datos_contacto = []
+    tipos_alojamiento = []
+
     for bloque in bloques:
         lineas = bloque.split("\n")
-        alojamiento_info = []
-        tipos_alojamiento = []
-        
         for linea in lineas:
             if "Alojamiento:" in linea:
                 nombre_alojamiento = linea  # Guardamos el nombre del alojamiento
-                alojamiento_info.append(linea)
-            elif "Plazas:" in linea:
-                tipos_alojamiento.append(linea)
-            elif "Descripción:" in linea:
+            elif any(tag in linea for tag in ["Dirección:", "Teléfono:", "Email:", "Plazas Totales:", "Mascotas:", "Wifi:", "Directv:", "Ropa blanca:", "Habilitación provincial:"]):
+                datos_contacto.append(linea)
+            elif "Plazas:" in linea or "Descripción:" in linea or "Servicios:" 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(f"{nombre_alojamiento}\nTipos de alojamiento:\n" + "\n".join(tipos_alojamiento))
+
+    if nombre_alojamiento:
+        resultado.append(nombre_alojamiento)
+    if datos_contacto:
+        resultado.append("\n".join(datos_contacto))
+    if tipos_alojamiento:
+        resultado.append("Tipos de alojamiento:\n" + "\n".join(tipos_alojamiento))
     
     return "\n\n".join(resultado)
 
@@ -132,3 +129,4 @@ with gr.Blocks(title="Chat de Turismo") as iface:
     input_box.submit(mostrar_resultados, inputs=input_box, outputs=[output_box, extra_box])
 
 iface.launch(share=True, inbrowser=True)
+