Update app.py

app.py

@@ -38,17 +38,16 @@ class_labels = {
     9: "mitm"
 }
 
-def detect_intrusion(feature_string, model_choice="Random Forest"):
-    # Split input string into key-value pairs and map them to the feature values
-    features = dict(item.split(" : ") for item in feature_string.split(", "))
-    
-    # Ensure all feature names are in the input
-    feature_values = []
-    for feature_name in feature_names:
-        if feature_name in features:
-            feature_values.append(float(features[feature_name]))
-        else:
-            return f"Missing feature: {feature_name}"
+def detect_intrusion(feature_values, model_choice="Random Forest"):
+    # Ensure the length of feature_values matches feature_names
+    if len(feature_values) != len(feature_names):
+        return "Please fill in all the required feature values."
+
+    # Convert the input values to floats and match them with feature names
+    try:
+        feature_values = [float(value) for value in feature_values]
+    except ValueError:
+        return "Please enter valid numerical values for all fields."
 
     # Choose the model based on user selection
     if model_choice == "Random Forest":
@@ -72,18 +71,20 @@ def detect_intrusion(feature_string, model_choice="Random Forest"):
     else:
         return f"Intrusion Detected: {class_labels.get(predicted_class, 'Unknown Attack')}"
 
-# Create a Gradio interface
-iface = gr.Interface(fn=detect_intrusion, 
-                     inputs=[gr.Textbox(label="Input Features (e.g., src_ip : 16 ,  src_port : 49321 ,  ...)"), 
-                             gr.Dropdown(choices=["Random Forest", "Decision Tree", "Bagging Classifier", "AdaBoost Classifier"], label="Select Model")], 
-                     outputs="text", 
-                     title="Intrusion Detection System",
-                     description="""
-                         Enter features as key-value pairs in the format: 
-                         feature_name1:value1, feature_name2:value2, ...
-                         Example: src_ip:16, src_port:49321, dst_ip:210, dst_port:8080, proto:1, service:5, ...
-                         Note: Ensure all required features are provided and take extra care for spaces around the colons and the commas a space before and after both characters.
-                     """)
+# Create Gradio input fields for each feature
+inputs = [gr.Textbox(label=feature_name) for feature_name in feature_names[:-1]]  # Exclude "label" field from inputs
+
+# Add model choice dropdown
+inputs.append(gr.Dropdown(choices=["Random Forest", "Decision Tree", "Bagging Classifier", "AdaBoost Classifier"], label="Select Model"))
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=detect_intrusion, 
+    inputs=inputs, 
+    outputs="text", 
+    title="Intrusion Detection System",
+    description="Fill in the blank fields for the network traffic features, and choose the model to detect intrusions."
+)
 
 # Launch the interface locally for testing
 iface.launch()