mirror of
https://github.com/open-webui/open-webui
synced 2024-11-16 21:42:58 +00:00
85 lines
2.0 KiB
Python
85 lines
2.0 KiB
Python
import os
|
|
from fastapi import (
|
|
FastAPI,
|
|
Request,
|
|
Depends,
|
|
HTTPException,
|
|
status,
|
|
UploadFile,
|
|
File,
|
|
Form,
|
|
)
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from faster_whisper import WhisperModel
|
|
|
|
from constants import ERROR_MESSAGES
|
|
from utils.utils import (
|
|
decode_token,
|
|
get_current_user,
|
|
get_verified_user,
|
|
get_admin_user,
|
|
)
|
|
from utils.misc import calculate_sha256
|
|
|
|
from config import CACHE_DIR, UPLOAD_DIR, WHISPER_MODEL, WHISPER_MODEL_DIR, DEVICE_TYPE
|
|
|
|
if DEVICE_TYPE != "cuda":
|
|
whisper_device_type = "cpu"
|
|
|
|
|
|
app = FastAPI()
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
|
|
@app.post("/transcribe")
|
|
def transcribe(
|
|
file: UploadFile = File(...),
|
|
user=Depends(get_current_user),
|
|
):
|
|
print(file.content_type)
|
|
|
|
if file.content_type not in ["audio/mpeg", "audio/wav"]:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail=ERROR_MESSAGES.FILE_NOT_SUPPORTED,
|
|
)
|
|
|
|
try:
|
|
filename = file.filename
|
|
file_path = f"{UPLOAD_DIR}/{filename}"
|
|
contents = file.file.read()
|
|
with open(file_path, "wb") as f:
|
|
f.write(contents)
|
|
f.close()
|
|
|
|
model = WhisperModel(
|
|
WHISPER_MODEL,
|
|
device=whisper_device_type,
|
|
compute_type="int8",
|
|
download_root=WHISPER_MODEL_DIR,
|
|
)
|
|
|
|
segments, info = model.transcribe(file_path, beam_size=5)
|
|
print(
|
|
"Detected language '%s' with probability %f"
|
|
% (info.language, info.language_probability)
|
|
)
|
|
|
|
transcript = "".join([segment.text for segment in list(segments)])
|
|
|
|
return {"text": transcript.strip()}
|
|
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail=ERROR_MESSAGES.DEFAULT(e),
|
|
)
|