better version 03

app.py

@@ -66,7 +66,11 @@ step01 = "Step 01: Upload Files"
 step02 = "Step 02: Edit Knowledge Base"
 step03 = "Step 03: Split text"
 tabs = [step01, step02, step03]
-selected_tab = st.sidebar.radio("Choose a tab", tabs)
+if "selected_tab" not in st.session_state:
+    st.session_state.selected_tab = step01
+
+selected_tab = st.sidebar.radio(
+    "Choose a tab", tabs, index=tabs.index(st.session_state.selected_tab))
 
 if "text_content" not in st.session_state:
     st.session_state.text_content = ""
@@ -95,11 +99,13 @@ elif selected_tab == step01:
                           for uploaded_file in uploaded_files]
         file_names = [uploaded_file.name for uploaded_file in uploaded_files]
 
-        if st.button('Convert and add to knowledge database'):
+        if st.button('Convert to text'):
             converting_message = st.text("Converting PDFs...")
             converted_text = "\n".join(pdf_to_text(pdf_files_data, file_names))
             st.session_state.text_content += converted_text
             converting_message.empty()
+            st.session_state.selected_tab = step02
+            st.experimental_rerun()
 
 elif selected_tab == step03:
     st.subheader("Splitting Options")