segoedu commited on
Commit
34af858
1 Parent(s): bbb03b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -5
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
- #return_source_documents=True,
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(respuesta)}]")
58
 
59
  # Guardar pregunta y respuesta en los logs
60
- logs.append({"Pregunta": user_question, "Respuesta": 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)