add: task confirmation button

app.py

@@ -19,39 +19,49 @@ OPTION1="Chat with a file 💬📖"
 OPTION2="Text summarization 🔎"
 OPTION_N="Other 🤔"
 
-option = st.radio(
-    "Please select a task 🤖",
-    options=[OPTION1, OPTION2, OPTION_N],
-                      )
-
-if option == OPTION1:
-    from qa import qa_main 
-    with st.container(): 
-        qa_main()
-
-elif option == OPTION2:
-    from summarization import summarization_main
-    with st.container():
-        summarization_main()
-
-elif option == OPTION_N:
-    from mailing import mailing_main
-    with st.container():
-        user_suggestion = st.text_input(
-            "What other task would you like to perform?", 
-            placeholder="Transform meeting transcripts into rainbow-colored unicorns"
+col1, col2 = st.columns(2)
+with col1:
+    option = st.radio(
+        "Please select a task 🤖",
+        options=[OPTION1, OPTION2, OPTION_N],
+        key="task_selection"
         )
-        if user_suggestion:
-            st.write("""
-            Thanks for contributing with your suggestion! We are carefully reviewing every suggestion. 
-            If you wish to further discuss your task suggestion, consider reaching out to francesco.capuano@bain.com.
-            We will get back to you as soon as possible!
-            """)
-
-        mailing_main(
-            subject="**NEW TASK SUGGESTION** - Automatic email.",
-            body=f"User suggestion\n: {user_suggestion}",
-            to_address="francesco.capuano@bain.com"
-        )
-        st.stop()
+
+with col2: 
+    confirm = st.button("Confirm", key="task_selection_confirm")
+    if confirm: 
+        st.session_state["task_confirmed"] = True
+
+if st.session_state.get("task_confirmed"):
+    # only execute the actual app code when the user confirms the task selection
+    if st.session_state.get("task_selection") == OPTION1:
+        from qa import qa_main 
+        with st.container(): 
+            qa_main()
+
+    elif st.session_state.get("task_selection") == OPTION2:
+        from summarization import summarization_main
+        with st.container():
+            summarization_main()
+
+    elif st.session_state.get("task_selection") == OPTION_N:
+        from mailing import mailing_main
+        with st.container():
+            user_suggestion = st.text_input(
+                "What other task would you like to perform?", 
+                placeholder="Transform meeting transcripts into rainbow-colored unicorns"
+            )
+            if user_suggestion:
+                st.write("""
+                Thanks for contributing with your suggestion! We are carefully reviewing every suggestion. 
+                If you wish to further discuss your task suggestion, consider reaching out to francesco.capuano@bain.com.
+                We will get back to you as soon as possible!
+                """)
+
+            mailing_main(
+                subject="**NEW TASK SUGGESTION** - Automatic email.",
+                body=f"User suggestion\n: {user_suggestion}",
+                to_address="francesco.capuano@bain.com"
+            )
+            st.stop()