Upload eeg_processor.py

modules/eeg_processor.py

@@ -135,4 +135,27 @@ class EEGProcessor:
             'activity': activity,
             'mobility': mobility,
             'complexity': complexity
+        }
+    
+    def process_file(self, file_path: str) -> Dict:
+        """Process an EEG file and extract features.
+        
+        Args:
+            file_path: Path to the EEG file (EDF, BDF, or CNT format)
+            
+        Returns:
+            Dict containing processed EEG data and features
+        """
+        # Load EEG file using MNE
+        raw = mne.io.read_raw_edf(file_path, preload=True)
+        
+        # Preprocess the data
+        raw_processed = self.preprocess(raw)
+        
+        # Extract features
+        features = self.extract_features(raw_processed)
+        
+        return {
+            'raw_data': raw_processed,
+            'features': features
         }