Update app.py

app.py

@@ -1,6 +1,7 @@
-import gradio as gr
-import spaces
 from audio_separator.separator import Separator
+from flask import Flask, request, send_file
+import requests
+from pathlib import Path
 
 separator = Separator()
 
@@ -94,7 +95,6 @@ demucs_models = [
     'hdemucs_mmi.yaml',
 ]
 
-@spaces.GPU(duration=300)
 def roformer_separator(audio, checkpoint_name):
     full_checkpoint_name = roformer_models[checkpoint_name]
     separator.load_model(full_checkpoint_name)
@@ -103,7 +103,6 @@ def roformer_separator(audio, checkpoint_name):
     stem2 = output_files[1]
     return stem1, stem2
 
-@spaces.GPU(duration=300)
 def mdx_vr_separator(audio, checkpoint_name):
     separator.load_model(checkpoint_name)
     output_files = separator.separate(audio)
@@ -111,7 +110,6 @@ def mdx_vr_separator(audio, checkpoint_name):
     stem2 = output_files[1]
     return stem1, stem2
 
-@spaces.GPU(duration=300)
 def demucs_separator(audio, checkpoint_name):
     separator.load_model(checkpoint_name)
     output_files = separator.separate(audio)
@@ -121,56 +119,56 @@ def demucs_separator(audio, checkpoint_name):
     stem4 = output_files[3]
     return stem1, stem2, stem3, stem4
 
-with gr.Blocks(title="🎵 UVR5 UI 🎵") as demo:
-    gr.Markdown("<h1> 🎵 UVR5 UI 🎵 </h1>")
-    with gr.Tab("Vocal Separator (UVR)"):
-        gr.Markdown("Separate vocals and instruments from an audio file using UVR models.")
-        with gr.Tab("Mel/BS Roformer"):
-            roformer_audio_file = gr.Audio(label="Audio File", type="filepath")
-            with gr.Row():
-                roformer_model = gr.Dropdown(label="Model", choices=list(roformer_models.keys()))
-                roformer_button = gr.Button("Separate", variant="primary")
-            roformer_stem1 = gr.Audio(type="filepath", label="Stem 1")
-            roformer_stem2 = gr.Audio(type="filepath", label="Stem 2")
-
-            roformer_button.click(roformer_separator, [roformer_audio_file, roformer_model], [roformer_stem1, roformer_stem2])
-        with gr.Tab("MDX23C"):
-            mdx23c_audio_file = gr.Audio(label="Audio File", type="filepath")
-            with gr.Row():
-                mdx23c_model = gr.Dropdown(label="Model", choices=mdx23c_models)
-                mdx23c_button = gr.Button("Separate", variant="primary")
-            mdx23c_stem1 = gr.Audio(type="filepath", label="Stem 1")
-            mdx23c_stem2 = gr.Audio(type="filepath", label="Stem 2")
-
-            mdx23c_button.click(mdx_vr_separator, [mdx23c_audio_file, mdx23c_model], [mdx23c_stem1, mdx23c_stem2])
-        with gr.Tab("MDX-NET"):
-            mdxnet_audio_file = gr.Audio(label="Audio File", type="filepath")
-            with gr.Row():
-                mdxnet_model = gr.Dropdown(label="Model", choices=mdxnet_models)
-                mdxnet_button = gr.Button("Separate", variant="primary")
-            mdxnet_stem1 = gr.Audio(type="filepath", label="Stem 1")
-            mdxnet_stem2 = gr.Audio(type="filepath", label="Stem 2")
-
-            mdxnet_button.click(mdx_vr_separator, [mdxnet_audio_file, mdxnet_model], [mdxnet_stem1, mdxnet_stem2])
-        with gr.Tab("VR-ARCH"):
-            vr_audio_file = gr.Audio(label="Audio File", type="filepath")
-            with gr.Row():
-                vr_model = gr.Dropdown(label="Model", choices=vrarch_models)
-                vr_button = gr.Button("Separate", variant="primary")
-            vr_stem1 = gr.Audio(type="filepath", label="Stem 1")
-            vr_stem2 = gr.Audio(type="filepath", label="Stem 2")
-
-            vr_button.click(mdx_vr_separator, [vr_audio_file, vr_model], [vr_stem1, vr_stem2])
-        with gr.Tab("Demucs"):
-            demucs_audio_file = gr.Audio(label="Audio File", type="filepath")
-            with gr.Row():
-                demucs_model = gr.Dropdown(label="Model", choices=demucs_models)
-                demucs_button = gr.Button("Separate", variant="primary")
-            demucs_stem1 = gr.Audio(type="filepath", label="Stem 1")
-            demucs_stem2 = gr.Audio(type="filepath", label="Stem 2")
-            demucs_stem3 = gr.Audio(type="filepath", label="Stem 3")
-            demucs_stem4 = gr.Audio(type="filepath", label="Stem 4")
-
-            demucs_button.click(demucs_separator, [demucs_audio_file, demucs_model], [demucs_stem1, demucs_stem2, demucs_stem3, demucs_stem4])
-
-demo.launch()
+app = Flask(__name__)
+
+@app.route('/', methods=['GET'])
+def hello():
+    return "Hello! This is an api server, and it is running successfully. For usage, please contact the person who hosted this api server."
+
+@app.route('/api/audio_separation', methods=['GET'])
+def audio_separation():
+    try:
+    	# 获取MP3文件的直链地址
+        mp3_url = request.args.get('url')
+        if not mp3_url:
+        	return "Error: URL parameter is required", 400
+
+    	# 下载MP3文件到本地
+        response = requests.get(mp3_url)
+        mp3_filename = mp3_url.split('/')[-1]  # 使用下载的文件名
+        with open("/tmp/" + mp3_filename, 'wb') as f:
+        	f.write(response.content)
+
+        # 执行音频分离操作
+        mdxnet_stem1, mdxnet_stem2 = mdx_vr_separator([response.content, UVR-MDX-NET-Inst_HQ_3.onnx])
+
+        # 生成分离后的文件名
+        vocals_filename = f"{os.path.splitext(mp3_filename)[0]}_vocals.wav"
+        no_vocals_filename = f"{os.path.splitext(mp3_filename)[0]}_no_vocals.wav"
+        
+        # 保存结果
+        with open(vocals_filename, 'wb') as file:
+            file.write(mdxnet_stem1)
+
+        with open(no_vocals_filename, 'wb') as file:
+            file.write(mdxnet_stem2)
+
+    	# 提供文件的永久直链
+        vocals_url = f"/download/{os.path.basename(vocals_filename)}"
+        no_vocals_url = f"/download/{os.path.basename(no_vocals_filename)}"
+
+    	# 返回结果
+        result = {
+        	"vocals_url": vocals_url,
+        	"no_vocals_url": no_vocals_url
+    	}
+        return result
+    except Exception as e:
+        return "Error: " + str(e), 500
+
+@app.route('/download/<filename>', methods=['GET'])
+def download(filename):
+    return send_file("/tmp/" + filename, as_attachment=True)
+
+if __name__ == '__main__':
+    app.run(debug=False)