Update app.py

app.py

@@ -17,15 +17,16 @@ numeric_features = [
     "door_state", "sphone_signal"
 ]
 
+# Class labels for attack types
 class_labels = {
-    0: "normal",
-    1: "backdoor",
-    2: "ddos",
-    3: "injection",
-    4: "password",
-    5: "ransomware",
-    6: "scanning",
-    7: "xss",
+    0: "Normal",
+    1: "Backdoor",
+    2: "DDoS",
+    3: "Injection",
+    4: "Password Attack",
+    5: "Ransomware",
+    6: "Scanning",
+    7: "XSS",
 }
 
 def convert_datetime_features(log_data):
@@ -65,8 +66,8 @@ def detect_intrusion(file):
     except Exception as e:
         return f"Error during prediction: {str(e)}"
 
-    # Only output whether it's an attack or normal
-    log_data['Prediction'] = ['Attack' if class_labels.get(pred, 'Unknown') != "normal" else "Normal" for pred in predictions]
+    # Map predictions to specific attack types
+    log_data['Prediction'] = [class_labels.get(pred, 'Unknown Attack') for pred in predictions]
 
     # Format date for output
     log_data['date'] = log_data['date'].dt.strftime('%Y-%m-%d')