Update app.py

app.py

@@ -35,9 +35,9 @@ def validate_user(phone, password):
 # Function to handle login
 def login(phone, password):
     if validate_user(phone, password):
-        return f"Welcome! You have successfully logged in."
+        return "Login successful! Redirecting to the main page...", gr.update(visible=True), gr.update(visible=False)
     else:
-        return "Invalid phone number or password. Please try again."
+        return "Invalid phone number or password. Please try again.", gr.update(visible=False), gr.update(visible=True)
 
 # Function to handle signup
 def signup(phone, name, email, password):
@@ -48,21 +48,32 @@ def signup(phone, name, email, password):
             file.write(f"{phone},{name},{email},{password}\n")
         return "Signup successful! You can now login."
 
-# Gradio interfaces
+# Main Page
+def main_page():
+    with gr.Blocks() as main:
+        gr.Markdown("# Welcome to the Main Page")
+        gr.Text("You are successfully logged in!")
+    return main
+
+# Login Interface
 def login_interface():
-    with gr.Blocks() as demo:
+    with gr.Blocks() as login_demo:
         gr.Markdown("# Login")
         phone = gr.Textbox(label="Phone Number")
         password = gr.Textbox(label="Password", type="password")
         output = gr.Textbox(label="Output")
         login_button = gr.Button("Login")
 
-        login_button.click(login, inputs=[phone, password], outputs=output)
+        main_content = main_page()
+        main_content.visible = False
+
+        login_button.click(login, inputs=[phone, password], outputs=[output, main_content, login_demo])
 
-    return demo
+    return login_demo, main_content
 
+# Signup Interface
 def signup_interface():
-    with gr.Blocks() as demo:
+    with gr.Blocks() as signup_demo:
         gr.Markdown("# Signup")
         phone = gr.Textbox(label="Phone Number")
         name = gr.Textbox(label="Name")
@@ -73,17 +84,19 @@ def signup_interface():
 
         signup_button.click(signup, inputs=[phone, name, email, password], outputs=output)
 
-    return demo
+    return signup_demo
 
 # Main interface
 with gr.Blocks() as main_interface:
     gr.Markdown("# Login and Signup System")
     with gr.Tabs():
         with gr.Tab("Login"):
-            login_interface()
+            login_demo, main_content = login_interface()
+            main_interface.append(main_content)
         with gr.Tab("Signup"):
             signup_interface()
 
 if __name__ == "__main__":
     main_interface.launch()
 
+