Update app.py

app.py

@@ -116,66 +116,39 @@ 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:")
         if st.button("Next", key="step1_next") and name:
             st.session_state.users.append(name)
-            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
-    elif st.session_state.step == 2:
+    if st.session_state.step == 2:
         st.header("Step 2: Select your drink(s)")
         drinks_options = [
             "Café con leche", "Colacao", "Descafeinado con leche", "Cortado", 
             "Aguasusia", "Aguasusia susia", "Café descafeinado con leche desnatada", 
             "Italiano", "Café con soja", "Té", "Manzanilla", "Nada"
         ]
+        selected_drinks = st.multiselect("Choose your drinks:", 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
 
-        # Use st.multiselect to select drinks
-        selected_drinks = st.multiselect("Choose your drinks:", drinks_options, key="drinks_multiselect")
-
-        # 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
-
-        # 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
-    elif st.session_state.step == 3:
+    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"
         ]
-
-        # 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
-
+        selected_food = st.multiselect("Choose your food:", 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)
+            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
 # "Current" view to display the current summary of all users' selections and submit to history
 elif menu == "Current":
     st.title("Current Selections of All Users")