Update app.py

app.py

@@ -5,20 +5,14 @@ import random
 users = {}
 otp_store = {}
 
-# =========================
-# STATE NAVIGATION
-# =========================
-def go_to(page):
-    return page
-
 # =========================
 # SIGNUP
 # =========================
 def signup(username, password):
     if username in users:
-        return "❌ User already exists", "signup"
+        return "❌ User already exists"
     users[username] = password
-    return "✅ Signup successful! Please login.", "login"
+    return "✅ Signup successful! Go to Login tab."
 
 # =========================
 # LOGIN
@@ -27,16 +21,16 @@ def login(username, password):
     if username in users and users[username] == password:
         otp = random.randint(1000, 9999)
         otp_store[username] = otp
-        return f"OTP: {otp} (demo)", "otp", username
-    return "❌ Invalid credentials", "login", ""
+        return f"OTP generated: {otp} (demo)"
+    return "❌ Invalid credentials"
 
 # =========================
 # OTP VERIFY
 # =========================
 def verify_otp(username, otp_input):
     if username in otp_store and str(otp_store[username]) == str(otp_input):
-        return "✅ Login Successful", "dashboard", True
-    return "❌ Invalid OTP", "otp", False
+        return "✅ Login Successful", True
+    return "❌ Invalid OTP", False
 
 # =========================
 # PREDICT
@@ -60,78 +54,49 @@ def predict(logged_in, *inputs):
 # =========================
 with gr.Blocks() as demo:
 
-    page = gr.State("home")
-    current_user = gr.State("")
     logged_in = gr.State(False)
 
-    # ================= HOME =================
-    with gr.Column(visible=True) as home_page:
-        gr.Markdown("# 🔐 Secure IDS System")
-        login_btn = gr.Button("Login")
-        signup_btn = gr.Button("Signup")
-
-    # ================= SIGNUP =================
-    with gr.Column(visible=False) as signup_page:
-        gr.Markdown("## Signup")
-        su_user = gr.Textbox(label="Username")
-        su_pass = gr.Textbox(label="Password", type="password")
-        su_btn = gr.Button("Create Account")
-        su_out = gr.Textbox()
-
-    # ================= LOGIN =================
-    with gr.Column(visible=False) as login_page:
-        gr.Markdown("## Login")
-        li_user = gr.Textbox(label="Username")
-        li_pass = gr.Textbox(label="Password", type="password")
-        li_btn = gr.Button("Login")
-        li_out = gr.Textbox()
-
-    # ================= OTP =================
-    with gr.Column(visible=False) as otp_page:
-        gr.Markdown("## Enter OTP")
-        otp_input = gr.Textbox()
-        otp_btn = gr.Button("Verify")
-        otp_out = gr.Textbox()
-
-    # ================= DASHBOARD =================
-    with gr.Column(visible=False) as dashboard_page:
-        gr.Markdown("## IDS Dashboard")
-
-        inputs = [gr.Number(label=f"Feature {i+1}") for i in range(10)]
-        pred_btn = gr.Button("Analyze")
-        output = gr.Markdown()
-
-    # ================= NAVIGATION LOGIC =================
-    def render(page_name):
-        return (
-            gr.update(visible=page_name=="home"),
-            gr.update(visible=page_name=="signup"),
-            gr.update(visible=page_name=="login"),
-            gr.update(visible=page_name=="otp"),
-            gr.update(visible=page_name=="dashboard")
-        )
-
-    # Home buttons
-    login_btn.click(lambda: "login", None, page).then(render, page,
-        [home_page, signup_page, login_page, otp_page, dashboard_page])
-
-    signup_btn.click(lambda: "signup", None, page).then(render, page,
-        [home_page, signup_page, login_page, otp_page, dashboard_page])
-
-    # Signup action
-    su_btn.click(signup, [su_user, su_pass], [su_out, page]).then(render, page,
-        [home_page, signup_page, login_page, otp_page, dashboard_page])
-
-    # Login action
-    li_btn.click(login, [li_user, li_pass], [li_out, page, current_user]).then(render, page,
-        [home_page, signup_page, login_page, otp_page, dashboard_page])
-
-    # OTP verify
-    otp_btn.click(verify_otp, [current_user, otp_input],
-                  [otp_out, page, logged_in]).then(render, page,
-        [home_page, signup_page, login_page, otp_page, dashboard_page])
-
-    # Prediction
-    pred_btn.click(predict, [logged_in] + inputs, output)
+    gr.Markdown("# 🔐 Secure IDS System")
+
+    with gr.Tabs():
+
+        # SIGNUP TAB
+        with gr.Tab("Signup"):
+            su_user = gr.Textbox(label="Username")
+            su_pass = gr.Textbox(label="Password", type="password")
+            su_btn = gr.Button("Signup")
+            su_out = gr.Textbox()
+
+            su_btn.click(signup, [su_user, su_pass], su_out)
+
+        # LOGIN TAB
+        with gr.Tab("Login"):
+            li_user = gr.Textbox(label="Username")
+            li_pass = gr.Textbox(label="Password", type="password")
+            li_btn = gr.Button("Login")
+            li_out = gr.Textbox()
+
+            li_btn.click(login, [li_user, li_pass], li_out)
+
+        # OTP TAB
+        with gr.Tab("OTP Verification"):
+            otp_user = gr.Textbox(label="Username")
+            otp_input = gr.Textbox(label="OTP")
+            otp_btn = gr.Button("Verify")
+            otp_out = gr.Textbox()
+
+            otp_btn.click(
+                verify_otp,
+                [otp_user, otp_input],
+                [otp_out, logged_in]
+            )
+
+        # DASHBOARD TAB
+        with gr.Tab("IDS Dashboard"):
+            inputs = [gr.Number(label=f"Feature {i+1}") for i in range(10)]
+            pred_btn = gr.Button("Analyze")
+            output = gr.Markdown()
+
+            pred_btn.click(predict, [logged_in] + inputs, output)
 
 demo.launch()