Update app.py

app.py

@@ -93,6 +93,28 @@ def visualize(hr, lr, recon, sr):
     ax3.set_ylabel('Частота, Гц')
     return fig
 
+
+
+def waveplot(hr, lr, recon, sr):
+    fig, (ax1, ax2, ax3) = plt.subplots(3, 1, sharey=True, sharex=True, figsize=(16, 12))
+    ax1.title.set_text('Оригинальный сигнал')
+    ax2.title.set_text('Сигнал с потерями')
+    ax3.title.set_text('Улучшенный сигнал')
+
+    canvas = FigureCanvas(fig)
+    p = librosa.display.waveplot(hr, ax=ax1, sr=sr)
+    p = librosa.display.waveplot(lr, ax=ax2, sr=sr)
+    p = librosa.display.waveplot(recon, ax=ax3, sr=sr)
+
+    ax1.set_xlabel('Время, с')
+    #ax1.set_ylabel('Частота, Гц')
+    ax2.set_xlabel('Время, с')
+    #ax2.set_ylabel('Частота, Гц')
+    ax3.set_xlabel('Время, с')
+    #ax3.set_ylabel('Частота, Гц')
+    return fig
+
+    
 packet_size = CONFIG.DATA.EVAL.packet_size
 window = CONFIG.DATA.window_size
 stride = CONFIG.DATA.stride
@@ -147,8 +169,20 @@ if st.button('Сгенерировать потери'):
         output = inference(re_im, session, onnx_model, input_names, output_names)
         st.text(str(time.time() - start_time))
         st.subheader('3. Визуализация')
-        fig = visualize(target, lossy_input, output, sr)
-        st.pyplot(fig)
+        fig_1 = visualize(target, lossy_input, output, sr)
+        fig_2 = waveplot(target, lossy_input, output, sr)
+        tab1, tab2 = st.tabs(["Частотная", "Временная"])
+
+        with tab1:
+           st.header("Частотная область - Спектрограмма")
+           st.pyplot(fig_1)
+
+        with tab2:
+           st.header("Временная область - вейвплот")
+           st.pyplot(fig_2)
+
+            
+        #st.pyplot(fig)
     st.success('Сделано!')
     sf.write('target.wav', target, sr)
     sf.write('lossy.wav', lossy_input, sr)