- Create lightsync/audio/waveform.py with extract_peaks() (AUD-05)
- MP3 via ffmpeg pipe, WAV/FLAC/OGG via soundfile+numpy
- Add GET /api/audio/waveform endpoint with asyncio.to_thread
- peaks param clamped 100-5000, returns {peaks, count, file}
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""Audio loading and status endpoints."""
|
|
import asyncio
|
|
from pathlib import Path
|
|
from fastapi import APIRouter, HTTPException, Request
|
|
from pydantic import BaseModel
|
|
from lightsync.audio.waveform import extract_peaks
|
|
|
|
router = APIRouter()
|
|
|
|
SUPPORTED_EXTENSIONS = {".mp3", ".wav", ".flac", ".ogg"}
|
|
|
|
|
|
class LoadRequest(BaseModel):
|
|
path: str
|
|
|
|
|
|
class AudioState(BaseModel):
|
|
position: float
|
|
paused: bool
|
|
duration: float | None
|
|
loaded: bool
|
|
file: str | None
|
|
|
|
|
|
@router.post("/load")
|
|
async def load_audio(req: LoadRequest, request: Request):
|
|
"""Load an audio file by server path (AUD-01). YouTube URL deferred to Phase 6 (AUD-02 stub)."""
|
|
engine = request.app.state.engine
|
|
if engine is None:
|
|
raise HTTPException(status_code=503, detail="Audio engine not ready")
|
|
|
|
p = Path(req.path)
|
|
if not p.exists():
|
|
raise HTTPException(status_code=404, detail=f"File not found: {req.path}")
|
|
if p.suffix.lower() not in SUPPORTED_EXTENSIONS:
|
|
raise HTTPException(status_code=400, detail=f"Unsupported format: {p.suffix}. Supported: {', '.join(sorted(SUPPORTED_EXTENSIONS))}")
|
|
|
|
await asyncio.to_thread(engine.load, req.path)
|
|
return {"status": "loaded", "path": req.path}
|
|
|
|
|
|
@router.get("/state")
|
|
async def get_audio_state(request: Request):
|
|
"""Get current audio playback state."""
|
|
engine = request.app.state.engine
|
|
if engine is None:
|
|
raise HTTPException(status_code=503, detail="Audio engine not ready")
|
|
return engine.get_state()
|
|
|
|
|
|
@router.get("/waveform")
|
|
async def get_waveform(request: Request, peaks: int = 1000):
|
|
"""Get waveform peak data for the currently loaded audio file (AUD-05).
|
|
|
|
Query params:
|
|
peaks: Number of peak samples (default 1000, max 5000)
|
|
"""
|
|
engine = request.app.state.engine
|
|
if engine is None:
|
|
raise HTTPException(status_code=503, detail="Audio engine not ready")
|
|
|
|
state = engine.get_state()
|
|
if not state["loaded"] or not state["file"]:
|
|
raise HTTPException(status_code=400, detail="No audio file loaded")
|
|
|
|
num_peaks = min(max(peaks, 100), 5000)
|
|
|
|
try:
|
|
data = await asyncio.to_thread(extract_peaks, state["file"], num_peaks)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Waveform extraction failed: {e}")
|
|
|
|
return {"peaks": data, "count": len(data), "file": state["file"]}
|