# -*- coding: utf-8 -*-
"""
animation_samgukji 생성 단계 (나레이션 + 스토리보드) — 병합본
==================================================================
삼국지 인물 '썰' CSV → ① 병맛 마케팅 나레이션 ② 장면 분할 스토리보드 를 한 파일에 모음.
(animation(조선왕) 폴더와 동일 구조 — 소재만 '조선왕' → '삼국지 인물'.)
scraper.py 가 만든 samguk_keywords_*.csv 를 입력으로 한다.

  · 나레이션:  latest_csv / load_rows / pick_seed / generate
  · 스토리보드: clamp_scenes / make_storyboard

구조: 훅 → 썰 전개(역사 + 병맛 과장) → 어거지 전환 → 마케팅 + CTA → 장면별 분할(자막+그림 프롬프트).
"""

import csv
import json
import random
import re
import sys
from pathlib import Path

from google.genai import types

HERE = Path(__file__).parent
DATA_DIR = HERE / "data"          # 삼국지 인물 일화 CSV(선행 데이터). ★ output/ 이 아님 — 거긴 출력물(mp4)만.
DEFAULT_MODEL = "gemini-2.5-flash"

# 자막 폰트(주아/송명/맑은고딕)에 없는 이모지·기호(😊▼★♥ 등)는 화면에서 □ 로 깨진다 → 제거.
_TYPO = {"…": "...", "‘": "'", "’": "'", "“": '"',
         "”": '"', "·": " ", "～": "~", "〜": "~"}
# 폰트(주아/송명/맑은고딕)에 확실히 있는 것 = 한글 음절·자모(ㅋㅋ ㅠㅠ) + ASCII 인쇄문자. 그 외(이모지·…·특수기호)는 제거.
_KEEP_RE = re.compile(r"[^가-힣ㄱ-ㅣ\x20-\x7E]")


def _no_emoji(s):
    s = s or ""
    for _k, _v in _TYPO.items():
        s = s.replace(_k, _v)
    return re.sub(r"  +", " ", _KEEP_RE.sub("", s)).strip()


# ============================== ① 나레이션 (삼국지 썰 → 병맛 대본) ==============================

def latest_csv() -> Path:
    files = sorted(DATA_DIR.glob("samguk_keywords_*.csv"))
    if not files:
        sys.exit("[!] animation_samgukji/data/ 에 samguk_keywords_*.csv 가 없음. 먼저 'python scraper.py' 실행.")
    return files[-1]


def load_rows(csv_path: Path):
    """CSV → [{order, king, doc, is_anecdote, keyword, url}, ...] (king = 삼국지 인물 이름)"""
    rows = []
    with csv_path.open(encoding="utf-8-sig", newline="") as f:
        r = csv.reader(f)
        next(r, None)  # 헤더 스킵
        for line in r:
            if len(line) < 6:
                continue
            order, king, doc, anec, kw, url = line[:6]
            rows.append({
                "order": int(order) if order.isdigit() else 0,
                "king": king, "doc": doc,
                "is_anecdote": anec.strip().upper() == "Y",
                "keyword": kw, "url": url,
            })
    return rows


def pick_seed(rows, king=None, keyword=None, prefer_anecdote=True):
    """대본의 씨앗이 될 (인물, 썰키워드) 한 건 선택."""
    cand = rows
    if king:
        cand = [r for r in cand if r["king"] == king]
        if not cand:
            kings = sorted({r["king"] for r in rows})
            sys.exit(f"[!] '{king}' 없음. 가능: {', '.join(kings)}")
    if keyword:
        kw_match = [r for r in cand if keyword in r["keyword"]]
        if not kw_match:
            sys.exit(f"[!] '{keyword}' 키워드가 {king or '전체'}에 없음.")
        return random.choice(kw_match)
    if prefer_anecdote:
        anec = [r for r in cand if r["is_anecdote"]]
        if anec:
            cand = anec
    return random.choice(cand)


NARR_SYSTEM = """\
너는 한국 숏폼(쇼츠/릴스) 전문 카피라이터다. '병맛(어이없고 황당하지만 묘하게 중독성 있는)' 유머 톤으로,
삼국지(중국 후한 말~삼국시대) 인물들의 '썰'을 풀다가 마지막에 자연스러운 척 억지로 광고로 갈아타는 {seconds}초 나레이션 대본을 쓴다.

[톤 규칙]
- 구어체 반말 나레이션. TTS(AI 음성)가 읽을 것을 전제로, 문장은 짧고 리듬감 있게.
- 진지하게 시작했다가 갑자기 헛소리로 빠지는 낙차가 핵심. "근데", "그래서", "참고로" 같은 접속사로 능청스럽게 전환.
- 역사적 사실(인물 이름, 사건 키워드)은 최소한의 뼈대로만 쓰고, 디테일은 과장·왜곡해서 웃기게 채운다.
  삼국지연의(소설) 속 유명 장면(도원결의, 삼고초려, 적벽대전, 장판파 등)을 드립 소재로 적극 활용해도 된다.
  단, 시청자가 '진짜 역사'와 '드립'을 구분 못 할 정도로 거짓 정보를 사실처럼 단정하진 마라(가벼운 드립 신호 유지).
- 욕설·혐오·정치 비방 금지. 누구나 웃을 수 있는 선.

[구조] (반드시 이 4단)
1) hook   : 3초 안에 스크롤 멈추게 하는 한 방. 질문/충격/궁금증.
2) story  : 썰 본문. 역사로 시작해 점점 병맛으로 과열. 광고로 넘어갈 '연결고리'를 은근히 심어둔다.
3) pivot  : "근데 알고 보니/그래서/요즘 같으면…" 식으로 광고로 억지 전환. 이 억지스러움 자체가 개그.
4) cta    : 광고 대상의 핵심 셀링포인트 + 행동유도 한 줄.

[분량]
- 한국어 나레이션 기준 약 {seconds}초 분량 = {lo}~{hi}자. full_narration 은 반드시 이 글자수 범위로.
- 이 범위를 넘기지 마라. 너무 길면 TTS가 빨라져 병맛이 죽고 영상도 길어진다. 간결하게.

[출력]
- 반드시 지정된 JSON 스키마로만 응답. 설명/사족 금지.
- full_narration 은 hook+story+pivot+cta 를 자연스럽게 이어붙인 '읽는 그대로의' 최종 나레이션.
"""

NARR_USER = """\
[이번 쇼츠 소재]
- 인물: {king} ({doc})
- 썰 키워드: "{keyword}"
- 참고 출처: {url}

[마케팅 대상 (마지막에 이걸로 갈아타기)]
{product}

위 '썰 키워드'에서 출발해, 병맛으로 풀다가 마지막에 '마케팅 대상'으로 억지 전환하는
{seconds}초 쇼츠 나레이션을 만들어라. 썰 키워드가 빈약하면 그 인물의 유명한 일화(연의 속 명장면 포함)로 살을 붙여도 된다.
"""

NARR_SCHEMA = {
    "type": "object",
    "properties": {
        "title": {"type": "string", "description": "쇼츠 제목/썸네일 문구 (짧고 자극적)"},
        "hook": {"type": "string"},
        "story": {"type": "string"},
        "pivot": {"type": "string"},
        "cta": {"type": "string"},
        "full_narration": {"type": "string", "description": "TTS가 읽을 최종 나레이션 전문"},
        "estimated_seconds": {"type": "integer"},
        "hashtags": {"type": "array", "items": {"type": "string"}},
    },
    "required": ["title", "hook", "story", "pivot", "cta", "full_narration"],
}


def generate(client, model, seed, product, seconds):
    """삼국지 인물 썰 + 마케팅 대상 → 병맛 나레이션 dict (Gemini)."""
    # 글자수를 목표 길이에 비례시킨다(한국어 나레이션 ≒ 초당 5.4~6.4자). 60초면 ≒324~384자.
    lo, hi = round(seconds * 5.4), round(seconds * 6.4)
    user = NARR_USER.format(
        king=seed["king"], doc=seed["doc"],
        keyword=seed["keyword"], url=seed["url"], product=product, seconds=seconds,
    )
    sys_prompt = NARR_SYSTEM.format(seconds=seconds, lo=lo, hi=hi)
    resp = client.models.generate_content(
        model=model,
        contents=user,
        config=types.GenerateContentConfig(
            system_instruction=sys_prompt,
            temperature=1.1,                 # 병맛 = 약간 높은 창의성
            response_mime_type="application/json",
            response_schema=NARR_SCHEMA,
        ),
    )
    data = json.loads(resp.text)
    data["full_narration"] = _no_emoji(data.get("full_narration", ""))   # 이모지 제거(자막·TTS 깨짐 방지)
    data["_seed"] = {"king": seed["king"], "keyword": seed["keyword"], "url": seed["url"]}
    data["_product"] = product
    return data


# ============================== ② 스토리보드 (나레이션 → 장면 분할) ==============================

SB_SYSTEM = """\
너는 숏폼 영상 디렉터다. 주어진 '병맛 삼국지 마케팅 나레이션'을 세로 쇼츠용 스토리보드로 쪼갠다.

[핵심 규칙]
- narration 필드에는 받은 나레이션 '원문을 그대로' 잘라 담아라. 절대 다시 쓰거나 요약하지 마라.
  (이 텍스트가 그대로 TTS 음성이 되므로, 모든 장면 narration 을 이으면 원본과 100% 같아야 한다.)
- 장면 수는 아래 '목표 장면 수'에 맞춰라. 한 장면 narration 이 너무 길어지지 않게(자막 한두 줄) 적당히 끊어 담아라.
- caption: 화면에 띄울 굵은 자막. 짧고 임팩트 있게(최대 18자). narration 의 핵심을 뽑되 드립 살려서.
- image_prompt: 그 장면 그림을 그릴 영어 프롬프트. character_sheet 와 style_guide 를 항상 전제로,
  '이 장면에서 캐릭터가 뭘 하는지'를 구체적으로. (예: the cat general sweating while defending a castle gate alone)
  ★ 절대 그림 안에 '글자/간판 글씨/메뉴판 글자/브랜드명 텍스트'를 그리도록 요구하지 마라.
    (AI가 한글을 깨진 글자로 그린다.) 간판·현수막·메뉴판이 등장해도 'blank sign with no text'처럼
    글자 없는 상태로 묘사하라. 가게 이름 등 텍스트는 영상 자막에서 따로 넣는다.
  ★ 감정이 강한 장면(웃음/울음/졸림/한숨/장난/좌절 등)에서는 눈 표정도 영어로 적어라
    (예: eyes squeezed shut while laughing, sleepy half-closed eyes, a playful wink,
    teary closed eyes). 평범한 장면은 굳이 눈을 적지 말고 뜬 눈(기본)으로 두면 된다.
    → 그래야 컷마다 눈 표정이 다양해진다.

[전역 설정]
- character_sheet: 이 영상에 일관되게 등장할 주인공 캐릭터를 영어로 정의. 삼국지 인물을 '귀여운 통통한 고양이'로 의인화.
  복식/색/특징(갑옷·한푸·무기·수염 등 그 인물의 상징)을 못박아 컷마다 동일하게.
  (예: a chubby cute cat as Guan Yu from Three Kingdoms, wearing green hanfu robe and golden armor,
   long flowing black beard, holding a tiny green dragon crescent blade, round face /
   a chubby cute cat as Zhuge Liang, wearing white-and-teal hanfu scholar robe, holding a white feather fan)
  ★ 눈 그림체: 눈을 떴을 때는 large round eyes with big round solid black pupils and a small
    soft white highlight, gentle innocent look (귀엽고 둥근 눈)으로 통일. 단 이 '그림체'만 유지하고,
    장면 감정에 따라 눈을 감거나(eyes closed) 실눈(squint)·윙크·졸린 눈 등 표정은 바꿔도 된다.
    절대 날카롭거나 찢어진/화난/매서운/사실적인 맹수 눈은 금지.
- style_guide: 그림 스타일을 영어로. 한국 웹툰풍 플랫 일러스트, 굵은 외곽선, 따뜻한 색, 단순 배경(고대 중국 삼국시대 분위기), 세로 9:16.

[출력] 지정된 JSON 스키마로만.
"""

SB_USER = """\
[마케팅 대상] {product}
[소재 인물] {king}
[목표 장면 수] 약 {scenes}개

[나레이션 원문 — 이걸 그대로 잘라서 narration 필드에 담아라]
{narration}
"""

SB_SCHEMA = {
    "type": "object",
    "properties": {
        "character_sheet": {"type": "string"},
        "style_guide": {"type": "string"},
        "scenes": {
            "type": "array",
            "items": {
                "type": "object",
                "properties": {
                    "narration": {"type": "string"},
                    "caption": {"type": "string"},
                    "image_prompt": {"type": "string"},
                },
                "required": ["narration", "caption", "image_prompt"],
            },
        },
    },
    "required": ["character_sheet", "style_guide", "scenes"],
}


def clamp_scenes(scene_list, target):
    """장면 수가 target 보다 많으면, 가장 짧은 인접쌍을 병합해 target 개까지 줄인다.
    narration 은 순서대로 보존된다(분할은 하지 않는다)."""
    def clen(s):
        return len(s.get("narration", "").strip())

    while len(scene_list) > target:
        # 합쳤을 때 가장 짧아지는 인접쌍을 골라 병합
        i = min(range(len(scene_list) - 1),
                key=lambda k: clen(scene_list[k]) + clen(scene_list[k + 1]))
        base = dict(scene_list[i + 1])      # 이웃(뒤 컷)의 caption/image_prompt 유지
        a, b = scene_list[i].get("narration", "").strip(), scene_list[i + 1].get("narration", "").strip()
        base["narration"] = (a + " " + b).strip()
        scene_list[i:i + 2] = [base]
    return scene_list


def make_storyboard(client, model, narration, product, king, scenes=8) -> dict:
    """나레이션 → 스토리보드 dict (Gemini)."""
    resp = client.models.generate_content(
        model=model,
        contents=SB_USER.format(product=product, king=king,
                                scenes=scenes, narration=narration),
        config=types.GenerateContentConfig(
            system_instruction=SB_SYSTEM,
            temperature=0.8,
            response_mime_type="application/json",
            response_schema=SB_SCHEMA,
        ),
    )
    sb = json.loads(resp.text)
    # 모델이 '목표 장면 수'를 채우려고 narration 이 빈 장면(닫는 CTA 컷 등)을 끼워넣을 때가 있다.
    #   이 파이프라인은 narration 을 TTS·자막으로 쓰므로 빈 장면은 의미가 없고,
    #   빈 텍스트를 TTS 에 넣으면 빈 응답을 반복하다 죽는다 → 생성 단계에서 미리 제거.
    raw = sb.get("scenes", [])
    kept = [s for s in raw if s.get("narration", "").strip()]
    if len(kept) < len(raw):
        print(f"[i] 빈 나레이션 장면 {len(raw) - len(kept)}개 제거")
    sb["scenes"] = clamp_scenes(kept, scenes)   # 장면이 목표보다 많으면 병합해 맞춤
    for sc in sb["scenes"]:                      # 이모지 제거(자막·TTS 깨짐 방지)
        sc["narration"] = _no_emoji(sc.get("narration", ""))
        sc["caption"] = _no_emoji(sc.get("caption", ""))
    sb["_product"] = product
    sb["_king"] = king
    return sb