Handle LSL stream disconnection by closing app after 2s of no data.

Author: PayalLakra

ffteeg.py

@@ -9,6 +9,7 @@
 from scipy.signal import butter, iirnotch, lfilter, lfilter_zi
 from scipy.fft import fft
 import math
+import time
 
 class EEGMonitor(QMainWindow):
     def __init__(self): 
@@ -17,6 +18,9 @@ def __init__(self):
         self.setWindowTitle("Real-Time EEG Monitor with FFT and Brainwave Power")
         self.setGeometry(100, 100, 1200, 800)
 
+        self.stream_active = True   # Flag to check if the stream is active
+        self.last_data_time = None  # Variable to store the last data time
+
         # Main layout split into two halves: top for EEG, bottom for FFT and Brainwaves
         self.central_widget = QWidget()
         self.main_layout = QVBoxLayout(self.central_widget)
@@ -108,6 +112,7 @@ def __init__(self):
     def update_plot(self):
         samples, _ = self.inlet.pull_chunk(timeout=0.0)
         if samples:
+            self.last_data_time = time.time()     # Store the last data time
             for sample in samples:
                 raw_point = sample[0]
 
@@ -129,6 +134,13 @@ def update_plot(self):
             time_axis = np.linspace(0, 4, len(plot_data))
             self.eeg_curve.setData(time_axis, plot_data)
 
+        else:
+            if self.last_data_time and (time.time() - self.last_data_time) > 2:
+                self.stream_active = False
+                print("LSL stream disconnected!")
+                self.timer.stop()
+                self.close()
+
     def process_fft_and_brainpower(self):
         window = np.hanning(len(self.moving_window))
         buffer_windowed = np.array(self.moving_window) * window