split eol test

app.py

@@ -39,9 +39,18 @@ def gradio_TSE(input_audio_path, enroll_audio_path1, enroll_audio_path2, audio_t
     eol_wav1 = "eol1.wav"
     sf.write(eol_wav1, enroll_wav1, 16000)
 
-    enroll_wav2 = fix_audio_format(enroll_audio_path2)
-    eol_wav2 = "eol2.wav"
-    sf.write(eol_wav2, enroll_wav2, 16000)
+    if len(enroll_wav1) > 4 * 16000:
+        middle_start = (len(enroll_wav1) - 3 * 16000) // 2
+        middle_end = middle_start + 3 * 16000
+        enroll_wav2 = enroll_wav1[middle_start:middle_end]
+        print("成功提取 enroll_wav2，长度:", len(enroll_wav2))
+        eol_wav2 = "eol2.wav"
+        sf.write(eol_wav2, enroll_wav2, 16000)
+    else:
+        enroll_wav2 = fix_audio_format(enroll_audio_path2)
+        print("成功导入 enroll_wav2，长度:", len(enroll_wav2))
+        eol_wav2 = "eol2.wav"
+        sf.write(eol_wav2, enroll_wav2, 16000)
 
     if audio_type == "clean":
         noise_folder_test = "noises/"