Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import torch
-import torchaudio
 import tempfile
 import os
 import numpy as np
@@ -19,28 +18,31 @@ def separate_audio(audio_file):
         return None, None
     
     try:
-        # Dùng librosa để load — hỗ trợ mọi định dạng và tần số
-        waveform, sample_rate = librosa.load(audio_file, sr=8000, mono=True)
+        # Load và resample về 8kHz bằng librosa
+        waveform, _ = librosa.load(audio_file, sr=8000, mono=True)
         
-        # Lưu file tạm về 8kHz
+        # Lưu file tạm
         tmp_path = "/tmp/input.wav"
         sf.write(tmp_path, waveform, 8000)
         
         # Tách nguồn âm
         est_sources = model.separate_file(path=tmp_path)
         
-        # Lưu kết quả
+        # Lấy kết quả
+        source1 = est_sources[:, :, 0].detach().cpu().numpy().squeeze()
+        source2 = est_sources[:, :, 1].detach().cpu().numpy().squeeze()
+        
+        # Lưu kết quả bằng soundfile
         source1_path = "/tmp/source1.wav"
         source2_path = "/tmp/source2.wav"
-        
-        torchaudio.save(source1_path, est_sources[:, :, 0].detach().cpu(), 8000)
-        torchaudio.save(source2_path, est_sources[:, :, 1].detach().cpu(), 8000)
+        sf.write(source1_path, source1, 8000)
+        sf.write(source2_path, source2, 8000)
         
         return source1_path, source2_path
     
     except Exception as e:
         print(f"Error: {e}")
-        raise gr.Error(f"Lỗi xử lý audio: {str(e)}")
+        raise gr.Error(f"Lỗi: {str(e)}")
 
 demo = gr.Interface(
     fn=separate_audio,
@@ -61,9 +63,9 @@ Demo ứng dụng kiến trúc SepFormer cho bài toán phân tách nguồn âm
 2. Nhấn Submit
 3. Nghe và download 2 nguồn âm đã được tách
 
-**Lưu ý:** Hỗ trợ định dạng WAV, MP3, FLAC. Mọi tần số lấy mẫu đều được tự động chuyển về 8kHz.
+**Lưu ý:** Hỗ trợ định dạng WAV, MP3, FLAC.
 
-> ⚠️ **Khuyến nghị:** Nên sử dụng file âm thanh có tần số lấy mẫu **8kHz** để đạt chất lượng phân tách tốt nhất.
+> ⚠️ **Khuyến nghị:** Nên sử dụng file âm thanh có tần số lấy mẫu **8kHz** và định dạng file có đuôi .wav để đạt chất lượng phân tách tốt nhất.
     """
 )