Update app.py

app.py

@@ -122,12 +122,12 @@ 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.selected_drinks = []  # Initialize an empty list to store selected drinks
+            st.session_state.selected_food = []  # Initialize an empty list to store selected food
             st.session_state.step = 2
 
     # Step 2: Select Drinks
-    if st.session_state.step == 2:
+    elif st.session_state.step == 2:
         st.header("Step 2: Select your drink(s)")
         drinks_options = [
             "Café con leche", "Colacao", "Descafeinado con leche", "Cortado", 
@@ -135,45 +135,46 @@ if menu == "Poll":
             "Italiano", "Café con soja", "Té", "Manzanilla", "Nada"
         ]
 
-        # Display the dropdown if `show_drinks_dropdown` is True
-        if st.session_state.get("show_drinks_dropdown", True):
-            st.session_state.selected_drinks = st.multiselect("Choose your drinks:", drinks_options)
+        # Use st.multiselect to select drinks
+        selected_drinks = st.multiselect("Choose your drinks:", drinks_options, key="drinks_multiselect")
 
-        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
+        # Collapse the dropdown after selection by storing the value in session state
+        if selected_drinks:
+            st.session_state.selected_drinks = selected_drinks
+            st.experimental_rerun()  # Force a rerun to collapse the dropdown
 
-        if st.button("Next", key="step2_next") and not st.session_state.get("show_drinks_dropdown", True):
-            # 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
+        # Show the "Next" button only if there are selected drinks
+        if st.session_state.selected_drinks:
+            if st.button("Next", key="step2_next"):
+                st.session_state.step = 3
 
     # Step 3: Select Food
-    if st.session_state.step == 3:
+    elif 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"
         ]
 
-        # Display the dropdown if `show_food_dropdown` is True
-        if st.session_state.get("show_food_dropdown", True):
-            st.session_state.selected_food = st.multiselect("Choose your food:", food_options)
-
-        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):
-            # 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()
-            st.success(f"Selections saved for {st.session_state.users[-1]}!")
-            st.session_state.step = 1
+        # Use st.multiselect to select food
+        selected_food = st.multiselect("Choose your food:", food_options, key="food_multiselect")
+
+        # Collapse the dropdown after selection by storing the value in session state
+        if selected_food:
+            st.session_state.selected_food = selected_food
+            st.experimental_rerun()  # Force a rerun to collapse the dropdown
+
+        # Show the "Save Selections" button only if there is selected food
+        if st.session_state.selected_food:
+            if st.button("Save Selections", key="save_selections"):
+                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()
+                st.success(f"Selections saved for {st.session_state.users[-1]}!")
+                st.session_state.step = 1  # Reset to step 1 for the next user
+                st.session_state.selected_drinks = []  # Reset the drinks selection
+                st.session_state.selected_food = []  # Reset the food selection
 
 # "Current" view to display the current summary of all users' selections and submit to history
 elif menu == "Current":