Update src/streamlit_app.py

src/streamlit_app.py

@@ -269,6 +269,27 @@ if st.button("Show Band Power Chart (TP9)"):
 # Button to show EEG animation
 if st.button("Show Animated EEG Trace (TP9)"):
     try:
+        response = requests.get(csv_url)
+        response.raise_for_status()
+        data = pd.read_csv(io.StringIO(response.text))
+        st.write(f"Loaded data with shape: {data.shape}")
+
+        # Extract timestamps and TP9
+        timestamps = data['timestamps'] if 'timestamps' in data.columns else data['TimeStamp']
+        tp10_raw = data['TP9']
+
+        # Calculate sampling rate
+        time_diffs = np.diff(timestamps)
+        average_interval = np.mean(time_diffs)
+        sampling_rate = 1 / average_interval
+
+        # Create MNE Raw object
+        info = mne.create_info(ch_names=['TP9'], sfreq=sampling_rate, ch_types='eeg')
+        raw = mne.io.RawArray(np.array([tp10_raw]), info)
+
+        # Apply bandpass filter
+        raw.filter(1, 50, fir_design='firwin')
+        tp10_filtered_mne = raw.get_data()[0]
         # Setup figure and axes
         fig, ax = plt.subplots()
         line, = ax.plot([], [], lw=2)