Update app.py

app.py

@@ -194,8 +194,22 @@ if st.button('Сгенерировать потери'):
     stoi_mass=[stoi_orig, stoi_lossy, stoi_enhanced]
 
 
-
-    
+    def get_power(x, nfft):
+        S = librosa.stft(x, n_fft=nfft)
+        S = np.log(np.abs(S) ** 2 + 1e-8)
+        return S
+    def LSD(x_hr, x_pr):
+        S1 = get_power(x_hr, nfft=2048)
+        S2 = get_power(x_pr, nfft=2048)
+        lsd = np.mean(np.sqrt(np.mean((S1 - S2) ** 2, axis=-1)), axis=0)
+        return lsd
+
+    lsd_orig = LSD(data_clean,data_clean)
+    lsd_lossy = LSD(data_lossy,data_clean)
+    lsd_enhanced = LSD(data_enhanced,data_clean)
+
+    lsd_mass=[lsd_orig, lsd_lossy, lsd_enhanced]
+        
     if samplerate != 8000:
         data_lossy = librosa.resample(data_lossy, orig_sr=48000, target_sr=8000)
         data_clean = librosa.resample(data_clean, orig_sr=48000, target_sr=8000)
@@ -209,8 +223,13 @@ if st.button('Сгенерировать потери'):
 
     psq_mas=[pesq_orig, pesq_lossy, pesq_enhanced]
 
+    
+
 
 
+
+    
+
     df = pd.DataFrame(columns=['Audio', 'PESQ', 'STOI', 'PLCMOS', 'LSD'])
 
     df['Audio'] = ['Clean', 'Lossy', 'Enhanced']
@@ -219,6 +238,8 @@ if st.button('Сгенерировать потери'):
 
     df['STOI'] = stoi_mass
 
+    df['LSD'] = lsd_mass
+
     st.table(df)