| 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 |
| } |
