Update app.py
Browse files
app.py
CHANGED
@@ -31,12 +31,18 @@ qa = RetrievalQA.from_chain_type(
|
|
31 |
llm=llm,
|
32 |
chain_type="stuff",
|
33 |
retriever=vectorstore.as_retriever(),
|
34 |
-
|
35 |
#verbose=True
|
36 |
)
|
37 |
|
38 |
# Función para mostrar logs
|
39 |
-
def mostrar_logs(logs):
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
st.sidebar.header("Registro de preguntas")
|
41 |
for entry in logs:
|
42 |
st.sidebar.write(f"**Pregunta: {entry['Pregunta']}**")
|
@@ -51,13 +57,15 @@ if user_question:
|
|
51 |
|
52 |
# Obtenemos la respuesta
|
53 |
respuesta = qa.invoke(user_question)
|
|
|
|
|
54 |
|
55 |
# Mostrar la variable en color verde
|
56 |
st.subheader("Respuesta")
|
57 |
-
st.write(f":green[{str(
|
58 |
|
59 |
# Guardar pregunta y respuesta en los logs
|
60 |
-
logs.append({"Pregunta": user_question, "Respuesta":
|
61 |
|
62 |
# Mostrar logs actualizados
|
63 |
-
mostrar_logs(logs)
|
|
|
31 |
llm=llm,
|
32 |
chain_type="stuff",
|
33 |
retriever=vectorstore.as_retriever(),
|
34 |
+
return_source_documents=True,
|
35 |
#verbose=True
|
36 |
)
|
37 |
|
38 |
# Función para mostrar logs
|
39 |
+
def mostrar_logs(logs,hints):
|
40 |
+
# Crear un contenedor desplegable
|
41 |
+
with st.expander("Chunks"):
|
42 |
+
for hint in hints:
|
43 |
+
st.write(hint.page_content)
|
44 |
+
st.write("-" * 30)
|
45 |
+
|
46 |
st.sidebar.header("Registro de preguntas")
|
47 |
for entry in logs:
|
48 |
st.sidebar.write(f"**Pregunta: {entry['Pregunta']}**")
|
|
|
57 |
|
58 |
# Obtenemos la respuesta
|
59 |
respuesta = qa.invoke(user_question)
|
60 |
+
docs = respuesta['source_documents']
|
61 |
+
result = respuesta['result']
|
62 |
|
63 |
# Mostrar la variable en color verde
|
64 |
st.subheader("Respuesta")
|
65 |
+
st.write(f":green[{str(result)}]")
|
66 |
|
67 |
# Guardar pregunta y respuesta en los logs
|
68 |
+
logs.append({"Pregunta": user_question, "Respuesta": result})
|
69 |
|
70 |
# Mostrar logs actualizados
|
71 |
+
mostrar_logs(logs,docs)
|