Update app.py

app.py

@@ -116,14 +116,15 @@ def reset_selections():
 if menu == "Poll":
     st.title("Breakfast Poll Application")
 
-    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.step = 2
-
-    if st.session_state.step == 2:
+    # Step 1: User's Name
+    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.step = 2  # Set the next step to be visible
+
+    # Show Step 2 only if Step 1 is completed
+    if st.session_state.step >= 2:
         st.header("Step 2: Select your drink(s)")
         drinks_options = [
             "Café con leche", "Colacao", "Descafeinado con leche", "Cortado", 
@@ -131,24 +132,27 @@ if menu == "Poll":
             "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
+            st.session_state.step = 3  # Set the next step to be visible
 
-    if st.session_state.step == 3:
+    # Show Step 3 only if Step 2 is completed
+    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)
+
         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
+            st.session_state.step = 1  # Reset to step 1 for the next user
 
 # "Current" view to display the current summary of all users' selections and submit to history
 elif menu == "Current":