Update app.py

app.py

@@ -123,6 +123,7 @@ if menu == "Poll":
         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.selected_drinks = []  # Initialize an empty list to store selected drinks
             st.session_state.step = 2
 
     # Step 2: Select Drinks
@@ -134,19 +135,19 @@ if menu == "Poll":
             "Italiano", "Café con soja", "Té", "Manzanilla", "Nada"
         ]
 
-        # Control visibility of the dropdown using session state
+        # Display the dropdown if `show_drinks_dropdown` is True
         if st.session_state.get("show_drinks_dropdown", True):
-            selected_drinks = st.multiselect("Choose your drinks:", drinks_options)
-        else:
-            st.write("Drinks selected successfully!")
+            st.session_state.selected_drinks = st.multiselect("Choose your drinks:", drinks_options)
 
-        if st.session_state.get("show_drinks_dropdown", True) and selected_drinks:
+        if st.session_state.get("show_drinks_dropdown", True) and st.session_state.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})
+            # Use the session state to access the selected drinks
+            st.session_state.current_selections.append({"Name": st.session_state.users[-1], "Drinks": st.session_state.selected_drinks})
             st.session_state.show_food_dropdown = True  # Initialize the dropdown state for Step 3
+            st.session_state.selected_food = []  # Initialize an empty list to store selected food
             st.session_state.step = 3
 
     # Step 3: Select Food
@@ -157,18 +158,17 @@ if menu == "Poll":
             "Palmera de chocolate blanco", "Yogurt", "Pincho de tortilla", "Nada"
         ]
 
-        # Control visibility of the dropdown using session state
+        # Display the dropdown if `show_food_dropdown` is True
         if st.session_state.get("show_food_dropdown", True):
-            selected_food = st.multiselect("Choose your food:", food_options)
-        else:
-            st.write("Food selected successfully!")
+            st.session_state.selected_food = st.multiselect("Choose your food:", food_options)
 
-        if st.session_state.get("show_food_dropdown", True) and selected_food:
+        if st.session_state.get("show_food_dropdown", True) and st.session_state.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 not st.session_state.get("show_food_dropdown", True):
-            st.session_state.current_selections[-1]["Food"] = selected_food
+            # Use the session state to access the selected food
+            st.session_state.current_selections[-1]["Food"] = st.session_state.selected_food
             df = pd.DataFrame(st.session_state.current_selections)
             save_current_selection_to_file(df)
             upload_temp_file_to_repo()