66be55ba8a
- korean_preprocessor: 발음 사전 176 → 206개 (200+ 달성) - video_engine: SoraEngine 완전 제거 (2026-03-24 서비스 종료) - smart_video_router: veo3/seedance2 빈 문자열 반환 → ffmpeg_slides 폴백 - cli/init: gemini_web 서비스 설정 질문 추가 (user_profile 일치) - caption_renderer, tts_engine, video_assembler: --test 스탠드얼론 블록 추가 Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
433 lines
15 KiB
Python
433 lines
15 KiB
Python
"""
|
||
bots/shorts/caption_renderer.py
|
||
역할: 단어별 타임스탬프 → ASS 자막 파일 생성 (단어별 하이라이트)
|
||
|
||
스타일:
|
||
- 기본: 흰색 볼드, 검정 아웃라인 3px
|
||
- 하이라이트: 노란색 (#FFD700) — 현재 발음 중인 단어
|
||
- 훅 텍스트: 중앙 상단, 72px, 1.5초 표시
|
||
- 본문 자막: 하단 200px, 48px, 최대 2줄
|
||
|
||
출력:
|
||
data/shorts/captions/{timestamp}.ass
|
||
"""
|
||
import json
|
||
import logging
|
||
from pathlib import Path
|
||
from typing import Optional
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
BASE_DIR = Path(__file__).parent.parent.parent
|
||
|
||
CAPTION_TEMPLATES = {
|
||
'hormozi': {
|
||
'font_size': 64,
|
||
'highlight_color': '#FFD700',
|
||
'animation': 'pop_in',
|
||
'position': 'center',
|
||
'outline_width': 4,
|
||
'auto_emoji': False,
|
||
},
|
||
'tiktok_viral': {
|
||
'font_size': 56,
|
||
'highlight_color': '#FF6B6B',
|
||
'animation': 'bounce',
|
||
'auto_emoji': True,
|
||
'position': 'center_bottom',
|
||
},
|
||
'brand_4thpath': {
|
||
'font_size': 52,
|
||
'highlight_color': '#00D4FF',
|
||
'animation': 'typewriter',
|
||
'position': 'center',
|
||
'overlay_gradient': True,
|
||
},
|
||
}
|
||
|
||
# Corner → caption template mapping
|
||
CORNER_CAPTION_MAP = {
|
||
'쉬운세상': 'hormozi',
|
||
'숨은보물': 'tiktok_viral',
|
||
'바이브리포트': 'hormozi',
|
||
'팩트체크': 'brand_4thpath',
|
||
'한컷': 'tiktok_viral',
|
||
'웹소설': 'brand_4thpath',
|
||
}
|
||
|
||
|
||
def smart_line_break(text: str, max_chars: int = 18) -> list[str]:
|
||
"""
|
||
Break Korean text at semantic boundaries, not mid-word.
|
||
Never break before 조사 (particles) or 어미 (endings).
|
||
|
||
Returns list of line strings.
|
||
"""
|
||
# Common Korean particles/endings that should not start a new line
|
||
PARTICLES = ['은', '는', '이', '가', '을', '를', '의', '에', '에서', '으로', '로',
|
||
'과', '와', '도', '만', '까지', '부터', '보다', '처럼', '같이',
|
||
'한테', '에게', '이라', '라고', '이고', '이며', '고', '며', '면',
|
||
'이면', '이나', '나', '든지', '거나', '지만', '이지만', '지만',
|
||
'니까', '으니까', '이니까', '서', '아서', '어서', '며', '고']
|
||
|
||
if len(text) <= max_chars:
|
||
return [text] if text else []
|
||
|
||
lines = []
|
||
remaining = text
|
||
|
||
while len(remaining) > max_chars:
|
||
# Find best break point near max_chars
|
||
break_at = max_chars
|
||
|
||
# Look for space or punctuation near the limit
|
||
for i in range(max_chars, max(0, max_chars - 6), -1):
|
||
if i >= len(remaining):
|
||
continue
|
||
char = remaining[i]
|
||
prev_char = remaining[i-1] if i > 0 else ''
|
||
next_char = remaining[i+1] if i+1 < len(remaining) else ''
|
||
|
||
# Break at space
|
||
if char == ' ':
|
||
# Check if next word starts with a particle
|
||
next_word = remaining[i+1:i+4]
|
||
is_particle_start = any(next_word.startswith(p) for p in PARTICLES)
|
||
if not is_particle_start:
|
||
break_at = i
|
||
break
|
||
|
||
# Break after punctuation
|
||
if prev_char in ('。', '，', ',', '.', '!', '?', '~'):
|
||
break_at = i
|
||
break
|
||
|
||
lines.append(remaining[:break_at].strip())
|
||
remaining = remaining[break_at:].strip()
|
||
|
||
if remaining:
|
||
lines.append(remaining)
|
||
|
||
return [l for l in lines if l]
|
||
|
||
|
||
def get_template_for_corner(corner: str) -> dict:
|
||
"""
|
||
Get caption template config for a given content corner.
|
||
Falls back to 'hormozi' template if corner not in map.
|
||
"""
|
||
template_name = CORNER_CAPTION_MAP.get(corner, 'hormozi')
|
||
return CAPTION_TEMPLATES.get(template_name, CAPTION_TEMPLATES['hormozi'])
|
||
|
||
|
||
def _load_config() -> dict:
|
||
cfg_path = BASE_DIR / 'config' / 'shorts_config.json'
|
||
if cfg_path.exists():
|
||
return json.loads(cfg_path.read_text(encoding='utf-8'))
|
||
return {}
|
||
|
||
|
||
# ─── 색상 변환 ────────────────────────────────────────────────
|
||
|
||
def _hex_to_ass(hex_color: str, alpha: int = 0) -> str:
|
||
"""
|
||
HTML hex (#RRGGBB) → ASS 색상 &HAABBGGRR 변환.
|
||
ASS는 BGR 순서이며 alpha는 00(불투명)~FF(투명).
|
||
"""
|
||
c = hex_color.lstrip('#')
|
||
r, g, b = c[0:2], c[2:4], c[4:6]
|
||
return f'&H{alpha:02X}{b}{g}{r}'
|
||
|
||
|
||
# ─── 시간 포맷 ────────────────────────────────────────────────
|
||
|
||
def _sec_to_ass_time(seconds: float) -> str:
|
||
"""초(float) → ASS 시간 포맷 H:MM:SS.cc."""
|
||
cs = int(round(seconds * 100))
|
||
h = cs // 360000
|
||
cs %= 360000
|
||
m = cs // 6000
|
||
cs %= 6000
|
||
s = cs // 100
|
||
cs %= 100
|
||
return f'{h}:{m:02d}:{s:02d}.{cs:02d}'
|
||
|
||
|
||
# ─── ASS 헤더 ────────────────────────────────────────────────
|
||
|
||
def _ass_header(cfg: dict) -> str:
|
||
cap_cfg = cfg.get('caption', {})
|
||
font_ko = cap_cfg.get('font_ko', 'Pretendard')
|
||
font_size = cap_cfg.get('font_size', 48)
|
||
hook_size = cap_cfg.get('hook_font_size', 72)
|
||
default_color = _hex_to_ass(cap_cfg.get('default_color', '#FFFFFF'))
|
||
highlight_color = _hex_to_ass(cap_cfg.get('highlight_color', '#FFD700'))
|
||
outline_color = _hex_to_ass(cap_cfg.get('outline_color', '#000000'))
|
||
outline_w = cap_cfg.get('outline_width', 3)
|
||
margin_v = cap_cfg.get('position_from_bottom', 200)
|
||
|
||
return f"""[Script Info]
|
||
ScriptType: v4.00+
|
||
PlayResX: 1080
|
||
PlayResY: 1920
|
||
ScaledBorderAndShadow: yes
|
||
|
||
[V4+ Styles]
|
||
Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
|
||
Style: Default,{font_ko},{font_size},{default_color},{default_color},{outline_color},&H80000000,-1,0,0,0,100,100,0,0,1,{outline_w},1,2,20,20,{margin_v},1
|
||
Style: Highlight,{font_ko},{font_size},{highlight_color},{highlight_color},{outline_color},&H80000000,-1,0,0,0,100,100,0,0,1,{outline_w},1,2,20,20,{margin_v},1
|
||
Style: Hook,{font_ko},{hook_size},{default_color},{default_color},{outline_color},&H80000000,-1,0,0,0,100,100,0,0,1,{outline_w+1},2,5,20,20,100,1
|
||
|
||
[Events]
|
||
Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
|
||
"""
|
||
|
||
|
||
# ─── 단어 → 자막 라인 분할 ────────────────────────────────────
|
||
|
||
def _split_into_lines(words: list[dict], max_chars: int = 18) -> list[list[dict]]:
|
||
"""
|
||
단어 리스트 → 라인 리스트 (최대 max_chars 자).
|
||
반환: [[{word, start, end}, ...], ...]
|
||
"""
|
||
lines = []
|
||
cur_line: list[dict] = []
|
||
cur_len = 0
|
||
|
||
for w in words:
|
||
word_text = w['word']
|
||
if cur_line and cur_len + len(word_text) + 1 > max_chars:
|
||
lines.append(cur_line)
|
||
cur_line = [w]
|
||
cur_len = len(word_text)
|
||
else:
|
||
cur_line.append(w)
|
||
cur_len += len(word_text) + (1 if cur_line else 0)
|
||
|
||
if cur_line:
|
||
lines.append(cur_line)
|
||
|
||
return lines
|
||
|
||
|
||
def _line_start_end(line: list[dict]) -> tuple[float, float]:
|
||
return line[0]['start'], line[-1]['end']
|
||
|
||
|
||
# ─── ASS 이벤트 생성 ─────────────────────────────────────────
|
||
|
||
def _word_highlight_event(
|
||
line: list[dict],
|
||
highlight_color_hex: str,
|
||
default_color_hex: str,
|
||
outline_color_hex: str,
|
||
outline_w: int,
|
||
) -> str:
|
||
"""
|
||
한 라인의 모든 단어에 대해 단어별 하이라이트 오버라이드 태그 생성.
|
||
각 단어 재생 시간 동안 해당 단어만 highlight_color로 표시.
|
||
ASS override tag: {\\c&Hxxxxxx&} 로 색상 변경.
|
||
|
||
반환: 단어별 ASS 이벤트 문자열 목록
|
||
"""
|
||
hi_ass = _hex_to_ass(highlight_color_hex)
|
||
df_ass = _hex_to_ass(default_color_hex)
|
||
|
||
events = []
|
||
for i, w in enumerate(line):
|
||
start_t = w['start']
|
||
end_t = w['end']
|
||
|
||
# 전체 라인 텍스트: 현재 단어만 하이라이트
|
||
parts = []
|
||
for j, other in enumerate(line):
|
||
if j == i:
|
||
parts.append(f'{{\\c{hi_ass}}}{other["word"]}{{\\c{df_ass}}}')
|
||
else:
|
||
parts.append(other['word'])
|
||
text = ' '.join(parts)
|
||
|
||
event = (
|
||
f'Dialogue: 0,{_sec_to_ass_time(start_t)},{_sec_to_ass_time(end_t)},'
|
||
f'Default,,0,0,0,,{text}'
|
||
)
|
||
events.append(event)
|
||
|
||
return '\n'.join(events)
|
||
|
||
|
||
def _hook_event(hook_text: str, duration: float = 1.5) -> str:
|
||
"""훅 텍스트 — 중앙 상단, 72px, 1.5초 표시."""
|
||
return (
|
||
f'Dialogue: 1,{_sec_to_ass_time(0.0)},{_sec_to_ass_time(duration)},'
|
||
f'Hook,,0,0,0,,{hook_text}'
|
||
)
|
||
|
||
|
||
# ─── 균등 분할 타임스탬프 폴백 ───────────────────────────────
|
||
|
||
def _build_uniform_timestamps(script: dict, total_duration: float) -> list[dict]:
|
||
"""
|
||
Whisper 타임스탬프 없을 때 텍스트를 균등 시간으로 분할.
|
||
"""
|
||
parts = [script.get('hook', '')]
|
||
parts.extend(script.get('body', []))
|
||
parts.append(script.get('closer', ''))
|
||
text = ' '.join(p for p in parts if p)
|
||
words = text.split()
|
||
|
||
if not words:
|
||
return []
|
||
|
||
dur_per_word = total_duration / len(words)
|
||
return [
|
||
{
|
||
'word': w,
|
||
'start': round(i * dur_per_word, 3),
|
||
'end': round((i + 1) * dur_per_word, 3),
|
||
}
|
||
for i, w in enumerate(words)
|
||
]
|
||
|
||
|
||
# ─── 메인 엔트리포인트 ────────────────────────────────────────
|
||
|
||
def render_captions(
|
||
script: dict,
|
||
timestamps: list[dict],
|
||
output_dir: Path,
|
||
timestamp: str,
|
||
wav_duration: float = 0.0,
|
||
cfg: Optional[dict] = None,
|
||
corner: str = '',
|
||
) -> Path:
|
||
"""
|
||
스크립트 + 단어별 타임스탬프 → ASS 자막 파일 생성.
|
||
|
||
Args:
|
||
script: {hook, body, closer, ...}
|
||
timestamps: [{word, start, end}, ...] — 비어있으면 균등 분할
|
||
output_dir: data/shorts/captions/
|
||
timestamp: 파일명 prefix
|
||
wav_duration: TTS 오디오 총 길이 (균등 분할 폴백용)
|
||
cfg: shorts_config.json dict
|
||
corner: content corner name (e.g. '쉬운세상') for template selection
|
||
|
||
Returns:
|
||
ass_path
|
||
"""
|
||
if cfg is None:
|
||
cfg = _load_config()
|
||
|
||
output_dir.mkdir(parents=True, exist_ok=True)
|
||
ass_path = output_dir / f'{timestamp}.ass'
|
||
|
||
cap_cfg = cfg.get('caption', {})
|
||
|
||
# Apply corner-specific template overrides if corner is provided
|
||
if corner:
|
||
template = get_template_for_corner(corner)
|
||
# Override cfg caption section with template values
|
||
cap_cfg = dict(cap_cfg) # make a shallow copy to avoid mutating original
|
||
if 'font_size' in template:
|
||
cap_cfg['font_size'] = template['font_size']
|
||
if 'highlight_color' in template:
|
||
cap_cfg['highlight_color'] = template['highlight_color']
|
||
if 'outline_width' in template:
|
||
cap_cfg['outline_width'] = template['outline_width']
|
||
logger.info(f'[캡션] 코너 "{corner}" → 템플릿 적용: {template}')
|
||
|
||
max_chars = cap_cfg.get('max_chars_per_line_ko', 18)
|
||
highlight_color = cap_cfg.get('highlight_color', '#FFD700')
|
||
default_color = cap_cfg.get('default_color', '#FFFFFF')
|
||
outline_color = cap_cfg.get('outline_color', '#000000')
|
||
outline_w = cap_cfg.get('outline_width', 3)
|
||
|
||
# 타임스탬프 없으면 균등 분할
|
||
if not timestamps:
|
||
logger.warning('단어별 타임스탬프 없음 — 균등 분할 사용 (캡션 품질 저하)')
|
||
if wav_duration <= 0:
|
||
wav_duration = 20.0
|
||
timestamps = _build_uniform_timestamps(script, wav_duration)
|
||
|
||
# ASS 헤더 (rebuild cfg with updated cap_cfg so header reflects template overrides)
|
||
effective_cfg = dict(cfg)
|
||
effective_cfg['caption'] = cap_cfg
|
||
header = _ass_header(effective_cfg)
|
||
events = []
|
||
|
||
# 훅 이벤트 (첫 1.5초 중앙 표시)
|
||
hook_text = script.get('hook', '')
|
||
if hook_text and timestamps:
|
||
hook_end = min(1.5, timestamps[0]['start'] + 1.5) if timestamps else 1.5
|
||
events.append(_hook_event(hook_text, hook_end))
|
||
|
||
# 단어별 하이라이트 이벤트
|
||
lines = _split_into_lines(timestamps, max_chars)
|
||
for line in lines:
|
||
if not line:
|
||
continue
|
||
line_event = _word_highlight_event(
|
||
line, highlight_color, default_color, outline_color, outline_w
|
||
)
|
||
events.append(line_event)
|
||
|
||
ass_content = header + '\n'.join(events) + '\n'
|
||
ass_path.write_text(ass_content, encoding='utf-8-sig') # BOM for Windows compatibility
|
||
logger.info(f'ASS 자막 생성: {ass_path.name} ({len(timestamps)}단어, {len(lines)}라인)')
|
||
|
||
|
||
# ── Standalone test ──────────────────────────────────────────────
|
||
|
||
if __name__ == '__main__':
|
||
import sys
|
||
import tempfile
|
||
from pathlib import Path
|
||
|
||
if '--test' not in sys.argv:
|
||
print("사용법: python -m bots.shorts.caption_renderer --test")
|
||
sys.exit(0)
|
||
|
||
print("=== Caption Renderer Test ===")
|
||
|
||
# Test smart_line_break
|
||
test_texts = [
|
||
("AI를 활용한 자동화 방법입니다", 18),
|
||
("단 3가지만 알면 됩니다", 12),
|
||
]
|
||
print("\n[1] smart_line_break:")
|
||
for text, max_c in test_texts:
|
||
lines = smart_line_break(text, max_c)
|
||
print(f" 입력: {text!r}")
|
||
print(f" 결과: {lines}")
|
||
|
||
# Test template lookup
|
||
print("\n[2] get_template_for_corner:")
|
||
for corner in ['쉬운세상', '숨은보물', '팩트체크', '없는코너']:
|
||
tpl = get_template_for_corner(corner)
|
||
print(f" {corner}: font_size={tpl.get('font_size')}, animation={tpl.get('animation')}")
|
||
|
||
# Test render_captions with dummy timestamps
|
||
print("\n[3] render_captions (dry-run):")
|
||
sample_timestamps = [
|
||
{'word': '이거', 'start': 0.0, 'end': 0.3},
|
||
{'word': '모르면', 'start': 0.4, 'end': 0.8},
|
||
{'word': '손해입니다', 'start': 0.9, 'end': 1.5},
|
||
]
|
||
sample_script = {'hook': '이거 모르면 손해입니다'}
|
||
with tempfile.TemporaryDirectory() as tmpdir:
|
||
out = Path(tmpdir) / 'test.ass'
|
||
render_captions(
|
||
timestamps=sample_timestamps,
|
||
script=sample_script,
|
||
output_path=out,
|
||
corner='쉬운세상',
|
||
)
|
||
exists = out.exists()
|
||
size = out.stat().st_size if exists else 0
|
||
print(f" ASS 파일 생성: {exists}, 크기: {size}bytes")
|
||
assert exists and size > 0, "ASS 파일 생성 실패"
|
||
|
||
print("\n✅ 모든 테스트 통과")
|
||
return ass_path
|