from pydantic import AnyHttpUrl, BaseModel, Field

import app.shared.db.models as models


class PostJobPayload(BaseModel):
    url: AnyHttpUrl = Field(
        description=(
            "URL where the media file is available. This needs to be a direct link."
        )
    )

    type: models.JobType = Field(
        description="""Type of this job.
        `transcript` uses the original language of the audio.
        `translation` creates an automatic translation to english.
        `language_detection` detects language from the first 30 seconds of audio."""
    )

    language: str | None = Field(
        description=(
            "Spoken language in the media file. "
            "While optional, this can improve output when set."
        )
    )