File size: 1,390 Bytes
d9690c2 11fad67 d9690c2 11fad67 d9690c2 11fad67 d9690c2 11fad67 d9690c2 11fad67 d9690c2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | import os
import tempfile
from fastapi import FastAPI, UploadFile, File, Header, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from openai import OpenAI
APP_SHARED_SECRET = os.environ.get("APP_SHARED_SECRET", "")
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_methods=["*"],
allow_headers=["*"],
)
client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
def require_secret(x_app_secret: str | None) -> None:
if APP_SHARED_SECRET and x_app_secret != APP_SHARED_SECRET:
raise HTTPException(status_code=401, detail="Unauthorized")
@app.post("/v1/session")
async def session(x_app_secret: str | None = Header(default=None)):
require_secret(x_app_secret)
return {
"token": "hf-dev",
"expiresAt": None,
"requestId": None
}
@app.post("/v1/stt")
async def stt(
audio: UploadFile = File(...),
metadata: str | None = None,
x_app_secret: str | None = Header(default=None)
):
require_secret(x_app_secret)
with tempfile.NamedTemporaryFile(suffix=".m4a") as tmp:
tmp.write(await audio.read())
tmp.flush()
result = client.audio.transcriptions.create(
file=open(tmp.name, "rb"),
model="whisper-1"
)
return {
"transcript": result.text,
"language": "en",
"requestId": None
} |