Update app.py

app.py

@@ -68,13 +68,23 @@ def detect_intrusion(file):
     # 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]
 
-    return log_data[['date', 'time', 'Prediction']].to_string(index=False)
+    # Format date for output
+    log_data['date'] = log_data['date'].dt.strftime('%Y-%m-%d')
+
+    # Select final output columns
+    output_df = log_data[['date', 'time', 'Prediction']]
+
+    # Save the output to a CSV file for download
+    output_file = "intrusion_results.csv"
+    output_df.to_csv(output_file, index=False)
+
+    return output_df, output_file
 
 # Create Gradio interface
 iface = gr.Interface(
     fn=detect_intrusion,
     inputs=[gr.File(label="Upload Log File (CSV format)")],
-    outputs="text",
+    outputs=[gr.Dataframe(label="Intrusion Detection Results"), gr.File(label="Download Predictions CSV")],
     title="Intrusion Detection System",
     description=("""
         Upload a CSV log file with the following features: