Merge pull request #24 from DropThe8bit/feature/story

hhongyeahh · web-flow · commit d90db6ecb2c2 · 2025-08-31T18:00:33.000+09:00
[refactor] 질문/답변 기반 줄거리 생성 로직 프롬프트 구체화
diff --git a/everTale/app/api.py b/everTale/app/api.py
@@ -38,23 +38,31 @@ def create_next_story(request: dto.NextStoryRequest):
 
 
 @router.post("/question")
-def create_question(request: dto.QuestionRequest):
-    prompt = (
-        f"이전 장면:\n{request.previous}\n\n"
-        "아이의 참여를 유도하기 위해 질문 하나를 던져줘. 예를 들어 '주인공은 어떻게 해야 할까?'처럼 "
-        "선택이나 상상을 이끌어낼 수 있도록 해줘."
+def create_question(request: dto.NextStoryRequest):
+    question = story_service.generate_question_for_next_story(
+        previous=request.previous,
+        page_number=request.pageNum,
+        genre=request.genre,
+        name=request.name,
+        age=request.age,
+        gender=request.gender,
+        personalities=request.personalities
     )
-    question = story_service.generate_story(prompt)
     return {"message": question}
 
 @router.post("/next-from-answer")
 def create_next_story_with_answer(request: dto.NextFromAnswerRequest):
-    prompt = (
-        f"이전 장면:\n{request.previous}\n"
-        f"아이의 대답: {request.answer}\n\n"
-        "이 대답을 반영해서 다음 장면의 줄거리를 상상력 있게 이어서 써줘. 2~3문장으로 자연스럽게 전개해줘."
+    story = story_service.generate_story_from_question_and_answer(
+        question=request.question,
+        answer=request.answer,
+        previous=request.previous,
+        page_number=request.pageNum,
+        genre=request.genre,
+        name=request.name,
+        age=request.age,
+        gender=request.gender,
+        personalities=request.personalities
     )
-    story = story_service.generate_story(prompt)
     return {"message": story}
 
 @router.post("/init-character-image")
diff --git a/everTale/app/dto.py b/everTale/app/dto.py
@@ -22,29 +22,36 @@ class NextStoryRequest(BaseModel):
     gender: str = Field(..., json_schema_extra={"example": "female"})
     personalities: List[str] = Field(..., json_schema_extra={"example": ["용감함", "씩씩함"]})
 
-
-class QuestionRequest(BaseModel):
-    previous: str = Field(..., description="이전 줄거리")
-
-    class Config:
-        json_schema_extra = {
-            "example": {
-                "previous": "주인공은 친구와 함께 숲을 탐험하고 있었습니다."
-            }
-        }
+from pydantic import BaseModel, Field
+from typing import List
 
 class NextFromAnswerRequest(BaseModel):
-    previous: str = Field(..., description="이전 줄거리")
+    question: str = Field(..., description="이전에 던진 질문")
     answer: str = Field(..., description="아이의 대답")
+    previous: str = Field(..., description="이전 줄거리")
+    pageNum: int = Field(..., description="현재 페이지 번호")
+    genre: str = Field(..., description="스토리 장르")
+    name: str = Field(..., description="주인공 이름")
+    age: int = Field(..., description="주인공 나이")
+    gender: str = Field(..., description="주인공 성별")
+    personalities: List[str] = Field(..., description="주인공 성격 리스트")
 
     class Config:
         json_schema_extra = {
             "example": {
-                "previous": "주인공은 친구와 함께 숲을 탐험하고 있었습니다.",
-                "answer": "주인공은 용감하게 동굴 안으로 들어가야 해요."
+                "question": "주인공은 어떻게 해야 할까?",
+                "answer": "용감하게 문을 열어야 해요.",
+                "previous": "토토로는 첫 여행지로 북부의 마탑으로 향했어요...",
+                "pageNum": 2,
+                "genre": "ADVENTURE",
+                "name": "토토로",
+                "age": 8,
+                "gender": "female",
+                "personalities": ["용감함", "씩씩함"]
             }
         }
 
+
 class QuizRequest(BaseModel):
     previous: str = Field(...,json_schema_extra={"example":"숲 초입 나무 밑에서 부스럭 부스럭 소리가 들렸어요. 토로로는 나무로 가까이 다가갔어요. 찍찍찍..찍찍 소리는 점점 커지는데... 맙소사 아기 다람쥐가 나무에서 떨어서 풀 사이에 힘겹게 숨을 쉬고 있었어요."})
 
diff --git a/everTale/app/service/story_service.py b/everTale/app/service/story_service.py
@@ -186,7 +186,7 @@ def make_initial_prompt(genre, world_view, name, age, gender, personalities) ->
         f"[금지]\n"
         f"- " + "\n- ".join(guardrails["forbidden"]) + "\n\n"                                                                                     
         f"[출력 형식]\n"
-        f"- 한국어로 3~5문장.\n"
+        f"- 한국어로 7~8문장.\n"
         f"- 문단 머리표, 번호, 메타설명 없이 순수 서사만 출력.\n"
         f"- 마지막 문장은 다음 장면이 기대되게 가볍게 여운을 남김.\n\n"
         f"[이어서 작성]\n"
@@ -207,7 +207,7 @@ def generate_prompt_for_next_story(
     g = genre_guides.get(genre, genre_guides["ADVENTURE"])
     beat = beat_map.get(page_number, beat_map[1])
 
-    return (
+    prompt = f"""
         f"[역할]\n"
         f"너는 유아·아동 대상 동화 작가야. 번역투를 피하고 자연스러운 한국어 구어체로, 이전 줄거리에 자연스럽게 이어지게 작성해줘.\n\n"
         f"[장르]\n"
@@ -235,6 +235,114 @@ def generate_prompt_for_next_story(
         f"- 8페이지면 따뜻하게 마무리.\n\n"
         f"[이어서 작성]\n"
         f"위의 정보를 바탕으로 {page_number}번째 장면을 이어서 써줘."
-    )
+    """
+    return generate_story(prompt)
+
+
+def generate_question_for_next_story(
+    page_number: int,
+    genre: str,
+    previous: str,
+    name: str,
+    age: int,
+    gender: str,
+    personalities: list[str],
+) -> str:
+    personality_str = ", ".join(personalities)
+    g = genre_guides.get(genre, genre_guides["ADVENTURE"])
+    beat = beat_map.get(page_number, beat_map[1])
+
+    prompt = f"""
+        f"[역할]\n"
+        f"너는 유아·아동 대상 동화 작가야. 이전 줄거리와 주인공 정보를 참고해, 아이가 다음 이야기를 상상할 수 있도록 질문을 만들어줘.\n\n"
+        f"[장르]\n"
+        f"{genre} — 톤: {g['tone']} | 배경: {g['setting']} | 힌트: {g['hint']}\n\n"
+        f"[현재 장면] {page_number}페이지 / {beat['label']}\n"
+        f"- 장면 목적: {beat['goal']}\n"
+        f"- 반드시 반영: 이전 줄거리와 이어져야 하고, 주인공의 성격과 나이에 맞는 질문이어야 함.\n"
+        f"- 힌트: {', '.join(beat.get('hint', [])) if beat.get('hint') else '없음'}\n\n"
+        f"[이전 줄거리 요약]\n{previous}\n\n"
+        f"[주인공 정보]\n"
+        f"- 이름: {name}, 나이: {age}살, 성별: {gender}, 성격: {personality_str}\n\n"
+        f"[출력 규칙]\n"
+        f"- 한국어 질문 한 문장.\n"
+        f"- 아이가 선택하거나 상상으로 답할 수 있게 열려 있어야 함.\n"
+        f"- 예: '~해야 할까?' 등.\n"
+        f"- 기승전결의 흐름에 어울리도록 상황을 반영.\n\n"
+        f"[작성 요청]\n"
+        f"위 정보를 바탕으로 다음 줄거리를 이어가기 위한 질문을 1개 생성해줘."
+    """
+    return generate_story(prompt)
 
+def generate_story_from_question_and_answer(
+    question: str,
+    answer: str,
+    previous: str,
+    page_number: int,
+    genre: str,
+    name: str,
+    age: int,
+    gender: str,
+    personalities: list[str],
+) -> str:
+    # 간단 전처리: 공백 정리 및 과도한 길이 컷
+    def _clean(s: str, max_len: int = 1200) -> str:
+        s = " ".join((s or "").split())
+        return s[:max_len]
+
+    question = _clean(question, 200)
+    answer = _clean(answer, 300)
+    previous = _clean(previous, 1200)
+
+    personalities_str = ", ".join((personalities or [])[:6])
+    g = genre_guides.get(genre, genre_guides["ADVENTURE"])
+    beat = beat_map.get(page_number, beat_map[1])
+
+    prompt = f"""
+[역할]
+너는 유아·아동 대상 대화형 동화 작가다. 아이가 이해하기 쉬운 말로 따뜻하게 쓴다.
 
+[장르]
+{genre} — 톤: {g['tone']} | 배경: {g['setting']} | 힌트: {g.get('hint','')}
+
+[현재 장면] {page_number}페이지 / {beat['label']}
+- 장면 목적: {beat['goal']}
+- 반드시 포함: {', '.join(beat.get('must', [])) if beat.get('must') else '자연스러운 전개'}
+- 힌트: {', '.join(beat.get('hint', [])) if beat.get('hint') else '없음'}
+- 장면 마무리: {beat.get('ending','')} (선택적)
+
+[이전 줄거리]
+{previous}
+
+[주인공 정보]
+- 이름: {name} ({gender}, {age}세)
+- 성격: {personalities_str}
+
+[아이에게 던진 질문]
+{question}
+
+[아이의 대답]
+{answer}
+
+[작성 지침]
+- 아이의 대답을 **핵심 사건/선택**으로 직접 반영하여 **인과적으로** 이어질 것.
+- {genre} 톤을 유지하고, {name}의 성격({personalities_str})이 행동과 말투에 드러나게 할 것.
+- {beat['label']} 단계의 목적({beat['goal']})을 충족할 것.
+- 과도한 공포·폭력·비하·설교조는 금지. 설명 과다 대신 **장면 이미지/행동** 위주.
+- 대사는 0~1문장만 필요 시 사용.
+
+[스타일 규칙]
+- """ + "\n- ".join(guardrails["style_rules"]) + f"""
+
+[금지]
+- """ + "\n- ".join(guardrails["forbidden"]) + f"""
+
+[출력 형식]
+- 한국어로 **7~8문장**의 순수 서사만 출력(머리표/번호/해설 금지).
+- 8페이지면 따뜻하게 마무리, 아니면 다음 장면으로 이어질 수 있도록 작성한다..
+
+[작성 요청]
+위 정보를 바탕으로 다음 장면을 작성하라.
+""".strip()
+
+    return generate_story(prompt)
