Update app.py

app.py

@@ -2,147 +2,136 @@ import gradio as gr
 import numpy as np
 import random
 
-# =========================
-# DATABASE (temporary)
-# =========================
 users = {}
-
-# Store OTP temporarily
 otp_store = {}
 
+# =========================
+# STATE NAVIGATION
+# =========================
+def go_to(page):
+    return page
+
 # =========================
 # SIGNUP
 # =========================
 def signup(username, password):
     if username in users:
-        return "❌ User already exists"
-    
+        return "❌ User already exists", "signup"
     users[username] = password
-    return "✅ Signup successful! Please login."
-
+    return "✅ Signup successful! Please login.", "login"
 
 # =========================
-# LOGIN (STEP 1)
+# LOGIN
 # =========================
 def login(username, password):
     if username in users and users[username] == password:
         otp = random.randint(1000, 9999)
         otp_store[username] = otp
-        return f"✅ OTP generated: {otp} (demo purpose)"
-    else:
-        return "❌ Invalid credentials"
-
+        return f"OTP: {otp} (demo)", "otp", username
+    return "❌ Invalid credentials", "login", ""
 
 # =========================
-# OTP VERIFY (STEP 2)
+# OTP VERIFY
 # =========================
 def verify_otp(username, otp_input):
     if username in otp_store and str(otp_store[username]) == str(otp_input):
-        return "✅ Login Successful", True
-    else:
-        return "❌ Invalid OTP", False
-
+        return "✅ Login Successful", "dashboard", True
+    return "❌ Invalid OTP", "otp", False
 
 # =========================
-# PREDICT FUNCTION
+# PREDICT
 # =========================
-def predict(logged_in, duration, src_bytes, dst_bytes, failed_logins,
-            count, error_rate, same_srv_rate, diff_srv_rate,
-            host_count, login_attempts):
-
+def predict(logged_in, *inputs):
     if not logged_in:
         return "🚫 Please login first"
 
-    data = np.array([
-        duration, src_bytes, dst_bytes, failed_logins,
-        count, error_rate, same_srv_rate, diff_srv_rate,
-        host_count, login_attempts
-    ])
-    
+    data = np.array(inputs)
     score = np.sum(data)
 
     if score < 30:
-        status = "✅ Normal Traffic"
-        explanation = "Low activity detected."
+        return f"✅ Normal Traffic\n\nScore: {score:.2f}"
     elif score < 60:
-        status = "⚠️ Suspicious Activity"
-        explanation = "Moderate anomaly detected."
+        return f"⚠️ Suspicious Activity\n\nScore: {score:.2f}"
     else:
-        status = "🚨 Attack Detected!"
-        explanation = "High anomaly detected."
-
-    return f"""
-{status}
-
-Risk Score: {score:.2f}
-
-0–30 → Normal  
-30–60 → Suspicious  
-60+ → Attack  
-
-Analysis: {explanation}
-"""
-
+        return f"🚨 Attack Detected!\n\nScore: {score:.2f}"
 
 # =========================
 # UI
 # =========================
 with gr.Blocks() as demo:
 
-    gr.Markdown("# 🔐 Secure IDS with Authentication & 2FA")
-
-    # SIGNUP
-    gr.Markdown("## 📝 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, inputs=[su_user, su_pass], outputs=su_out)
-
-    gr.Markdown("---")
-
-    # LOGIN
-    gr.Markdown("## 🔑 Login")
-
-    li_user = gr.Textbox(label="Username")
-    li_pass = gr.Textbox(label="Password", type="password")
-    login_btn = gr.Button("Login")
-
-    login_out = gr.Textbox()
-
-    login_btn.click(login, inputs=[li_user, li_pass], outputs=login_out)
-
-    # OTP
-    gr.Markdown("## 🔢 Enter OTP")
-
-    otp_input = gr.Textbox(label="OTP")
-    otp_btn = gr.Button("Verify OTP")
-
-    otp_out = gr.Textbox()
-    logged_state = gr.State(False)
-
-    otp_btn.click(
-        verify_otp,
-        inputs=[li_user, otp_input],
-        outputs=[otp_out, logged_state]
-    )
-
-    gr.Markdown("---")
-
-    # IDS
-    gr.Markdown("## 📊 Network Analysis")
-
-    inputs = [gr.Number(label=f"Feature {i+1}") for i in range(10)]
-
-    predict_btn = gr.Button("Analyze")
-
-    output = gr.Markdown()
-
-    predict_btn.click(
-        predict,
-        inputs=[logged_state] + inputs,
-        outputs=output
-    )
+    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)
 
 demo.launch()