mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-25 19:15:18 +02:00
Added local Speech-to-Text (STT) support using Faster-Whisper
This commit is contained in:
parent
402039f02f
commit
dad79674c8
8 changed files with 396 additions and 7 deletions
|
|
@ -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)
|
||||
|
|
|
|||
96
surfsense_backend/app/routes/stt_routes.py
Normal file
96
surfsense_backend/app/routes/stt_routes.py
Normal 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)}"
|
||||
)
|
||||
Loading…
Add table
Add a link
Reference in a new issue