Upload 5 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+menu.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,212 @@
+from os import getenv
+from dotenv import load_dotenv
+import gradio as gr
+import os
+import json
+from langchain_huggingface import HuggingFaceEmbeddings
+from langchain_chroma import Chroma
+from langchain_openai import ChatOpenAI
+from langchain.schema import Document
+
+# Cargar variables de entorno desde un archivo .env
+load_dotenv()
+
+# Inicializar el modelo de lenguaje de OpenAI
+llm = ChatOpenAI(
+    openai_api_key=getenv("OPENROUTER_API_KEY"),
+    openai_api_base=getenv("OPENROUTER_BASE_URL"),
+    model_name="openai/gpt-4o",
+    model_kwargs={
+        "extra_headers": {
+            "Helicone-Auth": f"Bearer " + getenv("HELICONE_API_KEY")
+        }
+    },
+)
+
+# Cambiar el directorio de trabajo
+os.chdir(r"MIA/Proyecto 4")
+
+# Cargar datos de preguntas frecuentes desde un archivo JSON
+faqs_path = r"faqs.json"
+with open(faqs_path, "r") as file:
+    faqs_data = json.load(file)
+
+# Cargar datos del menú desde un archivo JSON
+menu_path = r"menu.json"
+with open(menu_path, "r") as file:
+    menu_data = json.load(file)
+
+# Definir la ruta del archivo de reservas
+reservas_path = r"reservas.json"
+
+# Definir la ruta del archivo PDF del menú
+menu_pdf_path = r"menu.pdf"
+
+# Crear el archivo de reservas si no existe
+if not os.path.exists(reservas_path):
+    with open(reservas_path, "w") as file:
+        json.dump([], file)
+
+# Cargar datos de reservas desde un archivo JSON
+with open(reservas_path, "r") as file:
+    reservas_data = json.load(file)
+
+# Crear documentos a partir de las preguntas frecuentes
+faqs_documents = [
+    Document(
+        page_content=pair["question"] + " " + pair["answer"],
+        metadata={"id": str(i)}
+    )
+    for i, pair in enumerate(faqs_data)
+]
+
+# Crear documentos a partir del menú
+menu_documents = [
+    Document(
+        page_content=f"{nombre}: Categoría: {info.get('categoria', 'Desconocida')}, "
+                     f"Alérgenos: {', '.join(info.get('alergenos', [])) if info.get('alergenos') else 'Ninguno'}, "
+                     f"Precio: ${info.get('precio', 'Desconocido')}, "
+                     f"Descripción: {info.get('descripcion', 'Sin descripción')}, "
+                     f"Sin Gluten: {'Sí' if info.get('sin_gluten') else 'No'}, "
+                     f"Vegan: {'Sí' if info.get('vegan') else 'No'}",
+        metadata={"tipo": "menu"}
+    )
+    for nombre, info in menu_data.items()
+]
+
+# Inicializar el modelo de embeddings de HuggingFace
+embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
+
+# Inicializar la tienda de vectores Chroma
+vectorstore = Chroma(embedding_function=embeddings)
+
+# Agregar documentos de preguntas frecuentes y menú a la tienda de vectores
+vectorstore.add_documents(faqs_documents)
+vectorstore.add_documents(menu_documents)
+
+# Estado de la reserva
+estado_reserva = {}
+
+# Función del chatbot
+def chatbot(message, history):
+    global estado_reserva
+
+    message_lower = message.lower()
+
+    # Palabras clave para mostrar el menú
+    menu_keywords = [
+        "menu completo", "menú completo", "carta completa",
+        "ver menú", "ver menu", "mostrar menú", "mostrar menu", "quiero ver el menú",
+        "quiero ver la carta", "mostrar la carta", "ver la carta", "enséñame el menú",
+        "enséñame la carta", "dame el menú", "dame la carta", "muéstrame el menú",
+        "muéstrame la carta", "quiero el menú", "quiero la carta", "menu por favor",
+        "menú por favor", "carta por favor", "puedo ver el menú", "puedo ver la carta"
+    ]
+    
+    # Mostrar imagen del menú si el mensaje coincide con alguna palabra clave
+    if any(message_lower.strip() == word for word in menu_keywords):
+        yield gr.Image("menu.png")
+        return
+
+    # Continuar con el proceso de reserva si está en proceso
+    if estado_reserva.get("en_proceso"):
+        if "dia" not in estado_reserva:
+            estado_reserva["dia"] = message
+            yield "¿A qué hora deseas la reserva? (Formato HH:MM)"
+            return
+
+        elif "hora" not in estado_reserva:
+            estado_reserva["hora"] = message
+            yield "¿Para cuántas personas será la reserva?"
+            return
+
+        elif "personas" not in estado_reserva:
+            estado_reserva["personas"] = message
+            yield "A nombre de quién será la reserva?"
+            return
+        
+        elif "nombre" not in estado_reserva:
+            estado_reserva["nombre"] = message
+            yield "(Opcional) Proporcione un número de teléfono de contacto o escriba 'no' para omitirlo."
+            return
+
+        elif "telefono" not in estado_reserva:
+            estado_reserva["telefono"] = message if message.lower() != "no" else "No proporcionado"
+
+            # Guardar la nueva reserva en el archivo JSON
+            nueva_reserva = {
+                "nombre": estado_reserva["nombre"],
+                "dia": estado_reserva["dia"],
+                "hora": estado_reserva["hora"],
+                "personas": estado_reserva["personas"],
+                "telefono": estado_reserva["telefono"]
+            }
+
+            with open(reservas_path, "r") as file:
+                reservas_actuales = json.load(file)
+
+            reservas_actuales.append(nueva_reserva)
+
+            with open(reservas_path, "w") as file:
+                json.dump(reservas_actuales, file, indent=2)
+
+            estado_reserva = {}
+            yield f"✅ ¡Reserva guardada con éxito! Aquí están los detalles:\n{json.dumps(nueva_reserva, indent=2)}"
+            return
+
+    # Iniciar el proceso de reserva si se menciona en el mensaje
+    if "reserva" in message_lower or "quiero reservar" in message_lower:
+        estado_reserva["en_proceso"] = True
+        yield "¿Para qué día quieres hacer la reserva? (Formato DD/MM/AAAA)"
+        return
+
+    # Buscar documentos relevantes en la tienda de vectores
+    relevant_docs = vectorstore.similarity_search(message)
+
+    # Crear el contexto para el modelo de lenguaje
+    context_text = "\n\n".join([doc.page_content for doc in relevant_docs])
+
+    final_prompt = (
+        "Eres un asistente virtual en un restaurante. Puedes responder preguntas sobre el menú, las reservas y las preguntas frecuentes. "
+        "Si el usuario menciona restricciones dietéticas, haz recomendaciones basadas en el menú disponible.\n\n"
+        f"{json.dumps(menu_data, indent=2)}\n\n"
+        f"Contexto relevante encontrado en la base de datos:\n{context_text}\n\n"
+        f"Pregunta: {message}\n"
+        "Respuesta:"
+    )
+
+    # Generar la respuesta del modelo de lenguaje
+    messages = [{"role": "user", "content": final_prompt}]
+    response = llm.stream(messages)
+
+    partial_response = ""
+    for chunk in response:
+        if chunk and hasattr(chunk, "content"):
+            content = chunk.content
+            if content is not None:
+                partial_response += content
+                yield partial_response
+
+# Configurar la interfaz de Gradio
+demo = gr.ChatInterface(
+    chatbot,
+    chatbot=gr.Chatbot(height=400, type="messages"),
+    textbox=gr.Textbox(placeholder="Escribe tu mensaje aquí...", container=False, scale=7),
+    title="ChatBot Restaurante",
+    description="Asistente virtual para reservas, menú y preguntas frecuentes.",
+    theme="ocean",
+    examples=[
+        "¿Cuáles son los horarios del restaurante?",
+        "¿Dónde están ubicados?",
+        "¿Aceptan pagos con tarjeta?",
+        "Quiero ver la carta",
+        "Quiero hacer una reserva"
+    ],
+    type="messages",
+    editable=True,
+    save_history=True,
+)
+
+# Ejecutar la aplicación
+if __name__ == "__main__":
+    demo.queue().launch()

faqs.json

@@ -0,0 +1,78 @@
+[
+  {
+    "question": "¿Cuáles son los horarios del restaurante?",
+    "answer": "Abrimos de 12:00 AM a 11:00 PM todos los días."
+  },
+  {
+    "question": "¿Dónde están ubicados?",
+    "answer": "Nos encontramos en Avenida Gastronómica 456, Ciudad."
+  },
+  {
+    "question": "¿Aceptan pagos con tarjeta?",
+    "answer": "Sí, aceptamos todas las tarjetas de crédito y débito, así como pagos móviles."
+  },
+  {
+    "question": "¿Ofrecen servicio de entrega a domicilio?",
+    "answer": "Sí, realizamos entregas a domicilio a través de nuestras plataformas asociadas."
+  },
+  {
+    "question": "¿Tienen opciones vegetarianas y veganas?",
+    "answer": "Sí, contamos con opciones vegetarianas y veganas en nuestro menú."
+  },
+  {
+    "question": "¿Tienen opciones sin gluten?",
+    "answer": "Sí, disponemos de opciones sin gluten. Consulte con nuestro personal para más detalles."
+  },
+  {
+    "question": "¿Cuentan con estacionamiento?",
+    "answer": "Sí, tenemos un estacionamiento privado para nuestros clientes."
+  },
+  {
+    "question": "¿Se pueden hacer pedidos para llevar?",
+    "answer": "Sí, ofrecemos servicio de comida para llevar. Puede realizar su pedido por teléfono o en persona."
+  },
+  {
+    "question": "¿Ofrecen menú infantil?",
+    "answer": "Sí, contamos con un menú especial para niños con porciones y opciones adecuadas."
+  },
+  {
+    "question": "¿Se necesita hacer una reserva?",
+    "answer": "No es necesario, pero recomendamos hacer una reserva en horarios pico o para grupos grandes."
+  },
+  {
+    "question": "¿Aceptan mascotas?",
+    "answer": "Sí, tenemos un área pet-friendly para que puedas disfrutar con tu mascota."
+  },
+  {
+    "question": "¿Tienen opciones sin lactosa?",
+    "answer": "Sí, ofrecemos opciones sin lactosa en nuestros platos y postres. Pregunta a nuestro personal."
+  },
+  {
+    "question": "¿Cuáles son las recomendaciones del chef?",
+    "answer": "Nuestras recomendaciones varían según la temporada, pero te sugerimos probar el 'Salmón a la Parrilla' y nuestra 'Pizza Margherita'."
+  },
+  {
+    "question": "¿Tienen menú para personas con diabetes?",
+    "answer": "Sí, tenemos opciones bajas en azúcar y adaptadas para personas con diabetes. Por favor, consulta con nuestro personal para más detalles."
+  },
+  {
+    "question": "¿Es necesario dejar propina?",
+    "answer": "La propina es completamente opcional, pero siempre es apreciada por nuestro equipo. Normalmente, se recomienda entre un 10% y un 15% del total de la cuenta."
+  },
+  {
+    "question": "¿Puedo personalizar los platos según mis preferencias?",
+    "answer": "Sí, estamos encantados de adaptar los platos a tus preferencias. Puedes pedir cambios en los ingredientes o ajustar los sabores."
+  },
+  {
+    "question": "¿Tienen algún menú especial para celebraciones?",
+    "answer": "Sí, ofrecemos menús especiales para celebraciones y eventos. Puedes pedir más información sobre nuestros paquetes de eventos."
+  },
+  {
+    "question": "¿Qué opciones de bebidas ofrecen?",
+    "answer": "Ofrecemos una amplia variedad de bebidas, incluyendo refrescos, jugos naturales, vinos, cervezas y cócteles."
+  },
+  {
+    "question": "¿Cómo puedo ponerme en contacto con ustedes?",
+    "answer": "Puedes contactarnos por teléfono, por email o a través de nuestras redes sociales. Todos los detalles están en nuestra página de contacto."
+  }
+]

menu.json

@@ -0,0 +1,147 @@
+{
+  "Pizza Margherita": {
+    "categoria": "Vegetariano",
+    "alergenos": ["Gluten", "Lácteos"],
+    "precio": 12.00,
+    "descripcion": "Pizza clásica con salsa de tomate, mozzarella fresca y albahaca.",
+    "sin_gluten": true,
+    "vegan": false
+  },
+  "Hamburguesa Clásica": {
+    "categoria": "Carnes",
+    "alergenos": ["Gluten", "Huevo"],
+    "precio": 15.00,
+    "descripcion": "Hamburguesa de carne de res con lechuga, tomate y mayonesa.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Ensalada César": {
+    "categoria": "Ensaladas",
+    "alergenos": ["Lácteos", "Huevo", "Pescado"],
+    "precio": 10.00,
+    "descripcion": "Lechuga romana, pollo a la parrilla, crutones, queso parmesano y aderezo César.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Salmón a la Parrilla": {
+    "categoria": "Pescados",
+    "alergenos": ["Pescado"],
+    "precio": 18.00,
+    "descripcion": "Filete de salmón a la parrilla con limón y hierbas.",
+    "sin_gluten": true,
+    "vegan": false
+  },
+  "Pasta Alfredo": {
+    "categoria": "Pastas",
+    "alergenos": ["Gluten", "Lácteos"],
+    "precio": 14.00,
+    "descripcion": "Pasta con una salsa cremosa de queso parmesano y ajo.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Tacos de Pollo": {
+    "categoria": "Mexicano",
+    "alergenos": ["Gluten"],
+    "precio": 13.00,
+    "descripcion": "Tacos suaves de pollo con guacamole, salsa picante y cebolla.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Sopa de Cebolla": {
+    "categoria": "Sopas",
+    "alergenos": ["Gluten", "Lácteos"],
+    "precio": 9.00,
+    "descripcion": "Sopa caliente de cebolla caramelizada con queso gratinado.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Tarta de Queso": {
+    "categoria": "Postres",
+    "alergenos": ["Lácteos", "Huevo"],
+    "precio": 7.00,
+    "descripcion": "Clásica tarta de queso con base de galleta y coulis de frambuesa.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Jugo de Naranja Natural": {
+    "categoria": "Bebidas",
+    "alergenos": ["Ninguno"],
+    "precio": 5.00,
+    "descripcion": "Jugo recién exprimido de naranjas frescas.",
+    "sin_gluten": true,
+    "vegan": true
+  },
+  "Café Espresso": {
+    "categoria": "Bebidas",
+    "alergenos": ["Ninguno"],
+    "precio": 3.00,
+    "descripcion": "Café espresso de grano molido recién hecho.",
+    "sin_gluten": true,
+    "vegan": true
+  },
+  "Sushi Roll Especial": {
+    "categoria": "Sushi",
+    "alergenos": ["Pescado", "Soya", "Gluten"],
+    "precio": 20.00,
+    "descripcion": "Rollos de sushi con salmón, aguacate, pepino y salsa de soja.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Hamburguesa Vegana": {
+    "categoria": "Vegana",
+    "alergenos": ["Gluten"],
+    "precio": 14.00,
+    "descripcion": "Hamburguesa vegana con proteína de soja, lechuga, tomate y mayonesa vegana.",
+    "sin_gluten": false,
+    "vegan": true
+  },
+  "Pizza Infantil": {
+    "categoria": "Menú Infantil",
+    "alergenos": ["Gluten", "Lácteos"],
+    "precio": 8.00,
+    "descripcion": "Mini pizza con salsa de tomate, mozzarella y orégano.",
+    "sin_gluten": true,
+    "vegan": false
+  },
+  "Nuggets de Pollo": {
+    "categoria": "Menú Infantil",
+    "alergenos": ["Gluten", "Huevo"],
+    "precio": 7.00,
+    "descripcion": "Nuggets de pollo empanados acompañados de salsa de barbacoa.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Pasta con Salsa de Tomate": {
+    "categoria": "Menú Infantil",
+    "alergenos": ["Gluten"],
+    "precio": 6.00,
+    "descripcion": "Pasta pequeña con salsa de tomate natural y queso rallado.",
+    "sin_gluten": false,
+    "vegan": false
+  },
+  "Helado de Vainilla": {
+    "categoria": "Menú Infantil",
+    "alergenos": ["Lácteos"],
+    "precio": 4.00,
+    "descripcion": "Helado cremoso de vainilla, servido con sirope de chocolate.",
+    "sin_gluten": true,
+    "vegan": false
+  },
+  "Limonada Natural": {
+    "categoria": "Menú Infantil",
+    "alergenos": ["Ninguno"],
+    "precio": 3.00,
+    "descripcion": "Limonada casera con agua, limón fresco y un toque de azúcar.",
+    "sin_gluten": true,
+    "vegan": true
+    
+  },
+  "Brownie de Chocolate": {
+  "categoria": "Postres",
+  "alergenos": ["Gluten", "Lácteos", "Nueces"],
+  "precio": 6.00,
+  "descripcion": "Brownie de chocolate casero con nueces y sirope de chocolate.",
+  "sin_gluten": false,
+  "vegan": false
+  }
+}

reservas.json

@@ -0,0 +1,44 @@
+[
+  {
+    "nombre": "Juan Fernandez",
+    "dia": "12/02/2025",
+    "hora": "21:00",
+    "personas": "10",
+    "telefono": "642128722"
+  },
+  {
+    "nombre": "Manuel",
+    "dia": "20/05/2025",
+    "hora": "20:05",
+    "personas": "5",
+    "telefono": "No proporcionado"
+  },
+  {
+    "nombre": "Pedro",
+    "dia": "20/05/2025",
+    "hora": "20:30",
+    "personas": "5",
+    "telefono": "No proporcionado"
+  },
+  {
+    "nombre": "David",
+    "dia": "20/03/2025",
+    "hora": "20:34",
+    "personas": "8",
+    "telefono": "No proporcionado"
+  },
+  {
+    "nombre": "David",
+    "dia": "20/03/2025",
+    "hora": "20:35",
+    "personas": "5",
+    "telefono": "No proporcionado"
+  },
+  {
+    "nombre": "David",
+    "dia": "20/05/2025",
+    "hora": "20:35",
+    "personas": "5",
+    "telefono": "No proporcionado"
+  }
+]