Update app.py

app.py

@@ -30,7 +30,7 @@ class NetworkAnomalyDetector:
     def parse_pcap_file(self, file_path):
         """
         Parse network packet file with multiple parsing strategies
-        
+
         :param file_path: Path to the packet capture file
         :return: DataFrame with packet features
         """
@@ -66,7 +66,7 @@ class NetworkAnomalyDetector:
                         'timestamp': getattr(packet, 'time', 0)
                     }
                     packet_features.append(features)
-                
+
                 if packet_features:
                     return pd.DataFrame(packet_features)
             except Exception as e:
@@ -76,7 +76,7 @@ class NetworkAnomalyDetector:
         if pyshark is not None:
             try:
                 import asyncio
-                
+
                 async def parse_with_pyshark():
                     capture = pyshark.FileCapture(file_path)
                     local_features = []
@@ -92,14 +92,14 @@ class NetworkAnomalyDetector:
                             local_features.append(features)
                         except Exception as packet_error:
                             parsing_errors.append(f"PyShark packet parsing error: {str(packet_error)}")
-                    
+
                     capture.close()
                     return local_features
 
                 # Run the async function
                 try:
                     packet_features = asyncio.run(parse_with_pyshark())
-                    
+
                     if packet_features:
                         return pd.DataFrame(packet_features)
                 except Exception as async_error:
@@ -109,7 +109,7 @@ class NetworkAnomalyDetector:
 
         # Fallback: Raw file reading
         packet_features = read_raw_file(file_path)
-        
+
         if packet_features:
             # Log parsing errors if any occurred
             if parsing_errors:
@@ -189,7 +189,7 @@ class NetworkAnomalyDetector:
         """
         # Parse packet file
         packets_df = self.parse_pcap_file(file_path)
-        
+
         if packets_df.empty:
             return {
                 'summary': {
@@ -201,30 +201,30 @@ class NetworkAnomalyDetector:
 
         # Prepare features for anomaly detection
         feature_columns = ['length', 'timestamp']
-        
+
         # Handle protocol and IP as categorical features
         packets_df['protocol_encoded'] = pd.Categorical(packets_df['protocol']).codes
         packets_df['src_ip_encoded'] = pd.Categorical(packets_df['src_ip']).codes
         packets_df['dst_ip_encoded'] = pd.Categorical(packets_df['dst_ip']).codes
-        
+
         feature_columns.extend(['protocol_encoded', 'src_ip_encoded', 'dst_ip_encoded'])
-        
+
         # Prepare features for anomaly detection
         features = packets_df[feature_columns]
-        
+
         # Scale features
         features_scaled = self.scaler.fit_transform(features)
-        
+
         # Detect anomalies
         anomaly_labels = self.isolation_forest.fit_predict(features_scaled)
         packets_df['is_anomaly'] = anomaly_labels == -1
-        
+
         # Create summary
         summary = {
             'total_packets': len(packets_df),
             'isolation_forest_anomalies': sum(packets_df['is_anomaly'])
         }
-        
+
         return {
             'summary': summary,
             'packets': packets_df
@@ -269,7 +269,7 @@ def create_gradio_interface():
 
         with gr.Row():
             file_input = gr.File(label="Upload PCAP File", type="filepath", file_types=['.pcap', '.pkt'])
-            
+
         analyze_button = gr.Button("Analyze Network File", variant="primary")
 
         # Outputs
@@ -279,8 +279,8 @@ def create_gradio_interface():
 
         # Event handlers
         analyze_button.click(
-            fn=analyze_network_file, 
-            inputs=[file_input], 
+            fn=analyze_network_file,
+            inputs=[file_input],
             outputs=[summary_output, results_dataframe, error_output]
         )