Update app.py

app.py

@@ -178,12 +178,19 @@ def main():
             message = st.text_input("Ask a question", key="message")
             submit_button = st.form_submit_button(label="Submit")
 
-        if submit_button:
-            with st.spinner("Generating response..."):
-                qa_chain, chat_history, response_answer, sources = conversation(st.session_state['qa_chain'], message, st.session_state['chat_history'])
-                st.session_state['qa_chain'] = qa_chain
-                st.session_state['chat_history'] = chat_history
-                st.session_state['sources'] = sources
+            if submit_button:
+                with st.spinner("Generating response..."):
+                    # Call the conversation function inside the form
+                    qa_chain, chat_history, response_answer, sources = conversation(st.session_state['qa_chain'], message, st.session_state['chat_history'])
+                    st.session_state['qa_chain'] = qa_chain
+                    st.session_state['chat_history'] = chat_history
+                    st.session_state['sources'] = sources
+
+                    # Clear the text input after submission
+                    st.session_state.message = ""
+                    st.experimental_rerun()  # Force a rerun to update the UI
 
 if __name__ == "__main__":
     main()
+
+