Update app.py

app.py

@@ -11,8 +11,9 @@ if 'historial' not in st.session_state:
 def get_historial():
     return st.session_state["historial"]
 
-def add_historial(value: str):
-    st.session_state["historial"].append(value)
+def add_historial(respuesta):
+    st.session_state["historial"].append(respuesta["query"])
+    st.session_state["historial"].append(respuesta["result"])
 
 
 os.environ["OPENAI_API_KEY"] = st.secrets['OPENAI_API_KEY']  # agregada en la config de hugginface
@@ -25,9 +26,9 @@ tmp_button = st.button("CLICK")
 chain = get_chain(st.secrets['OPENAI_API_KEY'])
 if tmp_button: #Esperar al boton
     out = chain.invoke(pregunta)
-    print(out)
+    add_historial(out)
     print(get_historial())
-    st.write(f"<p style='text-align: right;style='text-size: 30px; background-color: yellow;color: black;'>{out['query']}</p>", unsafe_allow_html=True)
+    st.write(f"<p style='text-align: right;style='text-size: 30px; background-color: yellow;color: black;'>{out['result']}</p>", unsafe_allow_html=True)
     #st.rerun() #Restart app
 else:
     st.stop()