Update app.py (colores y sidebar)

app.py

@@ -35,7 +35,7 @@ modelo_llm = st.selectbox('Modelo de lenguaje', list(modelos_llm))
 
 
 def save_to_file():
-    with open("./historial.txt", "a", encoding="utf-8") as archivo:
+    with open("historial.txt", "a", encoding="utf-8") as archivo:
         # Añadir la fecha y hora actual
         archivo.write("-" * 25 )
         fecha_hora_actual = datetime.now().strftime("%Y-%m-%d %H:%M")
@@ -68,6 +68,22 @@ def create_embeddings(pdf):
     return knowledge_base
 
 
+# Función para mostrar logs
+def mostrar_logs(logs,hints):
+    st.subheader("Chunks")
+    for hint in hints:
+        st.write(hint.page_content)
+        st.write("-" * 30)
+
+    st.sidebar.header("Registro de preguntas")
+    for entry in logs:
+        st.sidebar.write(f"**Pregunta: {entry['Pregunta']}**")
+        st.sidebar.write(f"Respuesta: {entry['Respuesta']}")
+
+
+# Lista para almacenar preguntas y respuestas
+logs = []
+
 if pdf_obj:
     file_name = pdf_obj.name
     knowledge_base = create_embeddings(pdf_obj)
@@ -82,7 +98,15 @@ if pdf_obj:
         chain = load_qa_chain(llm, chain_type="stuff")
         respuesta = chain.run(input_documents=docs, question=user_question)
 
-        st.write(respuesta)
+        # Mostrar la variable en color verde
+        st.subheader("Respuesta")
+        st.write(f":green[{str(respuesta)}]")
+
+        # Guardar pregunta y respuesta en los logs
+        logs.append({"Pregunta": user_question, "Respuesta": respuesta})
+
+        # Mostrar logs actualizados
+        mostrar_logs(logs,docs)
 
         # Guarda la consulta en un archivo
-        save_to_file()
+        save_to_file()