feat: 開放plotly控制選項

Keycatowo · Apr 23, 2023 · a6c341a · a6c341a
1 parent dc09472
commit a6c341a
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 5 deletions.
diff --git a/pages/1-Basic_Analysis.py b/pages/1-Basic_Analysis.py
@@ -6,7 +6,7 @@
 import numpy as np
 import librosa
 import pandas as pd
-from src.st_helper import convert_df, get_shift, update_sessions
+from src.st_helper import convert_df, get_shift, update_sessions, use_plotly
 from src.basic_info import plot_waveform, signal_RMS_analysis, plot_spectrogram
 
 
@@ -77,7 +77,7 @@
             st.write(f"Selected segment: `{start_time}` ~ `{end_time}`, duration: `{end_time-start_time}`")
             st.audio(y_sub, format="audio/ogg", sample_rate=sr)
 
-
+    use_plotly()
 
 
 #%% 功能分頁

diff --git a/pages/2-Pitch_Analysis.py b/pages/2-Pitch_Analysis.py
@@ -7,7 +7,7 @@
 import librosa
 import pandas as pd
 import seaborn as sns
-from src.st_helper import convert_df, get_shift, update_sessions
+from src.st_helper import convert_df, get_shift, update_sessions, use_plotly
 from src.pitch_estimation import (
     plot_mel_spectrogram, 
     plot_constant_q_transform, 
@@ -81,6 +81,7 @@
             st.write(f"Selected segment: `{start_time}` ~ `{end_time}`, duration: `{end_time-start_time}`")
             st.audio(y_sub, format="audio/ogg", sample_rate=sr)
 
+    use_plotly()
 
 
 #%% 功能分頁

diff --git a/pages/3-Time_Analysis.py b/pages/3-Time_Analysis.py
@@ -7,7 +7,7 @@
 import librosa
 import pandas as pd
 from src.beat_track import onsets_detection, plot_onset_strength, beat_analysis, predominant_local_pulse, static_tempo_estimation, plot_tempogram, onset_click_plot, beat_plot, plot_bpm
-from src.st_helper import convert_df, get_shift, update_sessions
+from src.st_helper import convert_df, get_shift, update_sessions, use_plotly
 import numpy as np
 
 st.title('Time Analysis')
@@ -73,6 +73,8 @@
         with st.expander("聲音片段(Segment of the audio)"):
             st.write(f"Selected segment: `{start_time}` ~ `{end_time}`, duration: `{end_time-start_time}`")
             st.audio(y_sub, format="audio/ogg", sample_rate=sr)
+
+    use_plotly()
 
 #%% 功能區塊
 if file is not None:

diff --git a/src/st_helper.py b/src/st_helper.py
@@ -64,4 +64,11 @@ def warning_region(text="This is a warning"):
         A warning region.
         If the user does not select a file, the warning will be shown.
     """
-    st.warning(text, icon="⚠️")
+    st.warning(text, icon="⚠️")
+
+
+def use_plotly():
+    st.session_state["use_plotly"] = st.sidebar.checkbox("Enable Dynamic Graphics", value=st.session_state["use_plotly"])
+    if st.session_state["use_plotly"]:
+        st.sidebar.info("Dynamic Graphics Enabled, more memory needed.\nWe will use Plotly to draw the figure if it is supported.")
+