Update src/streamlit_app.py

src/streamlit_app.py

@@ -100,3 +100,54 @@ if generate_button:
                 st.image(gif_path, caption="EEG Topomap Animation")
     except Exception as e:
         st.error(f"❌ Error: {str(e)}")
+
+
+plot_waveform_button = st.button("Plot TP9 Waveforms")
+
+if plot_waveform_button:
+    try:
+        # Download and read CSV
+        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]
+
+        # Plot raw waveform
+        fig_raw, ax_raw = plt.subplots(figsize=(12, 4))
+        ax_raw.plot(timestamps, tp10_raw, label='TP9', alpha=0.6, color='blue')
+        ax_raw.set_title('Raw Waveform of TP9')
+        ax_raw.set_xlabel('Time (seconds)')
+        ax_raw.set_ylabel('Amplitude')
+        ax_raw.legend()
+        ax_raw.grid(True)
+        st.pyplot(fig_raw)
+
+        # Plot filtered waveform
+        fig_filtered, ax_filtered = plt.subplots(figsize=(12, 4))
+        ax_filtered.plot(timestamps, tp10_filtered_mne, label='TP9 (1-50Hz)', linewidth=1.5, color='green')
+        ax_filtered.set_title('Filtered Waveform of TP9')
+        ax_filtered.set_xlabel('Time (seconds)')
+        ax_filtered.set_ylabel('Amplitude')
+        ax_filtered.legend()
+        ax_filtered.grid(True)
+        st.pyplot(fig_filtered)
+
+    except Exception as e:
+        st.error(f"❌ Error: {str(e)}")