Merge pull request #12 from PayalLakra/bio_amptool

Added - BioAmp CSV Plotter

Author: lorforlinux

BioAmp_CSV_Plotter.py

@@ -0,0 +1,82 @@
+import tkinter as tk
+from tkinter import filedialog, messagebox, ttk
+import pandas as pd
+import plotly.graph_objects as go
+from plotly.subplots import make_subplots
+
+class CSVPlotterApp:
+    def __init__(self, root):
+        self.root = root
+        self.root.title("CSV Plotter GUI")
+        self.filename = None
+        self.data = None
+        self.create_widgets()
+
+    def create_widgets(self):
+        # Create a frame for buttons and file name display
+        self.control_frame = tk.Frame(self.root, padx=10, pady=10)
+        self.control_frame.pack(pady=10)
+
+        # Button to load CSV file
+        self.load_button = tk.Button(self.control_frame, text="Load CSV", command=self.load_csv, bg='#4CAF50', fg='white', font=('Arial', 12))
+        self.load_button.pack(side=tk.LEFT, padx=5)
+
+        # Label to display the name of the selected CSV file
+        self.file_label = tk.Label(self.control_frame, text="No file selected", font=('Arial', 12))
+        self.file_label.pack(side=tk.LEFT, padx=5)
+
+        # Frame for dropdown menu
+        self.dropdown_frame = tk.Frame(self.root, padx=10, pady=10)
+        self.dropdown_frame.pack(pady=10)
+
+        self.dropdown_label = tk.Label(self.dropdown_frame, text="Select Channel to Plot:", font=('Arial', 12))
+        self.dropdown_label.pack(side=tk.LEFT)
+
+        # Dropdown menu for channel selection
+        self.channel_selection = tk.StringVar()
+        self.dropdown_menu = ttk.Combobox(self.dropdown_frame, textvariable=self.channel_selection, state="normal", font=('Arial', 12))
+        self.dropdown_menu.pack(side=tk.LEFT, padx=5, fill=tk.X, expand=True)
+
+        # Button to plot data
+        self.plot_button = tk.Button(self.root, text="Plot Data", command=self.plot_data, bg='#2196F3', fg='white', font=('Arial', 12))
+        self.plot_button.pack(pady=10)
+
+    def load_csv(self):
+        # Open file dialog to select a CSV file
+        self.filename = filedialog.askopenfilename(filetypes=[("CSV files", "*.csv")])
+        if self.filename:
+            try:
+                self.data = pd.read_csv(self.filename)   # Load the CSV file
+                self.setup_dropdown_menu()
+                self.file_label.config(text=f"File: {self.filename.split('/')[-1]}")
+            except Exception as e:
+                messagebox.showerror("Error", f"Could not load CSV file: {e}")
+
+    def setup_dropdown_menu(self):
+        # Populate dropdown menu with the CSV column names (channels)
+        columns = list(self.data.columns)
+        self.dropdown_menu['values'] = columns
+        if columns:
+            self.channel_selection.set(columns[0])  # Default selection
+
+    def plot_data(self):
+        selected_channel = self.channel_selection.get()   # Get the selected channel
+        if not selected_channel:
+            messagebox.showerror("Error", "No channel selected for plotting")
+            return
+
+        fig = go.Figure()          # Plot the selected channel using Plotly
+        fig.add_trace(go.Scatter(x=self.data.index, y=self.data[selected_channel], mode='lines', name=selected_channel))
+        
+        fig.update_layout(
+            title=f"Channel: {selected_channel}",
+            xaxis_title="Index",
+            yaxis_title="Value",
+            template="plotly_white"
+        )
+        fig.show()
+
+if __name__ == "__main__":
+    root = tk.Tk()
+    app = CSVPlotterApp(root)
+    root.mainloop()

ECG.ipynb

@@ -77113,8 +77113,8 @@
     "\n",
     "    # Define a window around the R-peak to capture the full PQRST complex\n",
     "    window_size = int(0.5 * 250)  # Half a second before and after the R-peak \n",
-    "    # start_idx = max(0, highest_r_peak_idx - window_size)  \n",
-    "    # end_idx = min(len(ecg_signal), highest_r_peak_idx + window_size) \n",
+    "    start_idx = max(0, highest_r_peak_idx - window_size)  \n",
+    "    end_idx = min(len(ecg_signal), highest_r_peak_idx + window_size) \n",
     "\n",
     "    # Extract the time and ECG data for the window\n",
     "    window_time = time[start_idx:end_idx]\n",