Update app.py

app.py

@@ -118,6 +118,9 @@ for i, chat in enumerate(st.session_state.history):
     message(chat['question'], is_user=True, key=f"user_{i}")
     message(chat['answer'], key=f"bot_{i}")
 
+if 'something' not in st.session_state:
+    st.session_state.something = ''
+
 user_input = st.text_input("Ask a question:", key="input")
 
 if st.button("Generate"):
@@ -125,4 +128,6 @@ if st.button("Generate"):
         result = st.session_state.qa_chain(user_input)
         answer = result["result"]
         st.session_state.history.append({"question": user_input, "answer": answer})
-        user_input = ''
+        st.session_state.something = st.session_state.input
+        st.session_state.input = ''
+