Update app.py

app.py

@@ -122,6 +122,7 @@ if menu == "Poll":
         name = st.text_input("Name:")
         if st.button("Next", key="step1_next") and name:
             st.session_state.users.append(name)
+            st.session_state.show_drinks_dropdown = True  # Initialize the dropdown state for Step 2
             st.session_state.step = 2
 
     # Step 2: Select Drinks
@@ -133,12 +134,19 @@ if menu == "Poll":
             "Italiano", "Café con soja", "Té", "Manzanilla", "Nada"
         ]
 
-        # Wrap the multiselect dropdown in an expander for collapsible behavior
-        with st.expander("Choose your drinks:", expanded=True):
-            selected_drinks = st.multiselect("", drinks_options)
+        # Control visibility of the dropdown using session state
+        if st.session_state.get("show_drinks_dropdown", True):
+            selected_drinks = st.multiselect("Choose your drinks:", drinks_options)
+        else:
+            st.write("Drinks selected successfully!")
 
-        if st.button("Next", key="step2_next") and selected_drinks:
+        if st.session_state.get("show_drinks_dropdown", True) and selected_drinks:
+            # If drinks are selected, hide the dropdown and show confirmation text
+            st.session_state.show_drinks_dropdown = False
+
+        if st.button("Next", key="step2_next") and not st.session_state.get("show_drinks_dropdown", True):
             st.session_state.current_selections.append({"Name": st.session_state.users[-1], "Drinks": selected_drinks})
+            st.session_state.show_food_dropdown = True  # Initialize the dropdown state for Step 3
             st.session_state.step = 3
 
     # Step 3: Select Food
@@ -149,11 +157,17 @@ if menu == "Poll":
             "Palmera de chocolate blanco", "Yogurt", "Pincho de tortilla", "Nada"
         ]
 
-        # Wrap the multiselect dropdown in an expander for collapsible behavior
-        with st.expander("Choose your food:", expanded=True):
-            selected_food = st.multiselect("", food_options)
+        # Control visibility of the dropdown using session state
+        if st.session_state.get("show_food_dropdown", True):
+            selected_food = st.multiselect("Choose your food:", food_options)
+        else:
+            st.write("Food selected successfully!")
+
+        if st.session_state.get("show_food_dropdown", True) and selected_food:
+            # If food is selected, hide the dropdown and show confirmation text
+            st.session_state.show_food_dropdown = False
 
-        if st.button("Save Selections", key="save_selections") and selected_food:
+        if st.button("Save Selections", key="save_selections") and not st.session_state.get("show_food_dropdown", True):
             st.session_state.current_selections[-1]["Food"] = selected_food
             df = pd.DataFrame(st.session_state.current_selections)
             save_current_selection_to_file(df)