File size: 3,881 Bytes
dbc3c35 5dadf47 dbc3c35 5dadf47 dbc3c35 5dadf47 dbc3c35 89e1dc4 dbc3c35 89e1dc4 dbc3c35 89e1dc4 dbc3c35 89e1dc4 dbc3c35 5dadf47 dbc3c35 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 | from datetime import datetime, timezone
from enum import Enum
from typing import Any, Literal
from pydantic import BaseModel, Field, HttpUrl, field_validator
def utc_now() -> datetime:
return datetime.now(timezone.utc)
class TargetPlatform(str, Enum):
tiktok = "tiktok"
youtube_shorts = "youtube_shorts"
instagram_reels = "instagram_reels"
class ChannelProfile(BaseModel):
niche: str = Field(default="education", min_length=2, max_length=80)
niche_custom: str = Field(default="", max_length=80)
channel_description: str = Field(default="", max_length=700)
clip_style: str = Field(default="informative", min_length=2, max_length=80)
clip_length_seconds: int = Field(default=60, ge=15, le=180)
clip_count: int = Field(default=5, ge=1, le=20)
primary_language: str = Field(default="Thai", min_length=2, max_length=40)
target_platform: TargetPlatform = TargetPlatform.tiktok
@field_validator("niche", "niche_custom", "channel_description", "clip_style", "primary_language")
@classmethod
def clean_text(cls, value: str) -> str:
return value.strip()
class YoutubeJobRequest(BaseModel):
youtube_url: HttpUrl
profile: ChannelProfile
class TranscriptSegment(BaseModel):
id: str
start_seconds: float = Field(ge=0)
end_seconds: float = Field(ge=0)
text: str
language: str | None = None
class SubtitleCue(BaseModel):
"""A single subtitle line with explicit timing relative to clip start."""
start_seconds: float = Field(ge=0)
end_seconds: float = Field(ge=0)
text: str = ""
class SkipRange(BaseModel):
"""A range to splice out of the middle of a clip (relative to clip start)."""
start_seconds: float = Field(ge=0)
end_seconds: float = Field(ge=0)
class ClipCandidate(BaseModel):
id: str
start_seconds: float = Field(ge=0)
end_seconds: float = Field(ge=0)
title: str
reason: str
score: float = Field(ge=0, le=100)
subtitle_text: str = ""
subtitle_cues: list[SubtitleCue] | None = None
skip_ranges: list[SkipRange] | None = None
video_url: str | None = None
download_url: str | None = None
approved: bool = False
deleted: bool = False
metadata: dict[str, Any] = Field(default_factory=dict)
class ClipPatch(BaseModel):
start_seconds: float | None = Field(default=None, ge=0)
end_seconds: float | None = Field(default=None, ge=0)
subtitle_text: str | None = None
subtitle_cues: list[SubtitleCue] | None = None
skip_ranges: list[SkipRange] | None = None
approved: bool | None = None
deleted: bool | None = None
class RegenerateClipRequest(BaseModel):
clip_style: str | None = None
clip_length_seconds: int | None = Field(default=None, ge=15, le=180)
subtitle_text: str | None = None
class TranslateSubtitlesRequest(BaseModel):
target_language: str = Field(min_length=2, max_length=40)
class PolishSubtitlesRequest(BaseModel):
style: str | None = None
class JobSnapshot(BaseModel):
id: str
status: Literal["queued", "running", "completed", "failed"]
progress: float = Field(ge=0, le=1)
message: str
current_step: str = ""
step_index: int = Field(default=0, ge=0)
step_total: int = Field(default=6, ge=1)
active_clip_index: int = Field(default=0, ge=0)
active_clip_total: int = Field(default=0, ge=0)
source: dict[str, Any]
profile: ChannelProfile
transcript: list[TranscriptSegment] = Field(default_factory=list)
clips: list[ClipCandidate] = Field(default_factory=list)
timings: dict[str, float] = Field(default_factory=dict)
error: str | None = None
created_at: datetime = Field(default_factory=utc_now)
updated_at: datetime = Field(default_factory=utc_now)
class HealthResponse(BaseModel):
ok: bool
app: str
demo_mode: bool
accelerator: dict[str, Any]
|