|
|
""" |
|
|
Transcriber interface definition |
|
|
""" |
|
|
|
|
|
from abc import ABC, abstractmethod |
|
|
from typing import Dict, Any, List, Optional |
|
|
from dataclasses import dataclass |
|
|
|
|
|
|
|
|
@dataclass |
|
|
class TranscriptionSegment: |
|
|
"""Transcription segment data class""" |
|
|
start: float |
|
|
end: float |
|
|
text: str |
|
|
speaker: Optional[str] = None |
|
|
confidence: Optional[float] = None |
|
|
|
|
|
|
|
|
@dataclass |
|
|
class TranscriptionResult: |
|
|
"""Transcription result data class""" |
|
|
text: str |
|
|
segments: List[TranscriptionSegment] |
|
|
language: str |
|
|
model_used: str |
|
|
audio_duration: float |
|
|
processing_time: float |
|
|
speaker_diarization_enabled: bool = False |
|
|
global_speaker_count: int = 0 |
|
|
error_message: Optional[str] = None |
|
|
|
|
|
|
|
|
class ITranscriber(ABC): |
|
|
"""Interface for audio transcription""" |
|
|
|
|
|
@abstractmethod |
|
|
async def transcribe( |
|
|
self, |
|
|
audio_file_path: str, |
|
|
model_size: str = "turbo", |
|
|
language: Optional[str] = None, |
|
|
enable_speaker_diarization: bool = False |
|
|
) -> TranscriptionResult: |
|
|
""" |
|
|
Transcribe audio file |
|
|
|
|
|
Args: |
|
|
audio_file_path: Path to audio file |
|
|
model_size: Model size to use |
|
|
language: Language code (None for auto-detect) |
|
|
enable_speaker_diarization: Whether to enable speaker detection |
|
|
|
|
|
Returns: |
|
|
TranscriptionResult object |
|
|
""" |
|
|
pass |
|
|
|
|
|
@abstractmethod |
|
|
def get_supported_models(self) -> List[str]: |
|
|
"""Get list of supported model sizes""" |
|
|
pass |
|
|
|
|
|
@abstractmethod |
|
|
def get_supported_languages(self) -> List[str]: |
|
|
"""Get list of supported language codes""" |
|
|
pass |
