Update app.py

app.py

@@ -116,6 +116,7 @@ def reset_selections():
 if menu == "Poll":
     st.title("Breakfast Poll Application")
 
+    # Step 1: Enter User's Name
     if st.session_state.step == 1:
         st.header("Step 1: Enter your name")
         name = st.text_input("Name:")
@@ -123,7 +124,8 @@ if menu == "Poll":
             st.session_state.users.append(name)
             st.session_state.step = 2
 
-    if st.session_state.step == 2:
+    # Step 2: Select Drinks
+    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,7 +137,8 @@ if menu == "Poll":
             st.session_state.current_selections.append({"Name": st.session_state.users[-1], "Drinks": selected_drinks})
             st.session_state.step = 3
 
-    if st.session_state.step == 3:
+    # Step 3: Select Food
+    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", 
@@ -149,6 +152,7 @@ if menu == "Poll":
             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")