Update app.py

app.py

@@ -278,8 +278,13 @@ def app():
         # 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):
+        # Components for pages
+        signup_page = gr.Column(visible=True)
+        login_page = gr.Column(visible=False)
+        menu_page = gr.Column(visible=False)
+
+        # Signup Page Logic
+        with signup_page:
             gr.Markdown("### Signup Page")
             name = gr.Textbox(label="Name")
             phone = gr.Textbox(label="Phone Number")
@@ -291,16 +296,19 @@ def app():
             # Signup callback
             def handle_signup(name, phone, email, password):
                 message, next_page = signup_user(name, phone, email, password)
-                return message, next_page
+                # Update visibility for redirection
+                if next_page == "login":
+                    return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), message
+                return gr.update(), gr.update(), gr.update(), message
 
             signup_btn.click(
                 handle_signup,
                 inputs=[name, phone, email, password],
-                outputs=[signup_message, current_page],
+                outputs=[signup_page, login_page, menu_page, signup_message],
             )
 
-        # Login Page
-        with gr.Row(visible=lambda state: state == "login", state=current_page):
+        # Login Page Logic
+        with login_page:
             gr.Markdown("### Login Page")
             login_email = gr.Textbox(label="Email")
             login_password = gr.Textbox(label="Password", type="password")
@@ -310,16 +318,19 @@ def app():
             # Login callback
             def handle_login(email, password):
                 message, next_page = login_user(email, password)
-                return message, next_page
+                # Update visibility for redirection
+                if next_page == "menu":
+                    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), message
+                return gr.update(), gr.update(), gr.update(), message
 
             login_btn.click(
                 handle_login,
                 inputs=[login_email, login_password],
-                outputs=[login_message, current_page],
+                outputs=[signup_page, login_page, menu_page, login_message],
             )
 
-        # Menu Page
-        with gr.Row(visible=lambda state: state == "menu", state=current_page):
+        # Menu Page Logic
+        with menu_page:
             gr.Markdown("### Menu Page (Accessible Only After Login)")
 
             # Radio button for selecting preference