app.py

"""
app.py — Dark Moon Study Report (Hugging Face Spaces Edition)
HF Spaces 전용: pyaudiowpatch/Live 기능 제외
포트: 7860 (HF 기본값)
"""

import logging
import os
import re
import tempfile
from pathlib import Path
from datetime import datetime

import gradio as gr

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger("study_hf")

# ── 공유 유틸 ────────────────────────────────────────────────

def _extract_text_from_srt(path: str) -> str:
    content = Path(path).read_text(encoding="utf-8", errors="replace")
    content = re.sub(r"^\d+\s*$", "", content, flags=re.MULTILINE)
    content = re.sub(r"\d{2}:\d{2}:\d{2},\d{3} --> \d{2}:\d{2}:\d{2},\d{3}", "", content)
    content = re.sub(r"<[^>]+>", "", content)
    return " ".join(content.split())


def _extract_youtube_id(url: str) -> str | None:
    """Extract YouTube video ID from various URL formats."""
    patterns = [
        r"(?:v=|youtu\.be/|embed/|shorts/)([A-Za-z0-9_-]{11})",
    ]
    for p in patterns:
        m = re.search(p, url)
        if m:
            return m.group(1)
    return None


# ── Gemini 분석 ──────────────────────────────────────────────

_ws_item = lambda props: {"type": "array", "items": {"type": "object", "properties": props}}

_SCHEMA = {
    "type": "object",
    "properties": {
        "title":      {"type": "string"},
        "subject":    {"type": "string"},
        "content_type": {"type": "string"},   # english_lecture | korean_lecture | tech_tutorial | documentary | general
        "summary": {
            "type": "object",
            "properties": {
                "intro": {"type": "string"},
                "main_points": _ws_item({
                    "point":   {"type": "string"},
                    "details": {"type": "array", "items": {"type": "string"}},
                }),
            },
        },
        "mermaid":    {"type": "string"},
        "keywords":   _ws_item({"term": {"type": "string"}, "definition": {"type": "string"}}),
        "questions":  _ws_item({"q": {"type": "string"}, "a": {"type": "string"}, "hint": {"type": "string"}}),
        "timeline":   _ws_item({"topic": {"type": "string"}, "description": {"type": "string"}}),
        # ── 워크시트 전용 필드 ──────────────────────────
        "worksheet": {
            "type": "object",
            "properties": {
                "pre_viewing":   _ws_item({"question": {"type": "string"}}),
                "vocabulary":    _ws_item({
                    "word":       {"type": "string"},
                    "definition": {"type": "string"},
                    "example":    {"type": "string"},
                }),
                "fill_blanks":   _ws_item({"sentence": {"type": "string"}, "answer": {"type": "string"}}),
                "comprehension": _ws_item({"q": {"type": "string"}, "a": {"type": "string"}, "type": {"type": "string"}}),
                "discussion":    _ws_item({"question": {"type": "string"}}),
                "self_check":    _ws_item({"item": {"type": "string"}}),
            },
        },
    },
    "required": ["title", "subject", "content_type", "summary", "mermaid",
                 "keywords", "questions", "timeline", "worksheet"],
}

PROMPT = """You are an expert educator and learning materials designer.
Analyze the lecture transcript below and produce a structured study report.

OUTPUT LANGUAGE: {lang}
All text fields (title, subject, intro, main_points, keywords definitions, questions, timeline, worksheet) MUST be written in {lang}.

【Core Rules】
- title: Lecture title (max 20 chars)
- subject: Subject/field (max 10 chars)
- content_type: one of english_lecture | korean_lecture | tech_tutorial | documentary | general
- summary.intro: Single sentence — the ONE most important takeaway
- summary.main_points: EXACTLY 3 key points. Each point must be a sharp, actionable insight (NOT a list of facts). Each point has 2 supporting details max. Be concise and essential, avoid redundancy.
- keywords: 8 core terms — the word on front, a SHORT clear definition (max 20 words) on back
- questions: 5 exam-quality questions (q, hint, a)
- timeline: 4~5 lecture flow steps
- worksheet fields: pre_viewing (2q), vocabulary (8 terms), fill_blanks (5), comprehension (5), discussion (2q), self_check (4 items)

Transcript:
{text}
"""





def _analyze(text: str, api_key: str, lang: str = "한국어") -> dict:
    import json
    import google.generativeai as genai

    genai.configure(api_key=api_key.strip())

    model = genai.GenerativeModel(
        "gemini-2.0-flash",
        generation_config=genai.GenerationConfig(
            response_mime_type="application/json",
            response_schema=_SCHEMA,
        ),
    )

    trimmed = text[:12000] + ("\n[생략]" if len(text) > 12000 else "")
    response = model.generate_content(PROMPT.format(text=trimmed, lang=lang))
    return json.loads(response.text)  # 스키마 적용으로 항상 유효한 JSON


# ── HTML 리포트 렌더러 ────────────────────────────────────────

def _render(data: dict, source_url: str = "") -> str:
    title    = data.get("title", "강의 정리")
    subject  = data.get("subject", "일반")
    summary  = data.get("summary", {})
    keywords = data.get("keywords", [])
    questions= data.get("questions", [])
    timeline = data.get("timeline", [])
    date_str = datetime.now().strftime("%Y년 %m월 %d일")

    intro = f'<p class="intro">{summary.get("intro","")}</p>'

    points = ""
    for i, mp in enumerate(summary.get("main_points", []), 1):
        details = "".join(f"<li>{d}</li>" for d in mp.get("details", []))
        points += f"""<div class="point-card">
          <div class="point-hd"><span class="pn">{i}</span><span class="pt">{mp.get('point','')}</span></div>
          <ul class="dl">{details}</ul></div>"""

    cards = ""
    for kw in keywords:
        cards += f"""<div class="flip-card">
          <div class="flip-inner">
            <div class="flip-front"><span class="term">{kw.get('term','')}</span><span class="ch">클릭 ↓</span></div>
            <div class="flip-back"><span class="defi">{kw.get('definition','')}</span></div>
          </div></div>"""

    qs = ""
    for i, q in enumerate(questions, 1):
        qs += f"""<div class="qcard">
          <div class="qhd" onclick="tog(this)">
            <span class="qn">Q{i}</span><span class="qt">{q.get('q','')}</span><span class="arr">▼</span>
          </div>
          <div class="qans" style="display:none">
            <div class="hint">💡 {q.get('hint','')}</div>
            <div class="ans">✅ {q.get('a','')}</div>
          </div></div>"""

    tl = ""
    for t in timeline:
        tl += f"""<div class="tli">
          <div class="dot"></div>
          <div class="tlc"><div class="tlto">{t.get('topic','')}</div>
          <div class="tlds">{t.get('description','')}</div></div></div>"""

    src = f'<div class="src">📎 <a href="{source_url}" target="_blank">{source_url}</a></div>' if source_url else ""

    # 키워드로 개념 버블 그리드 생성 (Mermaid 대체)
    bubble_items = ""
    colors = ["#e3f2fd","#fce4ec","#e8f5e9","#fff3e0","#f3e5f5","#e0f2f1","#fff8e1","#f1f8e9"]
    border_colors = ["#1565c0","#c62828","#2e7d32","#e65100","#6a1b9a","#00695c","#f57f17","#558b2f"]
    for idx, kw in enumerate(keywords[:8]):
        bg = colors[idx % len(colors)]
        bd = border_colors[idx % len(border_colors)]
        bubble_items += f'<div style="background:{bg};border:2px solid {bd};border-radius:12px;padding:14px 16px;text-align:center"><div style="font-weight:700;font-size:13px;color:{bd};margin-bottom:6px">{kw.get("term","")}</div><div style="font-size:12px;color:#333;line-height:1.5">{kw.get("definition","")[:80]}</div></div>'
    concept_grid = f'<div style="display:grid;grid-template-columns:repeat(auto-fill,minmax(200px,1fr));gap:14px;padding:18px 22px">{bubble_items}</div>' if bubble_items else '<p style="padding:18px;color:#999">키워드 데이터 없음</p>'

    return f"""<!DOCTYPE html>
<html lang="ko">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width,initial-scale=1">
<title>{title}</title>
<link href="https://fonts.googleapis.com/css2?family=Noto+Sans+KR:wght@300;400;500;700;900&display=swap" rel="stylesheet">
<style>
:root{{color-scheme:light;--navy:#1a237e;--navy2:#283593;--mint:#00bcd4;--mint-l:#e0f7fa;--yell:#fff9c4;--ydk:#f9a825;--purp:#f3e5f5;--pdk:#7b1fa2;--tx:#1c1c1e;--txl:#636e72;--r:14px;--sh:0 4px 20px rgba(0,0,0,.08)}}
*{{box-sizing:border-box;margin:0;padding:0}}
html,body{{font-family:'Noto Sans KR',sans-serif;background:#f4f6f9!important;color:#1c1c1e!important;line-height:1.7}}
.hero{{background:#ffffff;color:#1c1c1e;padding:48px 32px 40px;text-align:center;border-bottom:2px solid #e0e0e0}}
.badge{{display:inline-block;background:rgba(26,35,126,.1);border:1px solid rgba(26,35,126,.3);border-radius:20px;padding:4px 16px;font-size:12px;font-weight:500;margin-bottom:16px;letter-spacing:1px;color:var(--navy)}}
.hero h1{{font-size:2rem;font-weight:900;margin-bottom:8px;color:#1a237e!important}}
.meta{{font-size:13px;opacity:.7;margin-top:6px;color:#1c1c1e!important}}
.wrap{{max-width:900px;margin:0 auto;padding:32px 16px;display:flex;flex-direction:column;gap:24px}}
.sec{{background:#fff;border-radius:var(--r);box-shadow:var(--sh);overflow:hidden}}
.shd{{display:flex;align-items:center;gap:12px;padding:18px 22px 14px;border-bottom:1px solid #f0f0f0}}
.si{{font-size:20px}}.st{{font-size:17px;font-weight:700;color:var(--navy)}}
.sb{{padding:18px 22px}}
.intro{{color:var(--txl);font-size:15px;padding:12px 16px;background:var(--mint-l);border-left:4px solid var(--mint);border-radius:8px;margin-bottom:14px}}
.point-card{{border:1px solid #f0f0f0;border-radius:10px;margin-bottom:10px;overflow:hidden}}
.point-hd{{display:flex;align-items:center;gap:10px;padding:12px 14px;background:#fafafa}}
.pn{{width:26px;height:26px;border-radius:50%;background:#1a237e!important;color:#fff!important;display:flex;align-items:center;justify-content:center;font-size:12px;font-weight:700;flex-shrink:0}}
.pt{{font-weight:700;font-size:14px;color:#1a237e!important}}
.dl{{padding:10px 14px 10px 50px;list-style:none}}
.dl li{{position:relative;padding-left:14px;color:var(--txl);font-size:13px;margin-bottom:5px}}
.dl li::before{{content:'•';position:absolute;left:0;color:var(--mint);font-weight:700}}
.mw{{padding:22px;overflow-x:auto;text-align:center}}
.cards{{display:grid;grid-template-columns:repeat(auto-fill,minmax(180px,1fr));gap:14px;padding:18px 22px}}
.flip-card{{perspective:800px;min-height:140px;cursor:pointer}}
.flip-inner{{position:relative;width:100%;min-height:140px;transform-style:preserve-3d;transition:transform .5s}}
.flip-card:hover .flip-inner,.flip-card.on .flip-inner{{transform:rotateY(180deg)}}
.flip-front,.flip-back{{position:absolute;top:0;left:0;width:100%;min-height:140px;border-radius:10px;backface-visibility:hidden;display:flex;flex-direction:column;align-items:center;justify-content:center;padding:14px;text-align:center;overflow:hidden;word-break:break-word;overflow-wrap:break-word}}
.flip-front{{background:var(--yell);border:2px solid var(--ydk)}}
.flip-back{{background:#fff;color:#1c1c1e!important;transform:rotateY(180deg);border:2px solid #e0e0e0}}
.term{{font-size:14px;font-weight:700;color:#1a237e!important;margin-bottom:6px;word-break:break-word}}
.ch{{font-size:10px;color:var(--txl)}}
.defi{{font-size:11px;line-height:1.5;color:#1c1c1e!important;word-break:break-word;overflow-wrap:break-word}}
.qcard{{border:1px solid #f0f0f0;border-radius:10px;margin-bottom:10px;overflow:hidden}}
.qhd{{display:flex;align-items:flex-start;gap:10px;padding:14px;cursor:pointer;background:var(--purp);transition:background .2s}}
.qhd:hover{{background:#e8d5f5}}
.qn{{background:var(--pdk);color:#fff;border-radius:6px;padding:2px 7px;font-size:11px;font-weight:700;flex-shrink:0}}
.qt{{flex:1;font-size:13px;font-weight:500;color:var(--navy)}}
.arr{{color:var(--txl);font-size:11px;flex-shrink:0}}
.qans{{padding:14px;background:#fff}}
.hint{{font-size:13px;color:var(--txl);padding:7px 10px;background:#fafafa;border-radius:6px;margin-bottom:8px}}
.ans{{font-size:13px;color:#2e7d32;font-weight:600;padding:7px 10px;background:#e8f5e9;border-radius:6px}}
.tlw{{padding:18px 22px}}
.tli{{display:flex;gap:14px;margin-bottom:18px;position:relative}}
.tli:not(:last-child)::after{{content:'';position:absolute;left:10px;top:22px;width:2px;height:calc(100% + 2px);background:#e0e0e0}}
.dot{{width:22px;height:22px;border-radius:50%;background:var(--mint);border:3px solid #fff;box-shadow:0 0 0 2px var(--mint);flex-shrink:0;margin-top:2px}}
.tlc{{flex:1}}
.tlto{{font-weight:700;font-size:13px;color:var(--navy);margin-bottom:3px}}
.tlds{{font-size:12px;color:var(--txl)}}
.src{{text-align:center;font-size:12px;color:var(--txl);padding:8px 0}}
.src a{{color:var(--mint);text-decoration:none}}
.footer{{text-align:center;padding:22px;font-size:12px;color:#bdbdbd}}
@media print{{body{{background:#fff}}.hero{{-webkit-print-color-adjust:exact;print-color-adjust:exact}}}}
</style>
</head>
<body>
<div class="hero">
  <div class="badge">{subject} 스터디 노트</div>
  <h1>{title}</h1>
  <div class="meta">🌕 Dark Moon Study Report &nbsp;|&nbsp; {date_str}</div>
</div>
<div class="wrap">
  <div class="sec">
    <div class="shd"><span class="si">📋</span><span class="st">핵심 요점 정리</span></div>
    <div class="sb">{intro}{points}</div>
  </div>
  <div class="sec">
    <div class="shd"><span class="si">🗺️</span><span class="st">핵심 개념 맵</span></div>
    {concept_grid}
  </div>
  <div class="sec">
    <div class="shd"><span class="si">⚡</span><span class="st">핵심 암기 카드 (클릭)</span></div>
    <div class="cards">{cards}</div>
  </div>
  <div class="sec">
    <div class="shd"><span class="si">📝</span><span class="st">예상 시험 문제</span></div>
    <div class="sb">{qs}</div>
  </div>
  <div class="sec">
    <div class="shd"><span class="si">⏱️</span><span class="st">강의 타임라인</span></div>
    <div class="tlw">{tl}</div>
  </div>
  {src}
</div>
<div class="footer">🌕 Dark Moon Study Report — by 달의이성</div>
<script>
function tog(el){{const a=el.nextElementSibling,ar=el.querySelector('.arr');if(a.style.display==='none'){{a.style.display='block';ar.textContent='▲'}}else{{a.style.display='none';ar.textContent='▼'}}}}
document.querySelectorAll('.flip-card').forEach(c=>c.addEventListener('click',()=>c.classList.toggle('on')));
</script>
</body>
</html>"""


# ── 워크시트 렌더러 ───────────────────────────────────────────

_WS_CSS = """
html,body{color-scheme:light;font-family:'Noto Sans KR',sans-serif;background:#fff!important;color:#1c1c1e!important;margin:0;padding:0;line-height:1.7}
.ws-wrap{max-width:800px;margin:0 auto;padding:24px 20px}
.ws-hero{background:linear-gradient(135deg,#1a237e,#3949ab);color:#fff;padding:28px 24px;border-radius:12px;margin-bottom:24px}
.ws-hero h1{font-size:1.4rem;font-weight:900;margin:0 0 6px;color:#fff!important}
.ws-meta{font-size:12px;opacity:.8}
.ws-badge{display:inline-block;background:rgba(255,255,255,.2);border-radius:12px;padding:2px 10px;font-size:11px;margin-bottom:10px}
.section{background:#f8f9fa;border-radius:10px;padding:18px 20px;margin-bottom:16px}
.section h2{font-size:14px;font-weight:700;color:#1a237e!important;margin:0 0 12px;display:flex;align-items:center;gap:8px}
.vocab-table{width:100%;border-collapse:collapse;font-size:13px}
.vocab-table th{background:#1a237e;color:#fff!important;padding:7px 10px;text-align:left}
.vocab-table td{padding:7px 10px;border-bottom:1px solid #e0e0e0;color:#1c1c1e!important}
.vocab-table tr:nth-child(even) td{background:#f0f4ff}
.blank-item{padding:8px 12px;border-left:3px solid #00bcd4;margin-bottom:8px;background:#fff;color:#1c1c1e!important;font-size:13px;border-radius:0 6px 6px 0}
.q-item{padding:8px 12px;margin-bottom:8px;border:1px solid #e0e0e0;border-radius:6px;font-size:13px;color:#1c1c1e!important}
.q-num{font-weight:700;color:#7b1fa2!important;margin-right:6px}
.ans{color:#2e7d32!important;font-size:12px;margin-top:4px;padding:4px 8px;background:#e8f5e9;border-radius:4px}
.pre-q{padding:8px 12px;background:#e3f2fd;border-radius:6px;margin-bottom:8px;font-size:13px;color:#1c1c1e!important}
.disc-q{padding:8px 12px;background:#fce4ec;border-radius:6px;margin-bottom:8px;font-size:13px;color:#1c1c1e!important}
.check-item{padding:6px 0;font-size:13px;display:flex;align-items:center;gap:8px;color:#1c1c1e!important}
.check-box{width:16px;height:16px;border:2px solid #1a237e;border-radius:3px;flex-shrink:0}
.footer{text-align:center;padding:16px;font-size:11px;color:#bdbdbd}
.answer-key{background:#fff3e0;border-left:4px solid #f9a825;padding:4px 8px;font-size:12px;color:#e65100!important;border-radius:0 4px 4px 0;margin-top:4px}
@media print{.ws-hero{-webkit-print-color-adjust:exact;print-color-adjust:exact}}
"""

_TYPE_LABELS = {
    "english_lecture": "🇬🇧 English Lecture",
    "korean_lecture":  "📚 한국어 강의",
    "tech_tutorial":   "💻 Tech Tutorial",
    "documentary":     "🎬 Documentary",
    "general":         "📹 Video Lesson",
}


def _render_worksheet(data: dict) -> str:
    """학생용 워크시트 (정답 미포함)"""
    ws    = data.get("worksheet", {})
    title = data.get("title", "학습 워크시트")
    ct    = data.get("content_type", "general")
    date  = datetime.now().strftime("%Y-%m-%d")
    badge = _TYPE_LABELS.get(ct, "📹 Video Lesson")

    pre   = "".join(f'<div class="pre-q">💭 {q.get("question","")}</div>' for q in ws.get("pre_viewing", []))
    vocab_rows = "".join(
        f'<tr><td><b>{v.get("word","")}</b></td><td>{v.get("definition","")}</td>'
        f'<td style="color:#636e72;font-size:12px">{v.get("example","")}</td></tr>'
        for v in ws.get("vocabulary", [])
    )
    blanks = "".join(f'<div class="blank-item">{i+1}. {fb.get("sentence","")}</div>'
                     for i, fb in enumerate(ws.get("fill_blanks", [])))
    comps  = "".join(f'<div class="q-item"><span class="q-num">Q{i+1}</span>{c.get("q","")}</div>'
                     for i, c in enumerate(ws.get("comprehension", [])))
    discs  = "".join(f'<div class="disc-q">🗣️ {d.get("question","")}</div>'
                     for d in ws.get("discussion", []))
    checks = "".join(
        f'<div class="check-item"><div class="check-box"></div>{s.get("item","")}</div>'
        for s in ws.get("self_check", [])
    )

    return f"""<!DOCTYPE html>
<html lang="ko"><head><meta charset="UTF-8">
<link href="https://fonts.googleapis.com/css2?family=Noto+Sans+KR:wght@400;700;900&display=swap" rel="stylesheet">
<style>{_WS_CSS}</style></head><body>
<div class="ws-wrap">
<div class="ws-hero"><div class="ws-badge">{badge}</div>
<h1>{title} — 학습 워크시트</h1>
<div class="ws-meta">🌕 Dark Moon Lesson Builder &nbsp;|&nbsp; {date} &nbsp;|&nbsp; Student Copy</div>
</div>
<div class="section"><h2>🔍 시청 전 활동</h2>{pre or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>📖 핵심 어휘</h2>
<table class="vocab-table"><thead><tr><th>단어/용어</th><th>의미</th><th>예문/예시</th></tr></thead>
<tbody>{vocab_rows}</tbody></table></div>
<div class="section"><h2>✏️ 시청 중 — 빈칸 채우기</h2>{blanks or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>🧠 이해 확인 문제</h2>{comps or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>💬 토론 질문</h2>{discs or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>✅ 자기 평가</h2>{checks or "<p style='color:#999'>-</p>"}</div>
<div class="footer">🌕 Dark Moon Lesson Builder — by 달의이성 | Student Copy</div>
</div></body></html>"""


def _render_teacher(data: dict) -> str:
    """교사용 가이드 (정답 포함)"""
    ws    = data.get("worksheet", {})
    title = data.get("title", "교사 가이드")
    ct    = data.get("content_type", "general")
    date  = datetime.now().strftime("%Y-%m-%d")
    badge = _TYPE_LABELS.get(ct, "📹 Video Lesson")

    pre   = "".join(f'<div class="pre-q">💭 {q.get("question","")}</div>' for q in ws.get("pre_viewing", []))
    vocab_rows = "".join(
        f'<tr><td><b>{v.get("word","")}</b></td><td>{v.get("definition","")}</td>'
        f'<td style="color:#636e72;font-size:12px">{v.get("example","")}</td></tr>'
        for v in ws.get("vocabulary", [])
    )
    blanks = "".join(
        f'<div class="blank-item">{i+1}. {fb.get("sentence","")}'
        f'<div class="answer-key">🔑 {fb.get("answer","")}</div></div>'
        for i, fb in enumerate(ws.get("fill_blanks", []))
    )
    comps = "".join(
        f'<div class="q-item"><span class="q-num">Q{i+1}</span>{c.get("q","")}'
        f'<div class="ans">✅ {c.get("a","")}</div></div>'
        for i, c in enumerate(ws.get("comprehension", []))
    )
    discs  = "".join(f'<div class="disc-q">🗣️ {d.get("question","")}</div>' for d in ws.get("discussion", []))
    checks = "".join(
        f'<div class="check-item"><div class="check-box"></div>{s.get("item","")}</div>'
        for s in ws.get("self_check", [])
    )
    study_q = "".join(
        f'<div class="q-item"><span class="q-num">Q{i+1}</span>{q.get("q","")}'
        f'<div class="ans">✅ {q.get("a","")}</div>'
        f'<div style="font-size:11px;color:#636e72;margin-top:2px">💡 힌트: {q.get("hint","")}</div></div>'
        for i, q in enumerate(data.get("questions", []))
    )

    return f"""<!DOCTYPE html>
<html lang="ko"><head><meta charset="UTF-8">
<link href="https://fonts.googleapis.com/css2?family=Noto+Sans+KR:wght@400;700;900&display=swap" rel="stylesheet">
<style>{_WS_CSS}
.teacher-banner{{background:#e8f5e9;border:2px solid #2e7d32;border-radius:8px;padding:10px 16px;margin-bottom:16px;font-size:13px;color:#1b5e20!important;font-weight:700}}
</style></head><body>
<div class="ws-wrap">
<div class="ws-hero"><div class="ws-badge">{badge}</div>
<h1>{title} — 교사 가이드</h1>
<div class="ws-meta">🌕 Dark Moon Lesson Builder &nbsp;|&nbsp; {date} &nbsp;|&nbsp; Teacher Copy</div>
</div>
<div class="teacher-banner">🗝️ 교사용 — 정답 포함 버전. 학생에게 배포하지 마세요.</div>
<div class="section"><h2>🔍 시청 전 활동</h2>{pre or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>📖 핵심 어휘 (정답)</h2>
<table class="vocab-table"><thead><tr><th>단어/용어</th><th>의미</th><th>예문/예시</th></tr></thead>
<tbody>{vocab_rows}</tbody></table></div>
<div class="section"><h2>✏️ 빈칸 채우기 (정답)</h2>{blanks or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>🧠 이해 문제 (정답)</h2>{comps or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>💬 토론 질문</h2>{discs or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>✅ 자기 평가</h2>{checks or "<p style='color:#999'>-</p>"}</div>
<div class="section"><h2>📝 추가 예상 문제 정답</h2>{study_q or "<p style='color:#999'>-</p>"}</div>
<div class="footer">🌕 Dark Moon Lesson Builder — by 달의이성 | CONFIDENTIAL</div>
</div></body></html>"""


# ── 메인 핸들러 ──────────────────────────────────────────────


def run_analysis(
    srt_file,
    text_in: str,
    gemini_key: str,
    lang: str,
) -> tuple:

    if not gemini_key or not gemini_key.strip():
        return "❌ Gemini API Key를 입력하세요.\naistudio.google.com에서 무료 발급.", "", None, None, None

    text = ""

    # ── 1. 텍스트 소스 결정 ──────────────────────────────
    if srt_file is not None:
        try:
            srt_path = srt_file if isinstance(srt_file, str) else srt_file.name
            text = _extract_text_from_srt(srt_path)
        except Exception as e:
            return f"❌ 파일 읽기 실패: {e}", "", None, None, None

    elif text_in and text_in.strip():
        text = text_in.strip()
    else:
        return "❌ SRT 파일을 업로드하거나 텍스트를 붙여넣기 하세요.", "", None, None, None

    if not text.strip():
        return "❌ 텍스트가 비어 있습니다.", "", None, None, None

    # ── 2. Gemini 분석 ────────────────────────────────────
    logger.info("🤖 Gemini AI 분석 중...")
    try:
        data = _analyze(text, gemini_key, lang)
    except Exception as e:
        return f"❌ Gemini 분석 실패: {str(e)[:150]}\n(API Key와 네트워크를 확인하세요)", "", None, None, None

    # ── 3. 파일 생성 (학습 노트 + 워크시트 + 교사 가이드) ────
    logger.info("✨ 파일 생성 중...")
    ts   = datetime.now().strftime("%Y%m%d_%H%M%S")
    title_raw = str(data.get("title") or "report")
    safe = re.sub(r'[^\w\-_가-힣]', '_', title_raw)[:30]
    tmp_dir = Path(tempfile.gettempdir())

    # 학습 노트 HTML
    notes_html = _render(data, "")
    notes_path = tmp_dir / f"{safe}_{ts}_notes.html"
    notes_path.write_text(notes_html, encoding="utf-8")

    # 학생 워크시트
    ws_html  = _render_worksheet(data)
    ws_path  = tmp_dir / f"{safe}_{ts}_worksheet.html"
    ws_path.write_text(ws_html, encoding="utf-8")

    # 교사 가이드
    tch_html = _render_teacher(data)
    tch_path = tmp_dir / f"{safe}_{ts}_teacher.html"
    tch_path.write_text(tch_html, encoding="utf-8")

    # Markdown 노트
    md_lines = [
        f"# {data.get('title','')}",
        f"> {data.get('subject','')} ({data.get('content_type','')}) | {datetime.now().strftime('%Y-%m-%d')}",
        "",
        f"## 요약\n{data.get('summary',{}).get('intro','')}",
        "", "## 핵심 요점",
    ]
    for i, mp in enumerate(data.get("summary",{}).get("main_points",[]), 1):
        md_lines.append(f"### {i}. {mp.get('point','')}")
        for d in mp.get("details",[]): md_lines.append(f"- {d}")
    md_lines += ["", "## 핵심 용어", ""]
    for kw in data.get("keywords",[]): md_lines.append(f"- **{kw.get('term','')}**: {kw.get('definition','')}")
    md_lines += ["", "## 예상 문제", ""]
    for i, q in enumerate(data.get("questions",[]), 1):
        md_lines += [f"**Q{i}.** {q.get('q','')}", f"> 힌트: {q.get('hint','')}", f"> 정답: {q.get('a','')}", ""]
    md_str = "\n".join(md_lines)
    md_path = notes_path.with_suffix(".md")
    md_path.write_text(md_str, encoding="utf-8")

    ct = data.get("content_type", "general")
    logger.info("✅ 완료!")
    status = (
        f"✅ 분석 완료!\n📌 제목: {data.get('title','')}\n"
        f"📚 과목: {data.get('subject','')} [{ct}]\n"
        f"📄 학습 노트 + 학생 워크시트 + 교사 가이드 생성됨"
    )
    return status, notes_html, str(notes_path), str(ws_path), str(tch_path)





# ── Gradio UI ────────────────────────────────────────────────


CSS = """
.title{font-size:2rem;font-weight:900;color:#1a237e;text-align:center;padding:12px 0 2px}
.sub{text-align:center;color:#636e72;font-size:14px;margin-bottom:8px}
"""


HOW_TO_MD = """
### 📋 자막 얻는 방법
**YouTube 영상**  
1. 영상 페이지 → `...` 더보기 → **자막 열기**  
2. 자막 텍스트 전체 복사 → 아래 **텍스트 붙여넣기** 탭에 붙여넣기

또는 [DownSub.com](https://downsub.com) / [savesubs.com](https://savesubs.com) 에서 SRT 다운로드 후 파일 업로드

**로컬 영상 / 시스템 소리 캡처** → [Pro 버전 구매](https://soundfury37.gumroad.com/l/blvcc) 필요
"""


def build_ui():
    with gr.Blocks(title="🌕 Dark Moon Lesson Builder", css=CSS) as demo:

        gr.HTML('<p class="title">🌕 Dark Moon Lesson Builder</p>')
        gr.HTML('<p class="sub">강의 자막 → Gemini AI → 학습 노트 + 워크시트 + 교사 가이드 자동 생성</p>')

        with gr.Row():
            # 입력 패널
            with gr.Column(scale=1, min_width=340):
                srt_in = gr.File(
                    label="📁 SRT / TXT 파일 업로드",
                    file_types=[".srt", ".txt"],
                )
                text_in = gr.Textbox(
                    label="📝 또는 텍스트 직접 붙여넣기 (파일 우선)",
                    placeholder="YouTube 자막, 강의 스크립트 등... (파일 업로드 시 이 칸은 무시됩니다)",
                    lines=6,
                )
                gr.Markdown(HOW_TO_MD)
                lang_in = gr.Dropdown(
                    label="🌐 출력 언어",
                    choices=["한국어", "English", "日本語", "中文"],
                    value="한국어",
                    info="학습 노트 출력 언어 선택",
                )
                key_in = gr.Textbox(
                    label="🔑 Gemini API Key",
                    type="password",
                    placeholder="AIza...",
                    info="무료 발급: aistudio.google.com",
                )
                run_btn = gr.Button("🚀 분석 시작", variant="primary", size="lg")
                status_out = gr.Textbox(label="상태", lines=4, interactive=False)
                gr.Markdown("**📥 다운로드**")
                with gr.Row():
                    dl_notes = gr.File(label="학습 노트")
                    dl_ws    = gr.File(label="학생 워크시트")
                    dl_tch   = gr.File(label="교사 가이드")

            # 학습 노트 미리보기
            with gr.Column(scale=2):
                gr.Markdown("### 📊 학습 노트 미리보기")
                notes_out = gr.HTML(
                    '<div style="text-align:center;padding:80px;color:#bdbdbd;font-size:16px">'
                    '👈 자막/텍스트 입력 후 🚀 분석 시작 클릭</div>'
                )

        run_btn.click(
            run_analysis,
            inputs=[srt_in, text_in, key_in, lang_in],
            outputs=[status_out, notes_out, dl_notes, dl_ws, dl_tch],
            api_name=False,
        )

        gr.Markdown(
            "---\n"
            "**🌕 Dark Moon Lesson Builder** — by 달의이성 &nbsp;|&nbsp; "
            "[Gemini API 무료 발급](https://aistudio.google.com) &nbsp;|&nbsp; "
            "[시스템 오디오 Pro 버전](https://soundfury37.gumroad.com/l/blvcc) &nbsp;|&nbsp; "
            "데이터는 서버에 저장되지 않습니다."
        )

    return demo


if __name__ == "__main__":
    demo = build_ui()
    demo.launch(server_name="0.0.0.0", server_port=7860)