use flac

app.py

@@ -7,7 +7,12 @@ import re
 import os
 import os.path as op
 import torch
+import soundfile as sf
+import numpy as np
+import tempfile
+
 from download import download_model
+from levo_inference import LeVoInference
 
 # 下载模型
 APP_DIR = op.dirname(op.abspath(__file__))
@@ -17,8 +22,6 @@ os.makedirs(base_full_path, exist_ok=True)
 download_model(base_full_path, repo_id="lglg666/SongGeneration-base-full", revision="19ebdb6")
 print("Successful downloaded model.")
 
-from levo_inference import LeVoInference
-
 # 模型初始化
 MODEL = LeVoInference(base_full_path)
 
@@ -73,6 +76,18 @@ with open(op.join(APP_DIR, 'conf/vocab.yaml'), 'r', encoding='utf-8') as file:
     STRUCTS = yaml.safe_load(file)
 
 
+def save_as_flac(sample_rate, audio_data):
+    if isinstance(audio_data, tuple):
+        sample_rate, audio_data = audio_data
+    
+    if audio_data.dtype == np.float64:
+        audio_data = audio_data.astype(np.float32)
+    
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".flac")
+    sf.write(temp_file, audio_data, sample_rate, format='FLAC')
+    return temp_file.name
+
+
 # 模拟歌曲生成函数
 def generate_song(lyric, description=None, prompt_audio=None, genre=None, cfg_coef=None, temperature=None, top_k=None, gen_type="mixed", progress=gr.Progress(track_tqdm=True)):
     global MODEL
@@ -138,7 +153,8 @@ def generate_song(lyric, description=None, prompt_audio=None, genre=None, cfg_co
         "timestamp": datetime.now().isoformat(),
     }
     
-    return (sample_rate, audio_data), json.dumps(input_config, indent=2)
+    filepath = save_as_flac(sample_rate, audio_data)
+    return filepath, json.dumps(input_config, indent=2)
 
 
 # 创建Gradio界面
@@ -227,7 +243,7 @@ lyrics
                 generate_bgm_btn = gr.Button("Generate Pure Music", variant="primary")
         
         with gr.Column():
-            output_audio = gr.Audio(label="Generated Song", type="numpy")
+            output_audio = gr.Audio(label="Generated Song", type="filepath")
             output_json = gr.JSON(label="Generated Info")
     
         # # 示例按钮