Added local Speech-to-Text (STT) support using Faster-Whisper

This commit is contained in:
Nabhan 2025-10-11 23:56:12 +05:00
parent 402039f02f
commit dad79674c8
8 changed files with 396 additions and 7 deletions

View file

@ -17,6 +17,7 @@ from .luma_add_connector_route import router as luma_add_connector_router
from .podcasts_routes import router as podcasts_router
from .search_source_connectors_routes import router as search_source_connectors_router
from .search_spaces_routes import router as search_spaces_router
from .stt_routes import router as stt_router
router = APIRouter()
@ -31,3 +32,4 @@ router.include_router(airtable_add_connector_router)
router.include_router(luma_add_connector_router)
router.include_router(llm_config_router)
router.include_router(logs_router)
router.include_router(stt_router)

View file

@ -0,0 +1,96 @@
"""Speech-to-Text API routes."""
from fastapi import APIRouter, File, Form, HTTPException, UploadFile
from fastapi.responses import JSONResponse
from app.services.stt_service import stt_service
router = APIRouter(prefix="/stt", tags=["Speech-to-Text"])
@router.post("/transcribe")
async def transcribe_audio(
audio: UploadFile = File(..., description="Audio file to transcribe"),
language: str = Form(None, description="Optional language code (e.g., 'en', 'es')"),
):
"""Transcribe uploaded audio file to text."""
# Validate file type
if not audio.content_type or not audio.content_type.startswith("audio/"):
raise HTTPException(
status_code=400,
detail="File must be an audio file"
)
try:
# Read audio bytes
audio_bytes = await audio.read()
# Transcribe
result = stt_service.transcribe_bytes(
audio_bytes,
filename=audio.filename or "audio.wav",
language=language if language else None
)
return JSONResponse(content={
"success": True,
"transcription": result["text"],
"metadata": {
"detected_language": result["language"],
"language_probability": result["language_probability"],
"duration_seconds": result["duration"],
"model_size": stt_service.model_size,
}
})
except Exception as e:
raise HTTPException(
status_code=500,
detail=f"Transcription failed: {str(e)}"
)
@router.get("/models")
async def get_available_models():
"""Get list of available Whisper models."""
return JSONResponse(content={
"models": [
{"name": "tiny", "size": "~39 MB", "speed": "fastest", "accuracy": "lowest"},
{"name": "base", "size": "~74 MB", "speed": "fast", "accuracy": "good"},
{"name": "small", "size": "~244 MB", "speed": "medium", "accuracy": "better"},
{"name": "medium", "size": "~769 MB", "speed": "slow", "accuracy": "high"},
{"name": "large-v3", "size": "~1550 MB", "speed": "slowest", "accuracy": "highest"},
],
"current_model": stt_service.model_size,
"note": "Models are downloaded automatically on first use"
})
@router.post("/change-model")
async def change_model(model_size: str = Form(...)):
"""Change the active Whisper model."""
valid_models = ["tiny", "base", "small", "medium", "large-v3"]
if model_size not in valid_models:
raise HTTPException(
status_code=400,
detail=f"Invalid model. Choose from: {valid_models}"
)
try:
# Create new service instance with different model
global stt_service
stt_service = type(stt_service)(model_size=model_size)
return JSONResponse(content={
"success": True,
"message": f"Model changed to {model_size}",
"note": "Model will be downloaded on next transcription if not cached"
})
except Exception as e:
raise HTTPException(
status_code=500,
detail=f"Failed to change model: {str(e)}"
)