use `on_change` to pass history on retrieve_answer()

app.py

@@ -88,14 +88,20 @@ def retrieve_document(query_input):
     related_doc = vector_database.similarity_search(query_input)
     return related_doc
 
-def retrieve_answer(query_input):
-    prompt_answer=  query_input + " " + "Try to elaborate as much as you can."
+def retrieve_answer():
+    prompt_answer=  st.session_state.my_text_input + " " + "Try to elaborate as much as you can."
     answer = qa_retriever.run(prompt_answer)
-    # output = st.text_area(label="Retrieved documents", value=answer[6:]) #this positional slicing helps remove "<pad> " at the beginning
-    
-    # st.markdown('---')
-    # score = st.radio(label = 'please select the rating score for overall satifaction and helpfullness of the bot answer', options=[0, 1,2,3,4,5], horizontal=True,
-    #                  on_change=update_worksheet_qa, key='rating')
+    log = {"timestamp": datetime.datetime.now(),
+        "question":st.session_state.my_text_input,
+        "generated_answer": answer[6:],
+        "rating":0 }
+
+    st.session_state.history.append(log)
+    update_worksheet_qa()
+    st.session_state.chat_history.append({"message": st.session_state.my_text_input, "is_user": True})
+    st.session_state.chat_history.append({"message": answer[6:] , "is_user": False})
+
+    st.session_state.my_text_input = ""
 
     return answer[6:] #this positional slicing helps remove "<pad> " at the beginning
     
@@ -187,22 +193,22 @@ st.write("""
 for chat in st.session_state.chat_history:
     st_message(**chat)
 
-query_input = st.text_input(label= 'What would you like to know about AIT?' , key = 'my_text_input')
-generate_button = st.button(label = 'Ask question!')
+query_input = st.text_input(label= 'What would you like to know about AIT?' , key = 'my_text_input', on_change= retrieve_answer )
+# generate_button = st.button(label = 'Ask question!')
 
-if generate_button:
-    answer = retrieve_answer(query_input)
-    log = {"timestamp": datetime.datetime.now(),
-        "question":query_input,
-        "generated_answer": answer,
-        "rating":0 }
+# if generate_button:
+#     answer = retrieve_answer(query_input)
+#     log = {"timestamp": datetime.datetime.now(),
+#         "question":query_input,
+#         "generated_answer": answer,
+#         "rating":0 }
 
-    st.session_state.history.append(log)
-    update_worksheet_qa()
-    st.session_state.chat_history.append({"message": query_input, "is_user": True})
-    st.session_state.chat_history.append({"message": answer, "is_user": False})
+#     st.session_state.history.append(log)
+#     update_worksheet_qa()
+#     st.session_state.chat_history.append({"message": query_input, "is_user": True})
+#     st.session_state.chat_history.append({"message": answer, "is_user": False})
 
-    print(st.session_state.chat_history)
+#     print(st.session_state.chat_history)
 
 
 clear_button = st.button("Start new convo",