Update app.py

app.py

@@ -116,6 +116,7 @@ def reset_selections():
 if menu == "Poll":
     st.title("Breakfast Poll Application")
 
+    # Step 1: User's Name
     if st.session_state.step == 1:
         st.header("Step 1: Enter your name")
         name = st.text_input("Name:")
@@ -123,6 +124,7 @@ if menu == "Poll":
             st.session_state.users.append(name)
             st.session_state.step = 2
 
+    # Step 2: Select Drinks
     if st.session_state.step == 2:
         st.header("Step 2: Select your drink(s)")
         drinks_options = [
@@ -130,18 +132,27 @@ if menu == "Poll":
             "Aguasusia", "Aguasusia susia", "Café descafeinado con leche desnatada", 
             "Italiano", "Café con soja", "Té", "Manzanilla", "Nada"
         ]
-        selected_drinks = st.multiselect("Choose your drinks:", drinks_options)
+
+        # Wrap the multiselect dropdown in an expander for collapsible behavior
+        with st.expander("Choose your drinks:", expanded=True):
+            selected_drinks = st.multiselect("", drinks_options)
+
         if st.button("Next", key="step2_next") and selected_drinks:
             st.session_state.current_selections.append({"Name": st.session_state.users[-1], "Drinks": selected_drinks})
             st.session_state.step = 3
 
+    # Step 3: Select Food
     if st.session_state.step == 3:
         st.header("Step 3: Select your food(s)")
         food_options = [
             "Barrita con aceite", "Barrita con tomate", "Palmera de chocolate", 
             "Palmera de chocolate blanco", "Yogurt", "Pincho de tortilla", "Nada"
         ]
-        selected_food = st.multiselect("Choose your food:", food_options)
+
+        # Wrap the multiselect dropdown in an expander for collapsible behavior
+        with st.expander("Choose your food:", expanded=True):
+            selected_food = st.multiselect("", food_options)
+
         if st.button("Save Selections", key="save_selections") and selected_food:
             st.session_state.current_selections[-1]["Food"] = selected_food
             df = pd.DataFrame(st.session_state.current_selections)