60 lines
1.3 KiB
Python
60 lines
1.3 KiB
Python
import os
|
|
from contextlib import asynccontextmanager
|
|
from pathlib import Path
|
|
|
|
from fastapi import FastAPI, HTTPException, BackgroundTasks
|
|
from fastapi.responses import FileResponse
|
|
from pydantic import BaseModel
|
|
|
|
from app import downloader
|
|
|
|
|
|
STATIC_DIR = Path(__file__).parent / "static"
|
|
AUDIO_TMP_DIR = Path("/tmp/apoena-audio")
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
AUDIO_TMP_DIR.mkdir(parents=True, exist_ok=True)
|
|
yield
|
|
|
|
|
|
app = FastAPI(title="apoena-transcript", lifespan=lifespan)
|
|
|
|
|
|
class ExtractAudioRequest(BaseModel):
|
|
url: str
|
|
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {"status": "ok"}
|
|
|
|
|
|
@app.get("/worker.js")
|
|
async def worker_js():
|
|
return FileResponse(STATIC_DIR / "worker.js", media_type="application/javascript")
|
|
|
|
|
|
@app.post("/extract-audio")
|
|
async def extract_audio(body: ExtractAudioRequest, background_tasks: BackgroundTasks):
|
|
try:
|
|
audio_path = await downloader.extract_audio(body.url)
|
|
except RuntimeError as e:
|
|
raise HTTPException(status_code=422, detail=str(e))
|
|
|
|
background_tasks.add_task(_delete_file, audio_path)
|
|
return FileResponse(audio_path, media_type="audio/mpeg", filename="audio.mp3")
|
|
|
|
|
|
def _delete_file(path):
|
|
try:
|
|
os.unlink(path)
|
|
except OSError:
|
|
pass
|
|
|
|
|
|
@app.get("/")
|
|
async def index():
|
|
return FileResponse(STATIC_DIR / "index.html")
|