Update app.py

app.py

@@ -259,28 +259,29 @@ modal_and_cart_js = """
 def app():
     with gr.Blocks() as demo:
         # State to track the current page
-        current_page = gr.State("login")  # Default page is the login page
+        current_page = gr.State("login")  # Default page is login
 
-        # === LOGIN PAGE ===
-        with gr.Row(visible_if=lambda x: x == "login", inputs=[current_page]):
+        # Login Page
+        with gr.Row(visible=lambda x: x == "login", inputs=[current_page]):
             gr.Markdown("### Login Page")
             login_email = gr.Textbox(label="Email")
             login_password = gr.Textbox(label="Password", type="password")
             login_btn = gr.Button("Login")
             login_message = gr.Label()
 
+            # Function to handle login
             def login(email, password):
-                message, page = validate_login(email, password)
-                return message, page  # Redirect based on login status
+                message, next_page = validate_login(email, password)
+                return message, next_page
 
             login_btn.click(
                 login,
                 inputs=[login_email, login_password],
-                outputs=[login_message, current_page],  # Update message and page
+                outputs=[login_message, current_page],  # Update message and redirect to the next page
             )
 
-        # === SIGNUP PAGE ===
-        with gr.Row(visible_if=lambda x: x == "signup", inputs=[current_page]):
+        # Signup Page
+        with gr.Row(visible=lambda x: x == "signup", inputs=[current_page]):
             gr.Markdown("### Signup Page")
             name = gr.Textbox(label="Name")
             phone = gr.Textbox(label="Phone Number")
@@ -289,20 +290,20 @@ def app():
             signup_btn = gr.Button("Signup")
             signup_message = gr.Label()
 
+            # Function to handle signup
             def signup(name, phone, email, password):
-                message, page = signup_user(name, phone, email, password)
-                return message, page  # Redirect to login after signup
+                message, next_page = signup_user(name, phone, email, password)
+                return message, next_page
 
             signup_btn.click(
                 signup,
                 inputs=[name, phone, email, password],
-                outputs=[signup_message, current_page],  # Update message and page
+                outputs=[signup_message, current_page],  # Update message and redirect to the next page
             )
 
-        # === MENU PAGE ===
-        with gr.Row(visible_if=lambda x: x == "menu", inputs=[current_page]):
+        # Menu Page
+        with gr.Row(visible=lambda x: x == "menu", inputs=[current_page]):
             gr.Markdown("### Menu Page")
-
             # Radio button for selecting preference
             selected_preference = gr.Radio(
                 choices=["All", "Vegetarian", "Halal/Non-Veg", "Guilt-Free"],