Update app.py

app.py

@@ -176,22 +176,20 @@ def main():
                     st.text_area(f"Source {i + 1} - Page {source['page']}", value=source["content"], height=100)
 
         with st.form(key="question_form"):
-            message = st.text_input("Ask a question", key="message_input")
+            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..."):
-                    # Process the form input
-                    qa_chain, chat_history, response_answer, sources = conversation(
-                        st.session_state['qa_chain'], message, st.session_state['chat_history']
-                    )
+                    # 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 form input field by simply not including the key update
-                    # The form input should be handled directly in the form submit button
-                    st.rerun()
+                    # Clear the text input after submission
+                    st.session_state.message = ""
+                    st.experimental_rerun()  # Force a rerun to update the UI
 
 if __name__ == "__main__":
     main()