Update app.py

app.py

@@ -271,13 +271,12 @@ modal_and_cart_js = """
 
 
 # Gradio app
-def app():
+ef app():
     with gr.Blocks() as demo:
         gr.Markdown("## Secure Food Ordering System")
 
-        # State variables to track redirection
-        with gr.State() as current_page:  # Holds the current visible page
-            current_page.set("signup")  # Default to signup page
+        # State variable to track the current page
+        current_page = gr.State("signup")  # Default page is Signup
 
         # Signup Page
         with gr.Row(visible=lambda state: state == "signup", state=current_page):
@@ -289,17 +288,15 @@ def app():
             signup_btn = gr.Button("Signup")
             signup_message = gr.Label()
 
-            # Handle Signup
+            # Signup callback
             def handle_signup(name, phone, email, password):
-                success, message = signup_user(name, phone, email, password)
-                if success:
-                    current_page.set("login")  # Redirect to login page
-                return message
+                message, next_page = signup_user(name, phone, email, password)
+                return message, next_page
 
             signup_btn.click(
                 handle_signup,
                 inputs=[name, phone, email, password],
-                outputs=[signup_message],
+                outputs=[signup_message, current_page],
             )
 
         # Login Page
@@ -310,17 +307,15 @@ def app():
             login_btn = gr.Button("Login")
             login_message = gr.Label()
 
-            # Handle Login
+            # Login callback
             def handle_login(email, password):
-                success, message = login_user(email, password)
-                if success:
-                    current_page.set("menu")  # Redirect to menu page
-                return message
+                message, next_page = login_user(email, password)
+                return message, next_page
 
             login_btn.click(
                 handle_login,
                 inputs=[login_email, login_password],
-                outputs=[login_message],
+                outputs=[login_message, current_page],
             )
 
         # Menu Page