Update the new electrode mapping & Plot text label. - 1213

app.py

@@ -50,11 +50,37 @@ def emg_plot(event_index, event_plot_name, left_std_ratio, left_delta_t, right_s
     # Add horizontal and vertical lines to create quadrants
     plt.axhline(y=0, color='black', linestyle='--')
     plt.axvline(x=0, color='black', linestyle='--')
+    
+    # Add quadrant labels
+    # Add styled text labels with colored box
+    def add_styled_text(x, y, text, va='bottom'):
+        # Create text box style
+        bbox_props = dict(
+            boxstyle='round,pad=0.5',
+            fc='#1f77b4',      # 背景顏色（白色）
+            ec='#1f77b4',   # 邊框顏色（藍色）
+            alpha=0.7,      # 背景透明度
+            lw=1.5          # 邊框寬度
+        )
+        
+        plt.text(x, y, text,
+                horizontalalignment='center',
+                verticalalignment=va,
+                bbox=bbox_props,
+                color='white',
+                fontweight='semibold',
+                fontsize=9)
+
+    # Add styled quadrant labels
+    add_styled_text(4, 0.5, "Exertion + / Coordination -", 'bottom')
+    add_styled_text(-4, 0.5, "Exertion - / Coordination -", 'bottom')
+    add_styled_text(-4, -0.5, "Exertion - / Coordination +", 'top')
+    add_styled_text(4, -0.5, "Exertion + / Coordination +", 'top')
 
     # Add title and axis labels
     plt.title(f'Muscle Coordination Analysis - {event_index}:{event_plot_name}', fontsize=14)
-    plt.xlabel('Std Ratio > 3', fontsize=12)
-    plt.ylabel('Delta T > 0', fontsize=12)
+    plt.xlabel('Exertion (Std Ratio > 3)', fontsize=12, fontweight='semibold')
+    plt.ylabel('Coordination (Delta T > 0)', fontsize=12, fontweight='semibold')
 
     # Remove axis numbers and labels
     ax.set_xticks([])
@@ -63,7 +89,7 @@ def emg_plot(event_index, event_plot_name, left_std_ratio, left_delta_t, right_s
     ax.set_yticklabels([])
     
     # Set plot legend with color
-    plt.legend(['Left', 'Right'], loc='upper left', fontsize=10)
+    plt.legend(['Left Swallowing Muscle', 'Right Swallowing Muscle'], loc='upper left', fontsize=10)
 
     # Set the limits of the plot
     plt.xlim(-10, 10)
@@ -103,8 +129,8 @@ def main():
     
     # Load test data button
     if st.button('Load Test Data', type="primary"):
-        st.session_state.emg_data = pd.read_csv('test-1/0-New_Task-recording-0.csv', skiprows=[0,1,3,4])
-        st.session_state.time_marker = pd.read_csv('test-1/time_marker.csv')
+        st.session_state.emg_data = pd.read_csv('test-new/0-New_Task-recording-0.csv', skiprows=[0,1,3,4])
+        st.session_state.time_marker = pd.read_csv('test-new/time_marker.csv')
         st.success("Test data loaded successfully!")
         st.session_state.data_isready = True
 
@@ -129,11 +155,11 @@ def main():
         # Reset emg data index with Channels
         emg_data = st.session_state.emg_data.set_index('Channels')
         # Get signal data from difference of emg_data
-        signal_left_lateral = emg_data['21'] - emg_data['3']
-        signal_left_medial = emg_data['22'] - emg_data['2']
+        signal_left_lateral = emg_data['17'] - emg_data['18']
+        signal_left_medial = emg_data['19'] - emg_data['20']
 
-        signal_right_lateral = emg_data['16'] - emg_data['6']
-        signal_right_medial = emg_data['17'] - emg_data['5']
+        signal_right_lateral = emg_data['23'] - emg_data['24']
+        signal_right_medial = emg_data['21'] - emg_data['22']
         
         # RMS caculation : Define the moving average window size
         N = 25
@@ -149,13 +175,13 @@ def main():
         signal_right_medial_RMS = moving_rms(signal_right_medial, N)
         
         # Time Marker Processing
-        time_marker = st.session_state.time_marker[['0-New_Task-recording_time(us)', 'name', 'tag']]
+        time_marker = st.session_state.time_marker[['0-New_Task-recording_time(us)', 'description', 'tag']]
         time_marker = time_marker.rename(columns={'0-New_Task-recording_time(us)': 'event_time'})
         
         # Select column value with odd/even index
         event_start_times = time_marker.loc[0::2]['event_time'].values.astype(int)
         event_end_times = time_marker.loc[1::2]['event_time'].values.astype(int)
-        event_names = time_marker.loc[0::2]['name'].values
+        event_names = time_marker.loc[0::2]['description'].values
         
         # Get signal basic 10s std
         signal_left_lateral_basics_10s_std = signal_left_lateral_RMS.loc[: 10000000].std()
@@ -173,11 +199,17 @@ def main():
             st.write(f"Start time: {float(event_start_time)/1000000: .3f} sec, End time: {float(event_end_time)/1000000: .3f} sec")
             
             # Get event signal data with event time duration
-            event_signal_LL = signal_left_lateral_RMS.loc[event_start_time:event_end_time]
-            event_signal_LM = signal_left_medial_RMS.loc[event_start_time:event_end_time]
+            mask_LL = (signal_left_lateral_RMS.index >= event_start_time) & (signal_left_lateral_RMS.index <= event_end_time)
+            event_signal_LL = signal_left_lateral_RMS.iloc[mask_LL]
+            
+            mask_LM = (signal_left_medial_RMS.index >= event_start_time) & (signal_left_medial_RMS.index <= event_end_time)
+            event_signal_LM = signal_left_medial_RMS.iloc[mask_LM]
+            
+            mask_RL = (signal_right_lateral_RMS.index >= event_start_time) & (signal_right_lateral_RMS.index <= event_end_time)
+            event_signal_RL = signal_right_lateral_RMS.iloc[mask_RL]
             
-            event_signal_RL = signal_right_lateral_RMS.loc[event_start_time:event_end_time]
-            event_signal_RM = signal_right_medial_RMS.loc[event_start_time:event_end_time]
+            mask_RM = (signal_right_medial_RMS.index >= event_start_time) & (signal_right_medial_RMS.index <= event_end_time)
+            event_signal_RM = signal_right_medial_RMS.iloc[mask_RM]
             
             # Calculate std ratio 
             left_event_std = event_signal_LL.std()

test-new/time_marker.csv

@@ -0,0 +1,13 @@
+Abs_time(us),0-New_Task-recording_time(us),topic,type,name,uuid,signal,description,color,order,tag
+1733899760533486.0,78357902.0,keyboard/c,keyboard,Keyboard,260efabb-3944-448b-88b4-1eef7de020dd,c,Cough,#baeea0,0,start
+1733899761846583.0,79670999.0,keyboard/c,keyboard,Keyboard,60fe4089-f190-4a34-9b49-7b6392f2dcb9,c,Cough,#baeea0,0,end
+1733899763403073.0,81227489.0,keyboard/b,keyboard,Keyboard,cb091860-8a06-460d-927d-c2006fa0d764,b,Bite,#fae2a3,1,start
+1733899764640907.0,82465323.0,keyboard/b,keyboard,Keyboard,72c6fbd9-8c83-48de-a89e-426636805944,b,Bite,#fae2a3,1,end
+1733899766415311.0,84239727.0,keyboard/s,keyboard,Keyboard,8fa85742-77ec-475c-a6dc-0fd4a986c943,s,Swallow,#fab6b6,2,start
+1733899767609930.0,85434346.0,keyboard/s,keyboard,Keyboard,8005446f-862c-446b-bc7e-7c428efc1f7a,s,Swallow,#fab6b6,2,end
+1733899768804131.0,86628547.0,keyboard/s,keyboard,Keyboard,8fa85742-77ec-475c-a6dc-0fd4a986c943,s,Swallow,#fab6b6,3,start
+1733899770127418.0,87951834.0,keyboard/s,keyboard,Keyboard,8005446f-862c-446b-bc7e-7c428efc1f7a,s,Swallow,#fab6b6,3,end
+1733899771849409.0,89673825.0,keyboard/s,keyboard,Keyboard,8fa85742-77ec-475c-a6dc-0fd4a986c943,s,Swallow,#fab6b6,4,start
+1733899773334247.0,91158663.0,keyboard/s,keyboard,Keyboard,8005446f-862c-446b-bc7e-7c428efc1f7a,s,Swallow,#fab6b6,4,end
+1733899774433363.0,92257779.0,keyboard/c,keyboard,Keyboard,260efabb-3944-448b-88b4-1eef7de020dd,c,Cough,#baeea0,5,start
+1733899775890252.0,93714668.0,keyboard/c,keyboard,Keyboard,60fe4089-f190-4a34-9b49-7b6392f2dcb9,c,Cough,#baeea0,5,end