Update app.py

app.py

@@ -2,87 +2,146 @@ import gradio as gr
 import numpy as np
 import random
 
+# =========================
+# DATABASE (temporary)
+# =========================
 users = {}
 otp_store = {}
 
 # =========================
-# SIGNUP
+# SIGNUP FUNCTION
 # =========================
 def signup(username, password):
     if username in users:
         return "❌ User already exists"
+    
     users[username] = password
-    return "✅ Signup successful! Go to Login tab."
+    return "✅ Signup successful! Now go to Login tab."
 
 # =========================
-# LOGIN
+# LOGIN FUNCTION
 # =========================
 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)"
-    return "❌ Invalid credentials"
+        return f"✅ OTP generated: {otp} (for demo use this OTP)"
+    
+    return "❌ Invalid username or password"
 
 # =========================
-# OTP VERIFY
+# OTP VERIFICATION
 # =========================
 def verify_otp(username, otp_input):
     if username in otp_store and str(otp_store[username]) == str(otp_input):
-        return "✅ Login Successful", True
+        return "✅ Login Successful! You can now use IDS Dashboard.", True
+    
     return "❌ Invalid OTP", False
 
 # =========================
-# PREDICT
+# IDS PREDICTION FUNCTION
 # =========================
-def predict(logged_in, *inputs):
+def predict(logged_in, duration, src_bytes, dst_bytes, failed_logins,
+            count, error_rate, same_srv_rate, diff_srv_rate,
+            host_count, login_attempts):
+
+    # 🔐 Access control
     if not logged_in:
-        return "🚫 Please login first"
+        return """
+🚫 **Access Denied**
+
+You must:
+1. Signup  
+2. Login  
+3. Verify OTP  
+
+👉 Only then you can access IDS Dashboard
+"""
+
+    # Prepare data
+    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)
 
+    # Decision logic
     if score < 30:
-        return f"✅ Normal Traffic\n\nScore: {score:.2f}"
+        status = "✅ Normal Traffic"
+        explanation = "Low activity and safe behavior detected."
+        recommendation = "No action needed."
     elif score < 60:
-        return f"⚠️ Suspicious Activity\n\nScore: {score:.2f}"
+        status = "⚠️ Suspicious Activity"
+        explanation = "Moderate anomaly detected."
+        recommendation = "Monitor the system."
     else:
-        return f"🚨 Attack Detected!\n\nScore: {score:.2f}"
+        status = "🚨 Attack Detected!"
+        explanation = "High anomaly detected. Possible malicious activity."
+        recommendation = "Immediate action required!"
+
+    return f"""
+{status}
+
+### 🔢 Risk Score: {score:.2f}
+The risk score represents combined network activity intensity.
+
+### 📊 Interpretation:
+- 0 – 30 → Normal (Low Risk)
+- 30 – 60 → Suspicious (Medium Risk)
+- 60+ → Attack (High Risk)
+
+### 🧠 Analysis:
+{explanation}
+
+### 🛡️ Recommendation:
+{recommendation}
+"""
 
 # =========================
-# UI
+# UI DESIGN
 # =========================
 with gr.Blocks() as demo:
 
     logged_in = gr.State(False)
 
-    gr.Markdown("# 🔐 Secure IDS System")
+    gr.Markdown("# 🔐 Secure Intrusion Detection System")
 
     with gr.Tabs():
 
-        # SIGNUP TAB
+        # ================= SIGNUP =================
         with gr.Tab("Signup"):
+            gr.Markdown("### Create a new account")
+
             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
+        # ================= LOGIN =================
         with gr.Tab("Login"):
+            gr.Markdown("### Enter credentials")
+
             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
+        # ================= OTP =================
         with gr.Tab("OTP Verification"):
+            gr.Markdown("### Enter OTP sent after login")
+
             otp_user = gr.Textbox(label="Username")
             otp_input = gr.Textbox(label="OTP")
-            otp_btn = gr.Button("Verify")
+
+            otp_btn = gr.Button("Verify OTP")
             otp_out = gr.Textbox()
 
             otp_btn.click(
@@ -91,12 +150,38 @@ with gr.Blocks() as demo:
                 [otp_out, logged_in]
             )
 
-        # DASHBOARD TAB
+        # ================= IDS DASHBOARD =================
         with gr.Tab("IDS Dashboard"):
-            inputs = [gr.Number(label=f"Feature {i+1}") for i in range(10)]
-            pred_btn = gr.Button("Analyze")
+
+            gr.Markdown("### Analyze Network Traffic")
+
+            duration = gr.Number(label="Connection Duration")
+            src_bytes = gr.Number(label="Source Bytes (Data Sent)")
+            dst_bytes = gr.Number(label="Destination Bytes (Data Received)")
+
+            failed_logins = gr.Number(label="Failed Login Attempts")
+            login_attempts = gr.Number(label="Total Login Attempts")
+            count = gr.Number(label="Connection Count")
+
+            error_rate = gr.Number(label="Error Rate")
+            same_srv_rate = gr.Number(label="Same Service Rate")
+            diff_srv_rate = gr.Number(label="Different Service Rate")
+
+            host_count = gr.Number(label="Host Count")
+
+            pred_btn = gr.Button("🔍 Analyze Traffic")
+
             output = gr.Markdown()
 
-            pred_btn.click(predict, [logged_in] + inputs, output)
+            pred_btn.click(
+                predict,
+                [
+                    logged_in,
+                    duration, src_bytes, dst_bytes, failed_logins,
+                    count, error_rate, same_srv_rate, diff_srv_rate,
+                    host_count, login_attempts
+                ],
+                output
+            )
 
 demo.launch()