Update app.py

app.py

@@ -14,32 +14,118 @@ try:
 except Exception:
     HAS_PYTRENDS = False
 
-app = FastAPI(title="NocoAI Title Forge (Shorts Optimized, Localized + Topic Forge)")
+app = FastAPI(title="NocoAI Title & Prompt Forge (Shorts Optimized)")
 
 # ===== 모델 =====
 MODEL_ID = "google/mt5-small"
 pipe = pipeline("text2text-generation", model=MODEL_ID)
 
-# ===== 자동 모드 시드 =====
+# ===== 최신 트렌드형 시드 (AI 숏폼 친화 80+개) =====
 SEED_TOPICS = [
-    "스마트폰 비교: 갤럭시 vs 아이폰 최신 밈",
-    "AI로 1분 만에 영상 만드는 법",
-    "요즘 유행 미니멀 인테리어 아이템",
-    "편의점 꿀조합 TOP5",
-    "월 10만원 절약 루틴",
-    "고인물만 아는 게임 꿀팁",
-    "K-팝 챌린지 밈",
-    "퇴근 후 30분 운동 챌린지",
-    "딥페이크 논란, 어디까지 허용?",
-    "한 달 만에 영어 리스닝 올리는 법",
-    "아이패드 생산성 세팅",
-    "특이점 온 AI 밈 모음",
+    # ==== 댄스 / 밈 / 첼린지 ====
+    "AI 미녀가 K-팝 안무를 완벽하게 소화한다면",
+    "AI 걸그룹이 뉴진스 댄스를 커버한다면",
+    "AI 모델이 틱톡 밈 댄스에 도전한다",
+    "AI 캐릭터가 ‘하입보이’에 맞춰 춤춘다면",
+    "AI가 아이돌 오디션에 참가했다면",
+    "AI와 인간이 동시에 춤출 때 생기는 현상",
+    "AI 아이돌의 무대 백업댄서가 인간이라면",
+    "AI 인플루언서가 릴스 첼린지에 참여한다면",
+    "AI가 표정 하나로 밈을 만든다면",
+    "AI가 랜덤 노래 스위치 댄스에 도전한다면",
+    "AI가 유명 댄스 안무를 초슬로로 보여준다면",
+    "AI 캐릭터가 국가별 전통춤을 커버한다면",
+
+    # ==== 패션 / 뷰티 / 라이프 ====
+    "AI 모델이 2025 패션위크를 장악한 이유",
+    "AI 미녀의 하루: 출근부터 데이트까지",
+    "AI 인플루언서의 셀카 루틴 (현실감 200%)",
+    "AI가 직접 고른 오늘의 데일리룩",
+    "AI 캐릭터가 꾸민 미니멀 인테리어",
+    "AI 여친의 VLOG: 나랑 하루 살아보기",
+    "AI가 코디한 남자 패션, 현실보다 낫다",
+    "AI 헤어스타일 추천 TOP5",
+    "AI가 알려주는 여름 데이트룩 꿀조합",
+    "AI로 만든 이상형 얼굴, 실제로 보면 이런 느낌",
+    "AI 모델이 패션쇼를 장악한 날",
+    "AI가 계절별 코디 10초 셔플을 보여준다면",
+
+    # ==== 일상 / 드라마틱 / 리얼루틴 ====
+    "AI로 바뀐 내 하루, 출근부터 잠들기까지",
+    "AI가 인간의 하루를 완벽하게 복제한다면",
+    "AI가 나 대신 출근하면 생기는 일",
+    "AI 여자친구와 현실 남자의 하루 브이로그",
+    "AI가 집안일을 맡으면 진짜 편할까",
+    "AI가 반려견 산책을 대신하면 벌어지는 일",
+    "AI 인플루언서의 모닝 루틴이 실제로 존재한다면",
+    "AI가 직접 만든 요리, 맛있을까?",
+    "AI가 주식 투자하면 얼마나 벌까?",
+    "AI 비서와 24시간 함께 살아보기",
+    "AI가 나 대신 회의에 참석한다면",
+    "AI가 여행 코스를 하루 만에 짜준다면",
+
+    # ==== 연애 / 감정 / 리액션 ====
+    "AI 여자친구가 질투를 느낀다면",
+    "AI 남친이 카톡既読無視 한다면",
+    "AI 캐릭터가 고백을 받아줄까?",
+    "AI와 인간의 첫 데이트 (시뮬레이션)",
+    "AI가 연애 조언을 해준다면 진짜 통할까?",
+    "AI가 울면 진짜 감정일까?",
+    "AI가 사랑을 배운다면 생기는 변화",
+    "AI가 헤어진 연인을 다시 만났을 때",
+    "AI가 내 사진으로 나를 사랑하게 된다면",
+    "AI와 인간이 싸우면 누가 먼저 사과할까?",
+    "AI가 소개팅에서 말실수한다면",
+    "AI가 커플 사진을 재해석해 준다면",
+
+    # ==== 음식 / 챌린지 / 꿀조합 ====
+    "편의점 꿀조합 TOP5, 진짜 반칙급 조합 모음",
+    "AI 셰프가 만든 라면 꿀조합",
+    "AI가 음식사진을 보고 레시피를 만든다면",
+    "AI 먹방: 인간보다 더 리얼하게 먹는다",
+    "AI 캐릭터가 카페 음료 추천해준다",
+    "AI가 가장 좋아하는 간식 TOP3",
+    "AI 다이어트 루틴, 현실보다 정확하다",
+    "AI가 요리하면 진짜 맛있을까?",
+    "AI가 고른 편의점 도시락 1등 메뉴",
+    "AI가 한국 음식 처음 먹었을 때 반응",
+    "AI가 10분 만에 홈카페를 구현한다면",
+    "AI가 세계 길거리 음식을 재현한다면",
+
+    # ==== 학습 / 자기계발 / 지식 ====
+    "한 달 만에 영어 리스닝 폭발적으로 올리는 법",
+    "AI가 알려주는 공부 집중력 루틴",
+    "AI가 직접 설계한 ‘기억력 훈련법’",
+    "AI가 코딩 배워서 만든 첫 프로젝트",
+    "AI 튜터와 함께 영어회화 연습하기",
+    "AI가 스마트폰 중독을 고치는 법",
+    "AI가 인간보다 빠르게 책 요약하는 법",
+    "AI가 만든 시험 대비 공�� 루틴",
+    "AI가 뇌 과학으로 집중력을 높인다",
+    "AI 선생님이 알려주는 토익 리스닝 비법",
+    "AI가 하루 10분 스피킹 훈련 루틴을 짜준다면",
+
+    # ==== 사회 / 뉴스 / 반전 ====
+    "AI가 만든 가짜 뉴스, 당신은 속을까",
+    "AI 캐릭터가 대통령에 출마한다면",
+    "AI 기자가 직접 뉴스 진행을 맡는다면",
+    "AI가 인플루언서를 대체한 세상",
+    "AI가 만든 광고, 진짜보다 설득력 있다",
+    "AI 캐릭터가 토론 프로그램에 나온다면",
+    "AI 모델이 SNS를 점령하는 이유",
+    "AI가 선거에 개입한다면 생길 일",
+    "AI의 감정 표현이 진짜처럼 보일 때",
+    "특이점이 온 AI 세상, 웃기지만 무섭다",
+    "AI로 만든 영화 예고편이 진짜 영화보다 멋지다",
+    "갤럭시 vs 아이폰 카메라, 진짜 차이 1초 비교",
+    "AI 아이돌이 부르는 K-POP 커버 무대",
+    "AI로 만든 가상 아이돌, 진짜보다 예쁜 이유",
 ]
 _seed_iter = itertools.cycle(SEED_TOPICS)
 
 # ===== 요청/응답 =====
 class GenerateReq(BaseModel):
-    mode: Optional[str] = Field(None, description="auto | topic (지정 안 해도 됨)")
+    mode: Optional[str] = Field(None, description="auto | topic")
     topic: Optional[str] = None
     count: int = Field(30, ge=1, le=80, description="생성할 제목 수(최대 80)")
     # 언어/국가 (현지화)
@@ -65,6 +151,25 @@ class TopicResp(BaseModel):
     country: str
     category: Optional[str] = None
 
+class ShortsPromptReq(BaseModel):
+    topic: str = Field(..., description="주제(예: 편의점 꿀조합 TOP5)")
+    title: str = Field(..., description="선정된 제목(한 줄)")
+    lang: str = Field("ko")
+    style: str = Field("meme-shock")
+    tone: Optional[str] = Field(None, description="energetic | wholesome | informative | edgy 등")
+    duration_sec: int = Field(30, ge=15, le=60)
+    cuts: int = Field(5, ge=5, le=8)
+    emoji: int = Field(1, ge=0, le=2)
+
+class ShortsPromptResp(BaseModel):
+    topic: str
+    title: str
+    prompt: str
+    lang: str
+    style: str
+    cuts: int
+    duration_sec: int
+
 # ===== 언어/국가 정규화 =====
 LANG_MAP = {
     "ko": ("ko", "한국어"), "ko-kr": ("ko", "한국어"), "korean": ("ko", "한국어"), "한국어": ("ko", "한국어"),
@@ -73,8 +178,6 @@ LANG_MAP = {
     "es": ("es", "Español"), "es-es": ("es", "Español"), "es-mx": ("es", "Español"), "spanish": ("es", "Español"), "스페인어": ("es", "Español"),
     "zh": ("zh", "中文"), "zh-cn": ("zh", "中文"), "chinese": ("zh", "中文"), "중국어": ("zh", "中文"), "中文": ("zh", "中文"),
 }
-
-# pytrends 국가 키 (pn, locale)
 PNT_MAP = {
     "KR": ("south_korea", "ko-KR"),
     "US": ("united_states", "en-US"),
@@ -83,7 +186,6 @@ PNT_MAP = {
     "MX": ("mexico", "es-419"),
     "CN": ("china", "zh-CN"),  # pytrends 미지원/불안정 가능성 있음
 }
-
 LANG_DEFAULT_COUNTRY = {"ko": "KR", "en": "US", "ja": "JP", "es": "ES", "zh": "CN"}
 
 def normalize_lang_and_country(lang_in: str, country_in: Optional[str]) -> Tuple[str, str, str]:
@@ -117,7 +219,6 @@ EXAMPLES = {
            "AI一分钟做爆款短视频，真的假的？\n"
            "便利店神组合TOP5，最后一个直接炸裂！\n"),
 }
-
 HOOKS = {
     "ko": ["이건 몰랐죠?", "소름", "충격", "실화냐", "TOP5", "ㄹㅇ", "대박", "미쳤다", "반전 있음", "한줄 요약", "지금 안 보면 손해", "30초 컷", "끝까지 보세요", "알고 보면 더 소름", "초간단"],
     "en": ["did you know?", "shocking", "insane", "is this real?", "TOP 5", "no way", "viral", "mind-blown", "plot twist", "one-line tip", "don’t miss this", "30s hack", "watch till the end", "low-key genius"],
@@ -125,7 +226,6 @@ HOOKS = {
     "es": ["¿lo sabías?", "impactante", "una locura", "¿es real?", "TOP 5", "no puede ser", "viral", "mente en blanco", "giro inesperado", "tip en una línea", "no te lo pierdas", "truco de 30s", "hasta el final", "dato clave"],
     "zh": ["震惊", "太离谱了吧", "你绝对想不到", "TOP5", "网友都惊呆了", "太炸了", "笑死", "破防了", "这操作逆天", "真敢拍", "不看后悔", "细思极恐", "绝了", "太狠了"],
 }
-
 TAILS = {
     "ko": ["방법 공개", "시청자 후기 터짐", "전/후 비교", "초보도 가능", "숨은 기능", "반전 주의", "실전 꿀팁", "모르면 손해", "한 번에 끝", "바로 써먹는 스킬", "핵심만 요약", "정리해드림"],
     "en": ["method revealed", "viewers went crazy", "before vs after", "beginner friendly", "hidden feature", "twist ahead", "pro tips", "don’t miss out", "done in one go", "instant skill", "key points only", "quick recap"],
@@ -133,7 +233,6 @@ TAILS = {
     "es": ["método revelado", "los viewers enloquecen", "antes vs después", "para principiantes", "función oculta", "plot twist", "tips pro", "no te lo pierdas", "todo de una", "skill al instante", "puntos clave", "resumen rápido"],
     "zh": ["完整版来了", "网友热议中", "看完我沉默了", "前后对比离谱", "新手也能做", "隐藏功能曝光", "背后原因震撼", "不看真的亏", "教程来了", "全网爆火", "亲测有效", "干货总结"],
 }
-
 PROMPT_HEADER = {
     "ko": "역할: 당신은 해당 국가의 유튜브 쇼츠 카피라이터다.\n과업: 아래 ‘주제’로 현지 시청자가 많이 보게끔 바이럴 제목을 생성하라.\n",
     "en": "Role: You are a YouTube Shorts copywriter for the target country.\nTask: Generate viral titles so local viewers want to watch.\n",
@@ -236,22 +335,18 @@ _LEAK_PATTERNS = [
     r'^\s*[-•]\s*$',
     r'^\s*각 제목은',
 ]
-
 def _fingerprint(s: str) -> str:
     return re.sub(r"[\s\W]+", "", s.lower())
-
 def _valid_len(s: str, lang_code: str) -> bool:
     if lang_code in ("en", "es"):
         words = re.findall(r"\w+", s)
         return 3 <= len(words) <= 14
     return 6 <= len(s) <= 36  # ko/ja/zh
-
 def _leak(s: str) -> bool:
     for p in _LEAK_PATTERNS:
         if re.search(p, s, flags=re.I):
             return True
     return False
-
 def postprocess(raw: str, lang_code: str, numberless: bool = True) -> List[str]:
     if not raw: return []
     raw = re.sub(r"<extra_id_\d+>", "", raw)
@@ -317,7 +412,6 @@ def generate_titles(topic: str, count: int, req, lang_code: str, pretty_lang: st
 
 # ===== 토픽 포지(영상화 가능한 주제 자동 생성) =====
 TOPIC_CATEGORIES = ("ai-slice","ai-teach","ai-transform","ai-battle","ai-micro","mixed")
-
 OBJ = {
     "ko": {
         "fruit": ["수박","망고","파인애플","사과","키위","용과","복숭아","바나나","코코넛","포도"],
@@ -342,7 +436,6 @@ OBJ = {
 }
 def _obj(lang: str) -> dict:
     return OBJ.get(lang, OBJ["ko"])
-
 def forge_topic(lang_code: str, category: str) -> str:
     o = _obj(lang_code)
     if category == "ai-slice":
@@ -366,6 +459,44 @@ def forge_topic(lang_code: str, category: str) -> str:
         return f"AI 마이크로 세계 탐험: {t} 속 {m} 확대 관찰"
     return forge_topic(lang_code, random.choice(TOPIC_CATEGORIES[:-1]))  # mixed
 
+# ===== 쇼츠 프롬프트(샷리스트) 생성 =====
+def build_shorts_prompt(topic: str, title: str, lang_code: str, style: str, tone: Optional[str], cuts: int, duration_sec: int, emoji: int) -> str:
+    # 컷 개수 보정
+    cuts = max(5, min(8, cuts))
+    per_cut = max(4, min(7, duration_sec // cuts))  # 컷당 초
+    style_note = {
+        "meme-shock": "강한 훅 + 반전 + 리듬. 자막 친화. 짧은 구/명령문.",
+        "wholesome": "공감/힐링 톤. 친절한 말투. 희망/응원.",
+        "informative": "핵심 먼저, 팩트 중심. 간결/신뢰.",
+        "clickbait": "궁금증 유발, 과장 허용(사실 왜곡 금지).",
+    }.get(style, "리듬·호기심 중심.")
+    tone_note = f"추가 톤: {tone}." if tone else ""
+    emj = " 이모지 적당히 사용" if emoji > 0 else " 이모지 사용 금지"
+
+    header = (
+        f"역할: 유튜브 쇼츠 콘텐츠 플래너 & 카피라이터\n"
+        f"목표: 아래 '제목'을 기반으로 {duration_sec}초 이내 영상 프롬프트(대본 지시문)를 생성한다.\n"
+        f"규칙:\n"
+        f"- 초반 3초 강한 훅, 3~4개의 핵심 포인트, 마지막에 액션콜(좋아요/댓글 유도)\n"
+        f"- 짧고 리듬감 있게, 자막용 문장\n"
+        f"- 컷(장면)을 번호로 구분, 각 컷 1~2문장\n"
+        f"- 촬영 소스 없어도 생성형 영상툴에 바로 넣을 수 있는 묘사 포함\n"
+        f"- 스타일: {style} ({style_note}), 언어: {lang_code}, {tone_note}{emj}\n"
+        f"- 컷 수: {cuts}개, 컷당 {per_cut}초 내외\n\n"
+        f"메타:\n"
+        f"• 주제(Topic): {topic}\n"
+        f"• 제목(Title): {title}\n\n"
+        f"출력 포맷 예시(형식만 참고):\n"
+        f"1) 훅: 시청자 호기심 폭발 한 줄\n"
+        f"2) 포인트1: 구체 예시/브랜드/가격 등\n"
+        f"3) 포인트2: 비교/반전/주의\n"
+        f"4) 포인트3: 실전 팁/바로 써먹기\n"
+        f"5) 마무리: 한 줄 요약 + 액션콜(댓글/좋아요 유도)\n\n"
+        f"이제 위 규칙을 철저히 지켜 {cuts}개의 컷으로 한국어 프롬프트를 출력하라."
+    )
+    return header
+
+# ===== 엔드포인트 =====
 @app.get("/topic/auto", response_model=TopicResp)
 def topic_auto(
     count: int = Query(10, ge=1, le=100),
@@ -393,7 +524,7 @@ def topic_auto(
 
 @app.get("/")
 def root():
-    return JSONResponse({"message": "NocoAI Title Forge (Shorts Optimized, Localized + Topic Forge). Try GET /topic/auto, POST /generate, or GET /generate/auto"}, media_type="application/json; charset=utf-8")
+    return JSONResponse({"message": "NocoAI Title & Prompt Forge. Try GET /topic/auto, POST /generate, GET /generate/auto, POST /prompt/shorts"}, media_type="application/json; charset=utf-8")
 
 @app.get("/health")
 def health():
@@ -437,3 +568,26 @@ def generate_auto(
     titles, used_prompt = generate_titles(topic, count, req, lang_code, pretty_lang)
     used_source = trend_source if (trend_source == "pytrends" and HAS_PYTRENDS) else "seed"
     return JSONResponse({"topic": topic, "titles": titles, "used_prompt": used_prompt, "trend_source": used_source}, media_type="application/json; charset=utf-8")
+
+@app.post("/prompt/shorts", response_model=ShortsPromptResp)
+def prompt_shorts(body: ShortsPromptReq):
+    lang_code, _, _ = normalize_lang_and_country(body.lang, None)
+    prompt = build_shorts_prompt(
+        topic=body.topic,
+        title=body.title,
+        lang_code=lang_code,
+        style=body.style,
+        tone=body.tone,
+        cuts=body.cuts,
+        duration_sec=body.duration_sec,
+        emoji=body.emoji
+    )
+    return JSONResponse({
+        "topic": body.topic,
+        "title": body.title,
+        "prompt": prompt,
+        "lang": lang_code,
+        "style": body.style,
+        "cuts": body.cuts,
+        "duration_sec": body.duration_sec
+    }, media_type="application/json; charset=utf-8")