Spaces:
Sleeping
Sleeping
File size: 2,695 Bytes
b99e090 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
import streamlit as st
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
# Set streamlit configuration with wide mode
st.set_page_config(layout="wide")
st.config.set_option("server.enableXsrfProtection", False)
def plot_signal(signal, index):
# Assuming df is your DataFrame and 'Signal1', 'Signal2', 'Signal3' are your columns
ch1 = pd.to_numeric(signal[index], errors='coerce')
# Define the moving average window size
N = 25
# Function to calculate moving RMS
def moving_rms(signal, window_size):
return np.sqrt(pd.Series(signal).rolling(window=window_size).mean()**2)
# Calculate moving RMS for each channel
ch1_rms = moving_rms(ch1, N)
# Calculate mean and standard deviation of the RMS signals
mean_ch1_rms = np.mean(ch1_rms)
std_ch1_rms = np.std(ch1_rms)
# Print mean and standard deviation values
st.write(f'Mean of RMS Signal {int(index)}: {mean_ch1_rms:.3f}, Std Dev of RMS Signal 1: {std_ch1_rms:.3f}')
# Plot the RMS signals
plt.figure(figsize=(15,6))
plt.plot(ch1_rms, label='Signal 1 RMS')
plt.title('RMS Signal Values, Moving Average Window Size N=25')
plt.xlabel('Index')
plt.ylabel('RMS Value')
plt.legend()
st.pyplot(plt.gcf())
# plt.show()
def main():
st.title('👅Dysphagia Analysis - by ITRI BDL')
uploaded_file1 = st.file_uploader("Choose the EMG_data CSV file", type="csv")
uploaded_file2 = st.file_uploader("Choose the time_marker CSV file", type="csv")
if uploaded_file1 is not None:
try:
data1 = pd.read_csv(uploaded_file1, skiprows=[0,1,3,4])
# st.dataframe(data1)
except Exception as e:
st.error(f"Error: {e}")
if uploaded_file2 is not None:
try:
data2 = pd.read_csv(uploaded_file2, skiprows=[0,1])
# st.dataframe(data2)
except Exception as e:
st.error(f"Error: {e}")
# Detect Upload File and choose signal with sidebar menu
if uploaded_file1 is not None and uploaded_file2 is not None:
signal = st.sidebar.selectbox('Select EMG Signal', data1.columns)
st.write(f'You selected: channel {signal}')
# Plot the selected signal
st.line_chart(data1[signal])
# Analysis button
if st.button('Start Analysis'):
st.write('Analysis started...')
# Add analysis code here
plot_signal(data1, signal)
st.write('Analysis completed...')
if __name__ == '__main__':
main() |