Update app.py

app.py

@@ -42,10 +42,10 @@ API_URL = "https://api.friendli.ai/dedicated/v1/chat/completions"
 MODEL_ID = "dep86pjolcjjnv8"
 DB_PATH = "webnovel_sessions_v1.db"
 
-# Target settings for web novel
+# Target settings for web novel - UPDATED FOR LONGER EPISODES
 TARGET_EPISODES = 40  # 40화 완결
-WORDS_PER_EPISODE = 200  # 각 화당 200-300 단어
-TARGET_WORDS = TARGET_EPISODES * WORDS_PER_EPISODE  # 총 8000 단어
+WORDS_PER_EPISODE = 400  # 각 화당 400-600 단어 (기존 200-300에서 증가)
+TARGET_WORDS = TARGET_EPISODES * WORDS_PER_EPISODE  # 총 16000 단어
 
 # Web novel genres
 WEBNOVEL_GENRES = {
@@ -397,7 +397,7 @@ class WebNovelSystem:
 
 **주제:** {query}
 **장르:** {genre}
-**목표:** 40화 완결, 총 8,000단어
+**목표:** 40화 완결, 총 16,000단어
 
 **장르 필수 요소:**
 - 핵심 요소: {', '.join(genre_info.get('key_elements', []))}
@@ -405,25 +405,26 @@ class WebNovelSystem:
 - 필수 포함: {', '.join(genre_info.get('must_have', []))}
 
 **전체 구성:**
-1. **1-5화**: 흥미로운 도입부, 주인공과 세계관 소개
-2. **6-15화**: 갈등 상승, 주요 인물 관계 형성
-3. **16-25화**: 중간 클라이맥스, 반전
-4. **26-35화**: 최종 갈등으로 치닫기
-5. **36-40화**: 클라이맥스와 결말
+1. **1-5화**: 흥미로운 도입부, 주인공과 세계관 소개, 핵심 갈등 제시
+2. **6-15화**: 갈등 상승, 주요 인물 관계 형성, 첫 번째 위기
+3. **16-25화**: 중간 클라이맥스, 반전, 새로운 진실 발견
+4. **26-35화**: 최종 갈등으로 치닫기, 모든 갈등의 수렴
+5. **36-40화**: 대결전, 클라이맥스와 결말, 감동적 마무리
 
 **각 화 구성 원칙:**
-- 200-300단어 분량
+- 400-600단어 분량 (충실한 내용)
+- 3-4개의 주요 장면 포함
 - 매 화 끝 강력한 후크
 - 빠른 전개와 몰입감
 - 독자가 다음 화를 기다리게 만들기
 
-구체적인 40화 플롯라인을 제시하세요.""",
+구체적인 40화 플롯라인을 제시하세요. 각 화마다 핵심 사건과 전개를 명시하세요.""",
 
             "English": f"""Plan a Korean-style web novel for {genre} genre.
 
 **Theme:** {query}
 **Genre:** {genre}
-**Goal:** 40 episodes, total 8,000 words
+**Goal:** 40 episodes, total 16,000 words
 
 **Genre Requirements:**
 - Key elements: {', '.join(genre_info.get('key_elements', []))}
@@ -431,26 +432,27 @@ class WebNovelSystem:
 - Must include: {', '.join(genre_info.get('must_have', []))}
 
 **Overall Structure:**
-1. **Episodes 1-5**: Engaging introduction, protagonist and world
-2. **Episodes 6-15**: Rising conflict, main relationships
-3. **Episodes 16-25**: Mid climax, plot twist
-4. **Episodes 26-35**: Building to final conflict
-5. **Episodes 36-40**: Climax and resolution
+1. **Episodes 1-5**: Engaging introduction, protagonist and world, core conflict
+2. **Episodes 6-15**: Rising conflict, main relationships, first crisis
+3. **Episodes 16-25**: Mid climax, plot twist, new revelations
+4. **Episodes 26-35**: Building to final conflict, convergence of all conflicts
+5. **Episodes 36-40**: Final battle, climax and resolution, emotional closure
 
 **Episode Principles:**
-- 200-300 words each
+- 400-600 words each (substantial content)
+- 3-4 major scenes per episode
 - Strong hook at episode end
 - Fast pacing and immersion
 - Make readers crave next episode
 
-Provide detailed 40-episode plotline."""
+Provide detailed 40-episode plotline with key events for each episode."""
         }
         
         return lang_prompts.get(language, lang_prompts["Korean"])
 
     def create_episode_prompt(self, episode_num: int, plot_outline: str, 
                             previous_content: str, genre: str, language: str) -> str:
-        """Create prompt for individual episode"""
+        """Create prompt for individual episode - UPDATED FOR LONGER CONTENT"""
         genre_info = GENRE_ELEMENTS.get(genre, {})
         hooks = EPISODE_HOOKS.get(genre, ["다음 순간, 충격적인 일이..."])
         
@@ -458,48 +460,78 @@ Provide detailed 40-episode plotline."""
             "Korean": f"""웹소설 {episode_num}화를 작성하세요.
 
 **장르:** {genre}
-**분량:** 200-300단어 (엄격히 준수)
+**분량:** 400-600단어 (엄격히 준수 - 충실한 내용으로)
 
 **전체 플롯에서 {episode_num}화 내용:**
 {self._extract_episode_plan(plot_outline, episode_num)}
 
 **이전 내용 요약:**
-{previous_content[-1000:] if previous_content else "첫 화입니다"}
+{previous_content[-1500:] if previous_content else "첫 화입니다"}
 
 **작성 지침:**
-1. 즉시 본론으로 - 불필요한 묘사 금지
-2. 대화와 행동 중심
-3. 독자를 사로잡는 전개
-4. 감정적 몰입 유도
-5. 마지막은 반드시 다음 화를 읽고 싶게 만드는 후크
+1. **구성**: 3-4개의 주요 장면으로 구성
+   - 도입부: 이전 화 연결 및 현재 상황
+   - 전개부: 2-3개의 핵심 사건/대화
+   - 클라이맥스: 긴장감 최고조
+   - 후크: 다음 화 예고
+
+2. **필수 요소:**
+   - 생생한 대화와 행동 묘사
+   - 캐릭터 감정과 내면 갈등
+   - 장면 전환과 템포 조절
+   - 독자 몰입을 위한 감각적 묘사
+
+3. **장르별 특색:**
+   - {genre_info.get('episode_structure', '빠른 전개')}
+   - 핵심 요소 1개 이상 포함
+
+4. **분량 배분:**
+   - 도입 (50-80단어)
+   - 주요 전개 (250-350단어)
+   - 클라이맥스와 후크 (100-150단어)
 
 **참고 후크 예시:**
 {random.choice(hooks)}
 
-빠르고 중독성 있는 {episode_num}화를 작성하세요.""",
+{episode_num}화를 풍성하고 몰입감 있게 작성하세요. 반드시 400-600단어로 작성하세요.""",
 
             "English": f"""Write episode {episode_num} of the web novel.
 
 **Genre:** {genre}
-**Length:** 200-300 words (strict)
+**Length:** 400-600 words (strict - with substantial content)
 
 **Episode {episode_num} from plot:**
 {self._extract_episode_plan(plot_outline, episode_num)}
 
 **Previous content:**
-{previous_content[-1000:] if previous_content else "First episode"}
+{previous_content[-1500:] if previous_content else "First episode"}
 
 **Guidelines:**
-1. Jump straight to action - no filler
-2. Dialogue and action focused
-3. Gripping development
-4. Emotional immersion
-5. End with hook for next episode
+1. **Structure**: 3-4 major scenes
+   - Opening: Connect from previous, current situation
+   - Development: 2-3 key events/dialogues
+   - Climax: Peak tension
+   - Hook: Next episode teaser
+
+2. **Essential elements:**
+   - Vivid dialogue and action
+   - Character emotions and conflicts
+   - Scene transitions and pacing
+   - Sensory details for immersion
+
+3. **Genre specifics:**
+   - {genre_info.get('episode_structure', 'Fast pacing')}
+   - Include at least 1 core element
+
+4. **Word distribution:**
+   - Opening (50-80 words)
+   - Main development (250-350 words)
+   - Climax and hook (100-150 words)
 
 **Hook example:**
 {random.choice(hooks)}
 
-Write fast-paced, addictive episode {episode_num}."""
+Write rich, immersive episode {episode_num}. Must be 400-600 words."""
         }
         
         return lang_prompts.get(language, lang_prompts["Korean"])
@@ -577,7 +609,8 @@ Provide specific improvements."""
             system_prompts = self.get_system_prompts(language)
             full_messages = [{"role": "system", "content": system_prompts.get(role, "")}, *messages]
             
-            max_tokens = 3000 if role == "writer" else 10000
+            # Increased max_tokens for longer episodes
+            max_tokens = 5000 if role == "writer" else 10000
             
             payload = {
                 "model": self.model_id,
@@ -639,18 +672,21 @@ Provide specific improvements."""
             yield f"❌ Error occurred: {str(e)}"
 
     def get_system_prompts(self, language: str) -> Dict[str, str]:
-        """System prompts for web novel roles"""
+        """System prompts for web novel roles - UPDATED FOR LONGER EPISODES"""
         base_prompts = {
             "Korean": {
                 "planner": """당신은 한국 웹소설 시장을 완벽히 이해하는 기획자입니다.
 독자를 중독시키는 플롯과 전개를 설계합니다.
 장르별 관습과 독자 기대를 정확히 파악합니다.
-40화 완결 구조로 완벽한 기승전결을 만듭니다.""",
+40화 완결 구조로 완벽한 기승전결을 만듭니다.
+각 화마다 충실한 내용과 전개를 계획합니다.""",
                 
                 "writer": """당신은 독자를 사로잡는 웹소설 작가입니다.
-간결하고 몰입감 있는 문체를 구사합니다.
-매 화 끝에 다음 화를 기다리게 만듭니다.
-대화와 행동으로 이야기를 전개합니다.""",
+풍부하고 몰입감 있는 문체를 구사합니다.
+각 화를 400-600단어로 충실하게 작성합니다.
+여러 장면과 전환을 통해 이야기를 전개합니다.
+대화, 행동, 내면 묘사를 균형있게 배치합니다.
+매 화 끝에 강력한 후크로 다음 화를 기다리게 만듭니다.""",
                 
                 "critic": """당신은 웹소설 독자의 마음을 읽는 평론가입니다.
 재미와 몰입감을 최우선으로 평가합니다.
@@ -661,12 +697,15 @@ Provide specific improvements."""
                 "planner": """You perfectly understand the Korean web novel market.
 Design addictive plots and developments.
 Accurately grasp genre conventions and reader expectations.
-Create perfect story structure in 40 episodes.""",
+Create perfect story structure in 40 episodes.
+Plan substantial content and development for each episode.""",
                 
                 "writer": """You are a web novelist who captivates readers.
-Use concise and immersive writing style.
-Make readers crave the next episode.
-Develop story through dialogue and action.""",
+Use rich and immersive writing style.
+Write each episode with 400-600 words faithfully.
+Develop story through multiple scenes and transitions.
+Balance dialogue, action, and inner descriptions.
+End each episode with powerful hook for next.""",
                 
                 "critic": """You read web novel readers' minds.
 Prioritize fun and immersion in evaluation.