Upload 6 files
Browse files- app.py +20 -0
- fn.py +28 -0
- install.bat +56 -0
- main.py +55 -0
- requirements.txt +4 -0
- venv.sh +7 -0
app.py
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import fn
|
2 |
+
import gradio as gr
|
3 |
+
|
4 |
+
fn.load_model()
|
5 |
+
|
6 |
+
with gr.Blocks() as demo:
|
7 |
+
title = gr.Markdown('# audiosep')
|
8 |
+
with gr.Row():
|
9 |
+
src_audio = gr.Audio(label="Source", sources="upload", interactive=True, type="filepath")
|
10 |
+
dst_audio_1 = gr.Audio(label="Primary", interactive=False, type="filepath")
|
11 |
+
dst_audio_2 = gr.Audio(label="Secondary", interactive=False, type="filepath")
|
12 |
+
|
13 |
+
src_audio.change(
|
14 |
+
fn=fn.run,
|
15 |
+
inputs=[src_audio],
|
16 |
+
outputs=[dst_audio_1, dst_audio_2],
|
17 |
+
)
|
18 |
+
|
19 |
+
if __name__ == '__main__':
|
20 |
+
demo.launch()
|
fn.py
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import os
|
2 |
+
import tempfile
|
3 |
+
from audio_separator.separator import Separator
|
4 |
+
|
5 |
+
separator = None
|
6 |
+
|
7 |
+
def load_model():
|
8 |
+
global separator
|
9 |
+
|
10 |
+
separator = Separator()
|
11 |
+
separator.load_model('Kim_Vocal_2.onnx')
|
12 |
+
|
13 |
+
def run(audio_path):
|
14 |
+
global separator
|
15 |
+
|
16 |
+
if not audio_path:
|
17 |
+
return None, None
|
18 |
+
|
19 |
+
primary_stem_output_path, secondary_stem_output_path = separator.separate(audio_path)
|
20 |
+
|
21 |
+
return primary_stem_output_path, secondary_stem_output_path
|
22 |
+
|
23 |
+
def raw(audio):
|
24 |
+
with tempfile.NamedTemporaryFile(delete=True) as t:
|
25 |
+
with open(t.name, 'w+b') as f:
|
26 |
+
f.write(audio)
|
27 |
+
|
28 |
+
return run(t.name)
|
install.bat
ADDED
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
@echo off
|
2 |
+
|
3 |
+
rem -------------------------------------------
|
4 |
+
rem NOT guaranteed to work on Windows
|
5 |
+
|
6 |
+
set REPOS=https://huggingface.co/spaces/aka7774/audiosep
|
7 |
+
set APPDIR=audiosep
|
8 |
+
set VENV=venv
|
9 |
+
|
10 |
+
rem -------------------------------------------
|
11 |
+
|
12 |
+
set INSTALL_DIR=%~dp0
|
13 |
+
cd /d %INSTALL_DIR%
|
14 |
+
|
15 |
+
:git_clone
|
16 |
+
set DL_URL=%REPOS%
|
17 |
+
set DL_DST=%APPDIR%
|
18 |
+
git clone %DL_URL% %APPDIR%
|
19 |
+
if exist %DL_DST% goto install_python
|
20 |
+
|
21 |
+
set DL_URL=https://github.com/git-for-windows/git/releases/download/v2.41.0.windows.3/PortableGit-2.41.0.3-64-bit.7z.exe
|
22 |
+
set DL_DST=PortableGit-2.41.0.3-64-bit.7z.exe
|
23 |
+
curl -L -o %DL_DST% %DL_URL%
|
24 |
+
if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
|
25 |
+
%DL_DST% -y
|
26 |
+
del %DL_DST%
|
27 |
+
|
28 |
+
set GIT=%INSTALL_DIR%PortableGit\bin\git
|
29 |
+
%GIT% clone %REPOS%
|
30 |
+
|
31 |
+
:install_python
|
32 |
+
set DL_URL=https://github.com/indygreg/python-build-standalone/releases/download/20240107/cpython-3.10.13+20240107-i686-pc-windows-msvc-shared-install_only.tar.gz
|
33 |
+
set DL_DST="%INSTALL_DIR%python.tar.gz"
|
34 |
+
curl -L -o %DL_DST% %DL_URL%
|
35 |
+
if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
|
36 |
+
tar -xzf %DL_DST%
|
37 |
+
|
38 |
+
set PYTHON=%INSTALL_DIR%python\python.exe
|
39 |
+
set PATH=%PATH%;%INSTALL_DIR%python310\Scripts
|
40 |
+
|
41 |
+
:install_venv
|
42 |
+
cd %APPDIR%
|
43 |
+
%PYTHON% -m venv %VENV%
|
44 |
+
set PYTHON=%VENV%\Scripts\python.exe
|
45 |
+
|
46 |
+
:install_pip
|
47 |
+
set DL_URL=https://bootstrap.pypa.io/get-pip.py
|
48 |
+
set DL_DST=%INSTALL_DIR%get-pip.py
|
49 |
+
curl -o %DL_DST% %DL_URL%
|
50 |
+
if not exist %DL_DST% bitsadmin /transfer dl %DL_URL% %DL_DST%
|
51 |
+
%PYTHON% %DL_DST%
|
52 |
+
|
53 |
+
%PYTHON% -m pip install gradio
|
54 |
+
%PYTHON% -m pip install -r requirements.txt
|
55 |
+
|
56 |
+
pause
|
main.py
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import os
|
2 |
+
import sys
|
3 |
+
import time
|
4 |
+
import signal
|
5 |
+
import psutil
|
6 |
+
import io
|
7 |
+
|
8 |
+
from fastapi import FastAPI, Request, status, Form, UploadFile
|
9 |
+
from fastapi.staticfiles import StaticFiles
|
10 |
+
from fastapi.middleware.cors import CORSMiddleware
|
11 |
+
from pydantic import BaseModel, Field
|
12 |
+
from fastapi.exceptions import RequestValidationError
|
13 |
+
from fastapi.responses import JSONResponse
|
14 |
+
|
15 |
+
import fn
|
16 |
+
import gradio as gr
|
17 |
+
from app import demo
|
18 |
+
|
19 |
+
app = FastAPI()
|
20 |
+
|
21 |
+
app.add_middleware(
|
22 |
+
CORSMiddleware,
|
23 |
+
allow_origins=['*'],
|
24 |
+
allow_credentials=True,
|
25 |
+
allow_methods=["*"],
|
26 |
+
allow_headers=["*"],
|
27 |
+
)
|
28 |
+
|
29 |
+
gr.mount_gradio_app(app, demo, path="/gradio")
|
30 |
+
|
31 |
+
fn.load_model()
|
32 |
+
|
33 |
+
@app.post("/raw1")
|
34 |
+
async def api_raw1(file: UploadFile = Form(...)):
|
35 |
+
try:
|
36 |
+
file_content = await file.read()
|
37 |
+
file_stream = io.BytesIO(file_content)
|
38 |
+
|
39 |
+
raw1, raw2 = fn.raw(file_stream)
|
40 |
+
|
41 |
+
return Response(content=raw1, media_type="audio/wav")
|
42 |
+
except Exception as e:
|
43 |
+
return {"error": str(e)}
|
44 |
+
|
45 |
+
@app.post("/raw2")
|
46 |
+
async def api_raw2(file: UploadFile = Form(...)):
|
47 |
+
try:
|
48 |
+
file_content = await file.read()
|
49 |
+
file_stream = io.BytesIO(file_content)
|
50 |
+
|
51 |
+
raw1, raw2 = fn.raw(file_stream)
|
52 |
+
|
53 |
+
return Response(content=raw2, media_type="audio/wav")
|
54 |
+
except Exception as e:
|
55 |
+
return {"error": str(e)}
|
requirements.txt
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
fastapi
|
2 |
+
uvicorn
|
3 |
+
audio-separator
|
4 |
+
python-multipart
|
venv.sh
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/usr/bin/bash
|
2 |
+
|
3 |
+
python3 -m venv venv
|
4 |
+
curl -kL https://bootstrap.pypa.io/get-pip.py | venv/bin/python
|
5 |
+
|
6 |
+
venv/bin/python -m pip install gradio
|
7 |
+
venv/bin/python -m pip install -r requirements.txt
|