mirror of
https://github.com/bellingcat/whisperbox-transcribe.git
synced 2026-06-13 05:58:35 +03:00
27 lines
685 B
Python
27 lines
685 B
Python
from abc import ABC
|
|
from typing import Any, Protocol, Tuple
|
|
from uuid import UUID
|
|
|
|
import app.shared.db.models as models
|
|
|
|
TaskReturnValue = Tuple[models.ArtifactType, Any]
|
|
|
|
|
|
class TaskProtocol(Protocol):
|
|
def __call__(self, job: models.Job) -> TaskReturnValue:
|
|
...
|
|
|
|
|
|
class BaseStrategy(ABC):
|
|
def transcribe(self, job: models.Job) -> TaskReturnValue:
|
|
raise NotImplementedError()
|
|
|
|
def translate(self, job: models.Job) -> TaskReturnValue:
|
|
raise NotImplementedError()
|
|
|
|
def detect_language(self, job: models.Job) -> TaskReturnValue:
|
|
raise NotImplementedError()
|
|
|
|
def cleanup(self, job_id: UUID) -> None:
|
|
raise NotImplementedError()
|