Update app.py

app.py

@@ -1,20 +1,19 @@
 import subprocess
 import sys
-import spacy
 import gradio as gr
 import re
 import os
 import random
 from fuzzywuzzy import fuzz
+from transformers import TFAutoModelForSequenceClassification, AutoTokenizer
 
 # Instalar dependencias
-subprocess.check_call([sys.executable, "-m", "pip", "install", "spacy", "fuzzywuzzy", "python-levenshtein", "gradio"])
+subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers", "fuzzywuzzy", "python-levenshtein", "gradio"])
 
-# Descargar el modelo de lenguaje
-spacy.cli.download("es_core_news_sm")
-
-# Cargar el modelo de spaCy
-nlp = spacy.load("es_core_news_sm")
+# Descargar el modelo y el tokenizador
+model_name = "huggingface/distilbert-base-uncased-finetuned-sst-2-english"
+model = TFAutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Diccionario de sinónimos para normalizar palabras clave
 sinonimos = {
@@ -250,40 +249,3 @@ def formatear_alojamiento(texto):
         print(f"Nombre extraído del primer bloque: {nombre}")
 
     return f"""🏠 Nombre: {nombre}
-📍 Dirección: {direccion}
-📞 Teléfono: {telefono}
-🛏 Plazas: {plazas}
-🐶 Política de mascotas: {mascotas}
-📶 Wi-Fi: {wifi}
-🛎 Servicios: {servicios}"""
-
-# Cargar el dataset
-dataset = cargar_dataset()
-
-# Interfaz con Gradio - Mantiene historial del chat
-with gr.Blocks(theme=gr.themes.Default()) 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() in ["ver mas", "ver más"]:
-            return ver_mas(historial), ""
-        else:
-            respuesta = responder_pregunta(nueva_pregunta, historial)
-
-        return respuesta, ""
-
-    preguntar_btn = gr.Button("Preguntar")
-
-    # Se borra la pregunta después de ser contestada
-    preguntar_btn.click(actualizar_chat, inputs=[chat_historial, pregunta_input], outputs=[chat_historial, pregunta_input])
-
-if __name__ == "__main__":
-    iface.launch()