Update app.py

app.py

@@ -174,15 +174,16 @@ def main():
                 for i, source in enumerate(st.session_state['sources']):
                     st.text_area(f"Source {i + 1} - Page {source['page']}", value=source["content"], height=100)
 
-        message = st.text_input("Ask a question", key="message")
+        with st.form(key="question_form"):
+            message = st.text_input("Ask a question", key="message")
+            submit_button = st.form_submit_button(label="Submit")
 
-        if st.button("Submit"):
+        if submit_button:
             with st.spinner("Generating response..."):
-                qa_chain, chat_history, response_answer, sources = conversation(st.session_state['qa_chain'], st.session_state['message'], st.session_state['chat_history'])
+                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
-                st.session_state['message'] = ""
 
 if __name__ == "__main__":
     main()