Spaces:

Pepguy
/

voice

Sleeping

App Files Files Community

Pepguy commited on Mar 13, 2025

Commit

a8336db

verified ·

1 Parent(s): 24d52af

Update app.py

Browse files

Files changed (1) hide show

app.py +133 -16

app.py CHANGED Viewed

@@ -1,28 +1,145 @@
-from fastapi import FastAPI
 import subprocess
 app = FastAPI()
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
-@app.get("/ffmpeg-version")
-def get_ffmpeg_version():
-    try:
-        result = subprocess.run(["ffmpeg", "-version"], capture_output=True, text=True)
-        return {"ffmpeg_version": result.stdout.splitlines()[0]}  # Return only the first line
-    except FileNotFoundError:
-        return {"error": "FFmpeg is not installed or not found in PATH"}
-# from fastapi import FastAPI
-# app = FastAPI()
-# @app.get("/")
-# def greet_json():
-#     return {"Hello": "World!"}

+import os
+import uuid
 import subprocess
+import requests
+import base64
+from fastapi import FastAPI, UploadFile, File, Form, HTTPException, Request
+from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
 app = FastAPI()
+# Allow CORS for testing
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+# Base directory to store task files
+BASE_DIR = "tasks"
+os.makedirs(BASE_DIR, exist_ok=True)
+@app.post("/process")
+async def process_audio(
+    request: Request,
+    file: UploadFile = File(None),
+    url: str = Form(None)
+):
+    if not file and not url:
+        raise HTTPException(status_code=400, detail="Provide either a file or a URL.")
+    # Generate unique task ID and create folder for this task
+    task_id = str(uuid.uuid4())
+    task_dir = os.path.join(BASE_DIR, task_id)
+    os.makedirs(task_dir, exist_ok=True)
+    # Download or read the file
+    if file:
+        filename = file.filename
+        input_path = os.path.join(task_dir, filename)
+        with open(input_path, "wb") as f:
+            content = await file.read()
+            f.write(content)
+    else:
+        # Download file from provided URL
+        resp = requests.get(url)
+        if resp.status_code != 200:
+            raise HTTPException(status_code=400, detail="Failed to download file from URL.")
+        ext = os.path.splitext(url)[1] or ".mp3"
+        filename = f"input{ext}"
+        input_path = os.path.join(task_dir, filename)
+        with open(input_path, "wb") as f:
+            f.write(resp.content)
+    # Run Spleeter to separate into 2 stems (vocals and accompaniment)
+    # Spleeter will create a subfolder named after the input file (without extension)
+    cmd = [
+        "spleeter", "separate",
+        "-i", input_path,
+        "-p", "spleeter:2stems",
+        "-o", task_dir
+    ]
+    try:
+        subprocess.run(cmd, check=True)
+    except subprocess.CalledProcessError as e:
+        raise HTTPException(status_code=500, detail=f"Spleeter processing failed: {e}")
+    # Determine output folder name and expected files
+    base_name = os.path.splitext(filename)[0]
+    output_dir = os.path.join(task_dir, base_name)
+    vocals_path = os.path.join(output_dir, "vocals.wav")
+    accomp_path = os.path.join(output_dir, "accompaniment.wav")
+    if not (os.path.exists(vocals_path) and os.path.exists(accomp_path)):
+        raise HTTPException(status_code=500, detail="Output files not found after processing.")
+    # Read output files and encode in base64
+    with open(vocals_path, "rb") as f:
+        vocals_data = f.read()
+    with open(accomp_path, "rb") as f:
+        accomp_data = f.read()
+    vocals_b64 = base64.b64encode(vocals_data).decode("utf-8")
+    accomp_b64 = base64.b64encode(accomp_data).decode("utf-8")
+    return JSONResponse(content={
+        "task_id": task_id,
+        "vocals": vocals_b64,
+        "accompaniment": accomp_b64
+    })
+@app.get("/", response_class=HTMLResponse)
+def index():
+    html_content = """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>Spleeter API Test</title>
+    </head>
+    <body>
+        <h1>Spleeter API Test</h1>
+        <form id="spleeterForm" enctype="multipart/form-data">
+            <div>
+                <label>Upload File:</label>
+                <input type="file" name="file">
+            </div>
+            <div>
+                <label>Or Enter URL:</label>
+                <input type="text" name="url" size="50">
+            </div>
+            <button type="submit">Process Audio</button>
+        </form>
+        <div id="result"></div>
+        <script>
+            const form = document.getElementById('spleeterForm');
+            form.onsubmit = async (e) => {
+                e.preventDefault();
+                const formData = new FormData(form);
+                const response = await fetch('/process', {
+                    method: 'POST',
+                    body: formData
+                });
+                const data = await response.json();
+                if (response.ok) {
+                    const vocalsSrc = "data:audio/wav;base64," + data.vocals;
+                    const accompSrc = "data:audio/wav;base64," + data.accompaniment;
+                    document.getElementById('result').innerHTML = `
+                        <h3>Vocals</h3>
+                        <audio controls src="${vocalsSrc}"></audio>
+                        <h3>Accompaniment</h3>
+                        <audio controls src="${accompSrc}"></audio>
+                    `;
+                } else {
+                    document.getElementById('result').innerHTML = `<p>Error: ${data.detail}</p>`;
+                }
+            };
+        </script>
+    </body>
+    </html>
+    """
+    return HTMLResponse(content=html_content)