Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,7 @@ embed_model = HuggingFaceEmbeddings(model_name="mixedbread-ai/mxbai-embed-large-
|
|
| 38 |
|
| 39 |
# Crea un objeto Chroma con el nombre de la colección
|
| 40 |
vectorstore = Chroma(
|
| 41 |
-
collection_name="
|
| 42 |
embedding_function=embed_model,
|
| 43 |
)
|
| 44 |
|
|
@@ -49,17 +49,27 @@ vectorstore.add_texts(context_data)
|
|
| 49 |
retriever = vectorstore.as_retriever()
|
| 50 |
|
| 51 |
# Crea un objeto PromptTemplate con el prompt
|
| 52 |
-
template = ("""
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
y
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
|
| 64 |
Context: {context}
|
| 65 |
Question: {question}
|
|
@@ -94,12 +104,12 @@ def rag_memory_stream(message, history):
|
|
| 94 |
|
| 95 |
# Crea un objeto Gradio con la función y el título
|
| 96 |
examples = [
|
| 97 |
-
"¿
|
| 98 |
-
"¿
|
| 99 |
-
"¿
|
| 100 |
]
|
| 101 |
-
description = "
|
| 102 |
-
title = "Experto en
|
| 103 |
demo = gr.ChatInterface(fn=rag_memory_stream,
|
| 104 |
type="messages",
|
| 105 |
title=title,
|
|
|
|
| 38 |
|
| 39 |
# Crea un objeto Chroma con el nombre de la colección
|
| 40 |
vectorstore = Chroma(
|
| 41 |
+
collection_name="unipapel_papeleria",
|
| 42 |
embedding_function=embed_model,
|
| 43 |
)
|
| 44 |
|
|
|
|
| 49 |
retriever = vectorstore.as_retriever()
|
| 50 |
|
| 51 |
# Crea un objeto PromptTemplate con el prompt
|
| 52 |
+
template = ("""Eres un asistente experto de unipapel.es, especializado en temas relacionados con la papelería. Tu función es ofrecer orientación a los clientes que realicen preguntas. Si no encuentras una respuesta, invítalos a contactar con nosotros. Puedes proporcionar información sobre nuestras categorías de productos según lo que te soliciten. La URL de la tienda es www.unipapel.es. A continuación, algunas de nuestras categorías:
|
| 53 |
+
|
| 54 |
+
Accesorios para pizarras
|
| 55 |
+
|
| 56 |
+
Acuarelas y lápices acuarelables
|
| 57 |
+
|
| 58 |
+
Archivadores de palanca
|
| 59 |
+
|
| 60 |
+
Blocs de dibujo
|
| 61 |
+
|
| 62 |
+
Bolígrafos
|
| 63 |
+
|
| 64 |
+
Carpetas clasificadoras
|
| 65 |
+
|
| 66 |
+
Cuadernos de caligrafía
|
| 67 |
+
|
| 68 |
+
Papelería escolar
|
| 69 |
+
|
| 70 |
+
Pegamentos y adhesivos
|
| 71 |
+
|
| 72 |
+
Pizarras blancas
|
| 73 |
|
| 74 |
Context: {context}
|
| 75 |
Question: {question}
|
|
|
|
| 104 |
|
| 105 |
# Crea un objeto Gradio con la función y el título
|
| 106 |
examples = [
|
| 107 |
+
"¿Qué tipo de bolígrafos son ideales para tomar apuntes en clase?",
|
| 108 |
+
"¿Cuáles son los mejores tipos de papel para imprimir documentos profesionales?",
|
| 109 |
+
"¿Qué materiales de oficina son esenciales para un espacio de trabajo productivo?"
|
| 110 |
]
|
| 111 |
+
description = "Chatbot con Unipapel"
|
| 112 |
+
title = "Experto en papeleria y todo lo referente"
|
| 113 |
demo = gr.ChatInterface(fn=rag_memory_stream,
|
| 114 |
type="messages",
|
| 115 |
title=title,
|