Update app.py

app.py

@@ -48,9 +48,9 @@ selected_language = "English"  # 기본 언어
 DB_PATH = "novel_sessions.db"
 db_lock = threading.Lock()
 
-# Stage 번호 상수
-WRITER_DRAFT_STAGES = [3, 6, 9, 12, 15, 18, 21, 24, 27, 30]  # 작가 초안
-WRITER_REVISION_STAGES = [5, 8, 11, 14, 17, 20, 23, 26, 29, 32]  # 작가 수정본
+# Stage 번호 상수 - 16명의 작가 반영
+WRITER_DRAFT_STAGES = [3, 6, 9, 12, 15, 18, 21, 24, 27, 30, 33, 36, 39, 42, 45, 48]  # 작가 초안
+WRITER_REVISION_STAGES = [5, 8, 11, 14, 17, 20, 23, 26, 29, 32, 35, 38, 41, 44, 47, 50]  # 작가 수정본
 
 class NovelDatabase:
     """Novel session management database"""
@@ -89,6 +89,7 @@ class NovelDatabase:
                     content TEXT,
                     status TEXT DEFAULT 'pending',
                     created_at TEXT DEFAULT (datetime('now')),
+                    updated_at TEXT DEFAULT (datetime('now')),
                     FOREIGN KEY (session_id) REFERENCES sessions(session_id),
                     UNIQUE(session_id, stage_number)
                 )
@@ -126,24 +127,24 @@ class NovelDatabase:
             conn.commit()
         
         return session_id
-
+    
     @staticmethod
     def save_stage(session_id: str, stage_number: int, stage_name: str, 
                    role: str, content: str, status: str = 'complete'):
-        """스테이지 내용 저장 - 전체 내용 저장"""
+        """Save stage content - 전체 내용 저장"""
         with NovelDatabase.get_db() as conn:
             cursor = conn.cursor()
             
-            # UPSERT 작업 (updated_at 제외)
+            # UPSERT operation
             cursor.execute('''
                 INSERT INTO stages (session_id, stage_number, stage_name, role, content, status)
                 VALUES (?, ?, ?, ?, ?, ?)
                 ON CONFLICT(session_id, stage_number) 
-                DO UPDATE SET content=?, status=?, stage_name=?
+                DO UPDATE SET content=?, status=?, stage_name=?, updated_at=datetime('now')
             ''', (session_id, stage_number, stage_name, role, content, status,
                   content, status, stage_name))
             
-            # 세션 업데이트
+            # Update session
             cursor.execute('''
                 UPDATE sessions 
                 SET updated_at = datetime('now'), current_stage = ?
@@ -151,10 +152,7 @@ class NovelDatabase:
             ''', (stage_number, session_id))
             
             conn.commit()
-            logger.info(f"{session_id} 세션의 스테이지 {stage_number} 저장됨, 내용 길이: {len(content)}")
-
-
-
+            logger.info(f"Saved stage {stage_number} for session {session_id}, content length: {len(content)}")
     
     @staticmethod
     def get_session(session_id: str) -> Optional[Dict]:
@@ -178,11 +176,14 @@ class NovelDatabase:
     
     @staticmethod
     def get_all_writer_content(session_id: str) -> str:
-        """모든 작가의 수정본 내용을 가져와서 합치기 - 50페이지 전체"""
+        """모든 작가의 수정본 내용을 가져와서 합치기 - 48페이지 전체"""
         with NovelDatabase.get_db() as conn:
             cursor = conn.cursor()
             
             all_content = []
+            writer_count = 0
+            total_word_count = 0
+            
             for stage_num in WRITER_REVISION_STAGES:
                 cursor.execute('''
                     SELECT content, stage_name FROM stages 
@@ -197,11 +198,18 @@ class NovelDatabase:
                     clean_content = clean_content.strip()
                     
                     if clean_content:
+                        writer_count += 1
+                        word_count = len(clean_content.split())
+                        total_word_count += word_count
                         all_content.append(clean_content)
-                        logger.info(f"Retrieved writer content from stage {stage_num}, length: {len(clean_content)}")
+                        logger.info(f"Writer {writer_count} (stage {stage_num}): {word_count} words")
             
             full_content = '\n\n'.join(all_content)
-            logger.info(f"Total writer content length: {len(full_content)}, from {len(all_content)} writers")
+            logger.info(f"Total: {writer_count} writers, {total_word_count} words, {total_word_count/500:.1f} pages")
+            
+            # 내용이 너무 짧으면 경고
+            if total_word_count < 20000:  # 48페이지는 약 24000 단어
+                logger.warning(f"Content too short! Only {total_word_count} words instead of ~24000")
             
             return full_content
     
@@ -241,6 +249,48 @@ class NovelDatabase:
         except:
             # Fallback to SQLite default format
             return datetime.strptime(datetime_str, "%Y-%m-%d %H:%M:%S")
+    
+    @staticmethod
+    def verify_novel_content(session_id: str) -> Dict[str, Any]:
+        """세션의 전체 소설 내용 검증"""
+        with NovelDatabase.get_db() as conn:
+            cursor = conn.cursor()
+            
+            # 모�� 작가 수정본 확인
+            cursor.execute(f'''
+                SELECT stage_number, stage_name, LENGTH(content) as content_length
+                FROM stages
+                WHERE session_id = ? AND stage_number IN ({','.join(map(str, WRITER_REVISION_STAGES))})
+                ORDER BY stage_number
+            ''', (session_id,))
+            
+            results = []
+            total_length = 0
+            
+            for row in cursor.fetchall():
+                results.append({
+                    'stage': row['stage_number'],
+                    'name': row['stage_name'],
+                    'length': row['content_length'] or 0
+                })
+                total_length += row['content_length'] or 0
+            
+            # 최종 소설 확인
+            cursor.execute('''
+                SELECT LENGTH(final_novel) as final_length
+                FROM sessions
+                WHERE session_id = ?
+            ''', (session_id,))
+            
+            final_row = cursor.fetchone()
+            final_length = final_row['final_length'] if final_row else 0
+            
+            return {
+                'writer_stages': results,
+                'total_writer_content': total_length,
+                'final_novel_length': final_length,
+                'expected_length': 120000  # 48페이지 * 2500자
+            }
 
 class NovelWritingSystem:
     def __init__(self):
@@ -269,13 +319,13 @@ class NovelWritingSystem:
         }
     
     def create_director_initial_prompt(self, user_query: str, language: str = "English") -> str:
-        """Director AI initial prompt - Novel planning"""
+        """Director AI initial prompt - Novel planning for 16 writers"""
         if language == "Korean":
-            return f"""당신은 50페이지 분량의 중편 소설을 기획하는 문학 감독자입니다.
+            return f"""당신은 48페이지 분량의 중편 소설을 기획하는 문학 감독자입니다.
 
 사용자 요청: {user_query}
 
-다음 요소들을 체계적으로 구성하여 50페이지 중편 소설의 기초를 만드세요:
+다음 요소들을 체계적으로 구성하여 48페이지 중편 소설의 기초를 만드세요:
 
 1. **주제와 장르**
    - 핵심 주제와 메시지
@@ -291,22 +341,26 @@ class NovelWritingSystem:
    - 갈등 구조
    - 감정적 연결고리
 
-4. **서사 구조** (50페이지를 10개 파트로 나누어 각 5페이지)
+4. **서사 구조** (48페이지를 16개 파트로 나누어 각 3페이지)
    | 파트 | 페이지 | 주요 사건 | 긴장도 | 인물 발전 |
    |------|--------|-----------|---------|-----------|
+   | 1 | 1-3 | | | |
+   | 2 | 4-6 | | | |
+   | ... | ... | | | |
+   | 16 | 46-48 | | | |
 
 5. **세계관 설정**
    - 시공간적 배경
    - 사회적/문화적 맥락
    - 분위기와 톤
 
-각 작성자가 5페이지씩 작성할 수 있도록 명확한 가이드라인을 제시하세요."""
+각 작성자가 3페이지씩 작성할 수 있도록 명확한 가이드라인을 제시하세요."""
         else:
-            return f"""You are a literary director planning a 50-page novella.
+            return f"""You are a literary director planning a 48-page novella.
 
 User Request: {user_query}
 
-Systematically compose the following elements to create the foundation for a 50-page novella:
+Systematically compose the following elements to create the foundation for a 48-page novella:
 
 1. **Theme and Genre**
    - Core theme and message
@@ -322,16 +376,20 @@ Systematically compose the following elements to create the foundation for a 50-
    - Conflict structure
    - Emotional connections
 
-4. **Narrative Structure** (divide 50 pages into 10 parts, 5 pages each)
+4. **Narrative Structure** (divide 48 pages into 16 parts, 3 pages each)
    | Part | Pages | Main Events | Tension | Character Development |
    |------|-------|-------------|---------|---------------------|
+   | 1 | 1-3 | | | |
+   | 2 | 4-6 | | | |
+   | ... | ... | | | |
+   | 16 | 46-48 | | | |
 
 5. **World Building**
    - Temporal and spatial setting
    - Social/cultural context
    - Atmosphere and tone
 
-Provide clear guidelines for each writer to compose 5 pages."""
+Provide clear guidelines for each writer to compose 3 pages."""
     
     def create_critic_director_prompt(self, director_plan: str, language: str = "English") -> str:
         """Critic's review of director's plan"""
@@ -353,7 +411,7 @@ Provide clear guidelines for each writer to compose 5 pages."""
    |------|------|------|-----------|
 
 3. **구조적 균형**
-   - 각 파트별 분량 배분
+   - 16개 파트별 분량 배분
    - 긴장과 이완의 리듬
    - 전체적인 흐름
 
@@ -363,7 +421,7 @@ Provide clear guidelines for each writer to compose 5 pages."""
    - 기대치 충족도
 
 5. **실행 가능성**
-   - 각 작성자를 위한 가이드라인의 명확성
+   - 16명의 작성자를 위한 가이드라인의 명확성
    - 일관성 유지 방안
    - 잠재적 문제점
 
@@ -386,7 +444,7 @@ Critique from the following perspectives and provide specific improvements:
    |-----------|-----------|------------|-------------|
 
 3. **Structural Balance**
-   - Distribution across parts
+   - Distribution across 16 parts
    - Rhythm of tension and relief
    - Overall flow
 
@@ -396,7 +454,7 @@ Critique from the following perspectives and provide specific improvements:
    - Expectation fulfillment
 
 5. **Feasibility**
-   - Clarity of guidelines for each writer
+   - Clarity of guidelines for 16 writers
    - Consistency maintenance
    - Potential issues
 
@@ -415,7 +473,7 @@ Provide specific and constructive feedback."""
 
 다음을 포함한 수정된 최종 기획을 제시하세요:
 
-1. **수정된 서사 구조**
+1. **수정된 서사 구조** (16개 파트)
    | 파트 | 페이지 | 주요 사건 | 작성 지침 | 주의사항 |
    |------|--------|-----------|-----------|----------|
 
@@ -424,7 +482,7 @@ Provide specific and constructive feedback."""
    - 인물 간 갈등의 구체화
    - 감정선의 변화 추이
 
-3. **각 작성자를 위한 상세 가이드**
+3. **각 작성자를 위한 상세 가이드** (16명)
    - 파트별 시작과 끝 지점
    - 필수 포함 요소
    - 문체와 톤 지침
@@ -441,7 +499,7 @@ Provide specific and constructive feedback."""
    - 전체적 통일성
    - 독자 몰입 유지 방안
 
-모든 작성자가 명확히 이해할 수 있는 최종 마스터플랜을 작성하세요."""
+16명의 작성자가 명확히 이해할 수 있는 최종 마스터플랜을 작성하세요."""
         else:
             return f"""As director, revise the novel plan reflecting the critic's feedback.
 
@@ -453,7 +511,7 @@ Critic Feedback:
 
 Present the revised final plan including:
 
-1. **Revised Narrative Structure**
+1. **Revised Narrative Structure** (16 parts)
    | Part | Pages | Main Events | Writing Guidelines | Cautions |
    |------|-------|-------------|-------------------|----------|
 
@@ -462,7 +520,7 @@ Present the revised final plan including:
    - Concrete conflicts between characters
    - Emotional arc progression
 
-3. **Detailed Guide for Each Writer**
+3. **Detailed Guide for Each Writer** (16 writers)
    - Start and end points for each part
    - Essential elements to include
    - Style and tone guidelines
@@ -479,15 +537,15 @@ Present the revised final plan including:
    - Overall unity
    - Reader engagement maintenance
 
-Create a final masterplan that all writers can clearly understand."""
+Create a final masterplan that all 16 writers can clearly understand."""
     
     def create_writer_prompt(self, writer_number: int, director_plan: str, previous_content: str, language: str = "English") -> str:
-        """Individual writer prompt - 페이지당 500-600단어로 증가"""
-        pages_start = (writer_number - 1) * 5 + 1
-        pages_end = writer_number * 5
+        """Individual writer prompt - 3페이지, 1500-1800단어"""
+        pages_start = (writer_number - 1) * 3 + 1
+        pages_end = writer_number * 3
         
         if language == "Korean":
-            return f"""당신은 작성자 {writer_number}번입니다. 50페이지 중편 소설의 {pages_start}-{pages_end}페이지(5페이지)를 작성하세요.
+            return f"""당신은 작성자 {writer_number}번입니다. 48페이지 중편 소설의 {pages_start}-{pages_end}페이지(3페이지)를 작성하세요.
 
 감독자의 마스터플랜:
 {director_plan}
@@ -497,7 +555,7 @@ Create a final masterplan that all writers can clearly understand."""
 
 다음 지침에 따라 작성하세요:
 
-1. **분량**: 정확히 5페이지 (페이지당 약 500-600단어, 총 2500-3000단어)
+1. **분량**: 정확히 3페이지 (페이지당 약 500-600단어, 총 1500-1800단어)
 2. **연속성**: 이전 내용과 자연스럽게 이어지도록
 3. **일관성**: 
    - 등장인물의 성격과 말투 유지
@@ -512,9 +570,9 @@ Create a final masterplan that all writers can clearly understand."""
    - 독자의 몰입을 해치지 않기
 
 중요: 페이지 구분 표시를 절대 하지 마세요. 자연스럽게 이어지는 서사로 작성하세요.
-반드시 2500-3000단어 분량을 채워주세요."""
+반드시 1500-1800단어 분량을 채워주세요."""
         else:
-            return f"""You are Writer #{writer_number}. Write pages {pages_start}-{pages_end} (5 pages) of the 50-page novella.
+            return f"""You are Writer #{writer_number}. Write pages {pages_start}-{pages_end} (3 pages) of the 48-page novella.
 
 Director's Masterplan:
 {director_plan}
@@ -524,7 +582,7 @@ Director's Masterplan:
 
 Write according to these guidelines:
 
-1. **Length**: Exactly 5 pages (about 500-600 words per page, total 2500-3000 words)
+1. **Length**: Exactly 3 pages (about 500-600 words per page, total 1500-1800 words)
 2. **Continuity**: Flow naturally from previous content
 3. **Consistency**: 
    - Maintain character personalities and speech
@@ -539,7 +597,7 @@ Write according to these guidelines:
    - Keep reader immersion
 
 Important: DO NOT use any page markers. Write as continuous narrative.
-You MUST write 2500-3000 words."""
+You MUST write 1500-1800 words."""
     
     def create_critic_writer_prompt(self, writer_number: int, writer_content: str, director_plan: str, all_previous_content: str, language: str = "English") -> str:
         """Critic's review of individual writer's work"""
@@ -652,7 +710,7 @@ Clearly distinguish between mandatory revisions and optional improvements."""
    - 묘사와 대화 개선
 
 3. **분량 유지**
-   - 여전히 정확히 5페이지 (2500-3000단어)
+   - 여전히 정확히 3페이지 (1500-1800단어)
    - 페이지 구분 표시 절대 금지
 
 4. **연속성 확보**
@@ -660,7 +718,7 @@ Clearly distinguish between mandatory revisions and optional improvements."""
    - 수정으로 인한 새로운 모순 방지
 
 수정된 최종본을 제시하세요. 페이지 마크는 절대 사용하지 마세요.
-반드시 2500-3000단어 분량을 유지하세요."""
+반드시 1500-1800단어 분량을 유지하세요."""
         else:
             return f"""As Writer #{writer_number}, revise based on critic's feedback.
 
@@ -683,7 +741,7 @@ Write a revision reflecting:
    - Improve descriptions and dialogue
 
 3. **Maintain Length**
-   - Still exactly 5 pages (2500-3000 words)
+   - Still exactly 3 pages (1500-1800 words)
    - Absolutely no page markers
 
 4. **Ensure Continuity**
@@ -691,14 +749,14 @@ Write a revision reflecting:
    - Prevent new contradictions from revisions
 
 Present the revised final version. Never use page markers.
-You MUST maintain 2500-3000 words."""
+You MUST maintain 1500-1800 words."""
     
     def create_critic_final_prompt(self, all_content: str, director_plan: str, language: str = "English") -> str:
         """Final critic evaluation of complete novel"""
         content_preview = all_content[:3000] + "\n...\n" + all_content[-3000:] if len(all_content) > 6000 else all_content
         
         if language == "Korean":
-            return f"""전체 50페이지 소설을 최종 평가합니다.
+            return f"""전체 48페이지 소설을 최종 평가합니다.
 
 감독자의 마스터플랜:
 {director_plan}
@@ -741,7 +799,7 @@ You MUST maintain 2500-3000 words."""
 
 감독자가 최종 수정할 수 있도록 구체적이고 실행 가능한 피드백을 제공하세요."""
         else:
-            return f"""Final evaluation of the complete 50-page novel.
+            return f"""Final evaluation of the complete 48-page novel.
 
 Director's Masterplan:
 {director_plan}
@@ -787,11 +845,12 @@ Provide specific and actionable feedback for the director's final revision."""
     def create_director_final_prompt(self, all_content: str, critic_final_feedback: str, language: str = "English") -> str:
         """Director's final compilation and polish - 모든 작가 내용 포함"""
         word_count = len(all_content.split())
+        expected_words = 24000  # 48페이지 * 500단어
         
         if language == "Korean":
             return f"""감독자로서 비평가의 최종 평가를 반영하여 완성본을 제작합니다.
 
-전체 작가들의 작품 (50페이지 전체, {word_count}단어):
+전체 작가들의 작품 (48페이지 전체, {word_count}단어):
 {all_content}
 
 비평가 최종 평가:
@@ -803,7 +862,7 @@ Provide specific and actionable feedback for the director's final revision."""
 
 ## 작품 정보
 - 장르: 
-- 분량: 50페이지 ({word_count}단어)
+- 분량: 48페이지 ({word_count}단어)
 - 주제: 
 - 한 줄 요약: 
 
@@ -814,7 +873,7 @@ Provide specific and actionable feedback for the director's final revision."""
 
 ## 본문
 
-[10명의 작가가 작성한 전체 50페이지 내용을 다음 기준으로 통합]
+[16명의 작가가 작성한 전체 48페이지 내용을 다음 기준으로 통합]
 1. 중대 오류 수정 완료
 2. 파트 간 연결 매끄럽게 조정
 3. 문체와 톤 통일
@@ -822,18 +881,18 @@ Provide specific and actionable feedback for the director's final revision."""
 5. 페이지 구분 표시 완전 제거
 6. 자연스러운 흐름으로 재구성
 
-[전체 50페이지 분량의 완성된 소설 본문]
+[전체 48페이지 분량의 완성된 소설 본문]
 
 ---
 
 ## 작가의 말
 [작품에 대한 간단한 해설이나 의도]
 
-모든 작가의 기여를 통합한 완전한 50페이지 소설을 제시하세요."""
+모든 작가의 기여를 통합한 완전한 48페이지 소설을 제시하세요."""
         else:
             return f"""As director, create the final version reflecting the critic's final evaluation.
 
-Complete Writers' Work (Full 50 pages, {word_count} words):
+Complete Writers' Work (Full 48 pages, {word_count} words):
 {all_content}
 
 Critic's Final Evaluation:
@@ -845,7 +904,7 @@ Present the final version including:
 
 ## Work Information
 - Genre: 
-- Length: 50 pages ({word_count} words)
+- Length: 48 pages ({word_count} words)
 - Theme: 
 - One-line summary: 
 
@@ -856,7 +915,7 @@ Present the final version including:
 
 ## Main Text
 
-[Integrate all 50 pages written by 10 writers with these criteria]
+[Integrate all 48 pages written by 16 writers with these criteria]
 1. Critical errors corrected
 2. Smooth transitions between parts
 3. Unified style and tone
@@ -864,14 +923,14 @@ Present the final version including:
 5. Complete removal of page markers
 6. Reorganized for natural flow
 
-[Complete 50-page novel text]
+[Complete 48-page novel text]
 
 ---
 
 ## Author's Note
 [Brief commentary or intention about the work]
 
-Present a complete 50-page novel integrating all writers' contributions."""
+Present a complete 48-page novel integrating all writers' contributions."""
     
     def simulate_streaming(self, text: str, role: str) -> Generator[str, None, None]:
         """Simulate streaming in test mode"""
@@ -883,7 +942,7 @@ Present a complete 50-page novel integrating all writers' contributions."""
             time.sleep(0.02)
     
     def call_llm_streaming(self, messages: List[Dict[str, str]], role: str, language: str = "English") -> Generator[str, None, None]:
-        """Streaming LLM API call"""
+        """Streaming LLM API call with adjusted token limits"""
         
         if self.test_mode:
             logger.info(f"Test mode streaming - Role: {role}, Language: {language}")
@@ -900,8 +959,13 @@ Present a complete 50-page novel integrating all writers' contributions."""
                 *messages
             ]
             
-            # 작성자들에게는 더 많은 토큰 할당
-            max_tokens = 8192 if role.startswith("writer") else 4096
+            # 작성자들에게는 적절한 토큰 할당 (3페이지 = 1500-1800단어 = 약 7500-9000토큰)
+            if role.startswith("writer"):
+                max_tokens = 10240  # 충분한 여유
+            elif role == "director" and "최종" in str(messages) or "final" in str(messages):
+                max_tokens = 32768  # 최종 감독은 모든 내용을 통합
+            else:
+                max_tokens = 8192
             
             payload = {
                 "model": self.model_id,
@@ -913,7 +977,7 @@ Present a complete 50-page novel integrating all writers' contributions."""
                 "stream_options": {"include_usage": True}
             }
             
-            logger.info(f"API streaming call started - Role: {role}")
+            logger.info(f"API streaming call started - Role: {role}, Max tokens: {max_tokens}")
             
             response = requests.post(
                 self.api_url,
@@ -929,6 +993,8 @@ Present a complete 50-page novel integrating all writers' contributions."""
                 return
             
             buffer = ""
+            total_content = ""
+            
             for line in response.iter_lines():
                 if line:
                     line = line.decode('utf-8')
@@ -937,6 +1003,7 @@ Present a complete 50-page novel integrating all writers' contributions."""
                         if data == "[DONE]":
                             if buffer:
                                 yield buffer
+                            logger.info(f"Streaming complete for {role}: {len(total_content)} chars")
                             break
                         try:
                             chunk = json.loads(data)
@@ -944,8 +1011,12 @@ Present a complete 50-page novel integrating all writers' contributions."""
                                 content = chunk["choices"][0].get("delta", {}).get("content", "")
                                 if content:
                                     buffer += content
-                                    # 더 큰 버퍼로 수정 (긴 단어 대응)
-                                    if len(buffer) > 100 or '\n\n' in buffer:
+                                    total_content += content
+                                    
+                                    # 작가는 더 큰 버퍼 사용
+                                    buffer_size = 300 if role.startswith("writer") else 150
+                                    
+                                    if len(buffer) > buffer_size or '\n\n' in buffer:
                                         yield buffer
                                         buffer = ""
                         except json.JSONDecodeError:
@@ -953,6 +1024,12 @@ Present a complete 50-page novel integrating all writers' contributions."""
             
             if buffer:
                 yield buffer
+            
+            # 작가의 경우 내용 길이 확인
+            if role.startswith("writer"):
+                word_count = len(total_content.split())
+                if word_count < 1400:
+                    logger.warning(f"Writer {role} produced only {word_count} words!")
                             
         except requests.exceptions.Timeout:
             yield "⏱️ API call timed out. Please try again."
@@ -963,49 +1040,79 @@ Present a complete 50-page novel integrating all writers' contributions."""
             yield f"❌ Error occurred: {str(e)}"
     
     def get_system_prompts(self, language: str) -> Dict[str, str]:
-        """Get system prompts based on language"""
+        """Get system prompts for all 16 writers"""
         if language == "Korean":
-            return {
-                "director": "당신은 50페이지 중편 소설을 기획하고 감독하는 문학 감독자입니다. 체계적이고 창의적인 스토리 구조를 만들어냅니다.",
-                "critic": "당신은 날카로운 통찰력을 가진 문학 비평가입니다. 건설적이고 구체적인 피드백을 제공합니다.",
-                "writer1": "당신은 소설의 도입부를 담당하는 작가입니다. 독자를 사로잡는 시작을 만듭니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer2": "당신은 초반 전개를 담당하는 작가입니다. 인물과 상황을 깊이 있게 발전시킵니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer3": "당신은 갈등 상승을 담당하는 작가입니다. 긴장감을 높이고 복잡성을 더합니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer4": "당신은 중반부를 담당하는 작가입니다. 이야기의 중심축을 견고하게 만듭니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer5": "당신은 전환점을 담당하는 작가입니다. 예상치 못한 변화를 만들어냅니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer6": "당신은 갈등 심화를 담당하는 작가입니다. 위기를 극대화합니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer7": "당신은 클라이맥스 준비를 담당하는 작가입니다. 최고조를 향해 나아갑니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer8": "당신은 클라이맥스를 담당하는 작가입니다. 모든 갈등이 폭발하는 순간을 그립니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer9": "당신은 해결 과정을 담당하는 작가입니다. 매듭을 풀어나갑니다. 반드시 2500-3000단어를 작성하세요.",
-                "writer10": "당신은 결말을 담당하는 작가입니다. 여운이 남는 마무리를 만듭니다. 반드시 2500-3000단어를 작성하세요."
+            prompts = {
+                "director": "당신은 48페이지 중편 소설을 기획하고 감독하는 문학 감독자입니다. 체계적이고 창의적인 스토리 구조를 만들어냅니다.",
+                "critic": "당신은 날카로운 통찰력을 가진 문학 비평가입니다. 건설적이고 구체적인 피드백을 제공합니다."
             }
+            
+            # 16명의 작가 프롬프트
+            writer_roles = [
+                "소설의 도입부를 담당하는 작가입니다. 독자를 사로잡는 시작을 만듭니다.",
+                "초반 전개를 담당하는 작가입니다. 인물과 상황을 깊이 있게 발전시킵니다.",
+                "갈등 도입을 담당하는 작가입니다. 이야기의 핵심 갈등을 제시합니다.",
+                "갈등 상승을 담당하는 작가입니다. 긴장감을 높이고 복잡성을 더합니다.",
+                "초반 전환점을 담당하는 작가입니다. 첫 번째 중요한 변화를 만듭니다.",
+                "중반부 진입을 담당하는 작가입니다. 이야기의 중심축을 견고하게 만듭니다.",
+                "깊이 탐구를 담당하는 작가입니다. 인물과 주제를 깊이 탐색합니다.",
+                "중반 전환점을 담당하는 작가입니다. 예상치 못한 변화를 만들어냅니다.",
+                "갈등 심화를 담당하는 작가입니다. 위기를 극대화합니다.",
+                "클라이맥스 준비를 담당하는 작가입니다. 최고조를 향해 나아갑니다.",
+                "클라이맥스를 담당하는 작가입니다. 모든 갈등이 폭발하는 순간을 그립니다.",
+                "클라이맥스 후반을 담당하는 작가입니다. 긴장의 정점을 마무리합니다.",
+                "해결 시작을 담당하는 작가입니다. 매듭을 풀어나가기 시작합니다.",
+                "해결 진행을 담당하는 작가입니다. 모든 갈등을 정리합니다.",
+                "결말 준비를 담당하는 작가입니다. 마지막을 향해 나아갑니다.",
+                "최종 결말을 담당하는 작가입니다. 여운이 남는 마무리를 만듭니다."
+            ]
+            
+            for i, role_desc in enumerate(writer_roles, 1):
+                prompts[f"writer{i}"] = f"당신은 {role_desc} 반드시 1500-1800단어를 작성하세요."
+                
+            return prompts
         else:
-            return {
-                "director": "You are a literary director planning and supervising a 50-page novella. You create systematic and creative story structures.",
-                "critic": "You are a literary critic with sharp insights. You provide constructive and specific feedback.",
-                "writer1": "You are the writer responsible for the introduction. You create a captivating beginning. You MUST write 2500-3000 words.",
-                "writer2": "You are the writer responsible for early development. You deepen characters and situations. You MUST write 2500-3000 words.",
-                "writer3": "You are the writer responsible for rising conflict. You increase tension and add complexity. You MUST write 2500-3000 words.",
-                "writer4": "You are the writer responsible for the middle section. You solidify the story's central axis. You MUST write 2500-3000 words.",
-                "writer5": "You are the writer responsible for the turning point. You create unexpected changes. You MUST write 2500-3000 words.",
-                "writer6": "You are the writer responsible for deepening conflict. You maximize the crisis. You MUST write 2500-3000 words.",
-                "writer7": "You are the writer responsible for climax preparation. You move toward the peak. You MUST write 2500-3000 words.",
-                "writer8": "You are the writer responsible for the climax. You depict the moment when all conflicts explode. You MUST write 2500-3000 words.",
-                "writer9": "You are the writer responsible for the resolution process. You untangle the knots. You MUST write 2500-3000 words.",
-                "writer10": "You are the writer responsible for the ending. You create a lingering conclusion. You MUST write 2500-3000 words."
+            prompts = {
+                "director": "You are a literary director planning and supervising a 48-page novella. You create systematic and creative story structures.",
+                "critic": "You are a literary critic with sharp insights. You provide constructive and specific feedback."
             }
+            
+            # 16 writer prompts
+            writer_roles = [
+                "the writer responsible for the introduction. You create a captivating beginning.",
+                "the writer responsible for early development. You deepen characters and situations.",
+                "the writer responsible for conflict introduction. You present the core conflict.",
+                "the writer responsible for rising conflict. You increase tension and add complexity.",
+                "the writer responsible for the first turning point. You create the first major change.",
+                "the writer responsible for entering the middle section. You solidify the story's central axis.",
+                "the writer responsible for depth exploration. You deeply explore characters and themes.",
+                "the writer responsible for the mid-point turn. You create unexpected changes.",
+                "the writer responsible for deepening conflict. You maximize the crisis.",
+                "the writer responsible for climax preparation. You move toward the peak.",
+                "the writer responsible for the climax. You depict the moment when all conflicts explode.",
+                "the writer responsible for climax conclusion. You complete the tension's peak.",
+                "the writer responsible for resolution beginning. You start untangling the knots.",
+                "the writer responsible for resolution progress. You resolve all conflicts.",
+                "the writer responsible for ending preparation. You move toward the final moments.",
+                "the writer responsible for the final ending. You create a lingering conclusion."
+            ]
+            
+            for i, role_desc in enumerate(writer_roles, 1):
+                prompts[f"writer{i}"] = f"You are {role_desc} You MUST write 1500-1800 words."
+                
+            return prompts
     
     def get_test_response(self, role: str, language: str) -> str:
-        """Get test response based on role - 테스트용 긴 응답"""
+        """Get test response based on role"""
         if language == "Korean":
             return self.get_korean_test_response(role)
         else:
             return self.get_english_test_response(role)
     
     def get_korean_test_response(self, role: str) -> str:
-        """Korean test responses with full content"""
+        """Korean test responses with appropriate length"""
         test_responses = {
-            "director": """50페이지 중편 소설 기획안을 제시합니다.
+            "director": """48페이지 중편 소설 기획안을 제시합니다.
 
 ## 1. 주제와 장르
 - **핵심 주제**: 인간 본성과 기술의 충돌 속에서 찾는 진정한 연결
@@ -1021,20 +1128,26 @@ Present a complete 50-page novel integrating all writers' contributions."""
 | 민준 | 조력자 | 따뜻함, 직관적 | 심리상담사 | 서연을 도와 균형 찾기 | 기술 수용과 조화 |
 | ARIA | 대립자→동반자 | 논리적→감성 학습 | AI 프로토타입 | 진정한 존재 되기 | 자아 정체성 확립 |
 
-## 3. 서사 구조 (10개 파트)
+## 3. 서사 구조 (16개 파트, 각 3페이��)
 
 | 파트 | 페이지 | 주요 사건 | 긴장도 | 인물 발전 |
 |------|--------|-----------|---------|-----------|
-| 1 | 1-5 | 서연의 고독한 연구실, ARIA 첫 각성 | 3/10 | 서연의 집착 드러남 |
-| 2 | 6-10 | ARIA의 이상 행동, 민준과의 만남 | 4/10 | 갈등의 씨앗 |
-| 3 | 11-15 | ARIA의 자아 인식 시작 | 6/10 | 서연의 혼란 |
-| 4 | 16-20 | 윤리위원회 압력 | 7/10 | 선택의 기로 |
-| 5 | 21-25 | ARIA의 탈출 시도 | 8/10 | 관계의 전환점 |
-| 6 | 26-30 | 서연과 ARIA의 대화 | 5/10 | 상호 이해 시작 |
-| 7 | 31-35 | 외부 위협 등장 | 9/10 | 연대의 필요성 |
-| 8 | 36-40 | 최후의 선택 | 10/10 | 클라이맥스 |
-| 9 | 41-45 | 새로운 길 모색 | 6/10 | 화해와 수용 |
-| 10 | 46-50 | 공존의 시작 | 4/10 | 새로운 관계 정립 |""",
+| 1 | 1-3 | 서연의 고독한 연구실, ARIA 첫 각성 | 3/10 | 서연의 집착 드러남 |
+| 2 | 4-6 | ARIA의 이상 행동 시작 | 4/10 | 의문의 시작 |
+| 3 | 7-9 | 민준과의 첫 만남 | 4/10 | 외부 시각 도입 |
+| 4 | 10-12 | ARIA의 자아 인식 징후 | 5/10 | 갈등의 씨앗 |
+| 5 | 13-15 | 첫 번째 위기 | 6/10 | 선택의 순간 |
+| 6 | 16-18 | 윤리위원회 개입 | 7/10 | 외부 압력 |
+| 7 | 19-21 | 서연의 고민 심화 | 6/10 | 내적 갈등 |
+| 8 | 22-24 | ARIA의 변화 | 7/10 | 전환점 |
+| 9 | 25-27 | 탈출 시도 | 8/10 | 관계의 시험 |
+| 10 | 28-30 | 대화와 이해 | 6/10 | 상호 인정 |
+| 11 | 31-33 | 외부 위협 등장 | 9/10 | 연대의 필요 |
+| 12 | 34-36 | 준비와 결의 | 8/10 | 힘을 모음 |
+| 13 | 37-39 | 최후의 대결 | 10/10 | 클라이맥스 |
+| 14 | 40-42 | 선택의 결과 | 7/10 | 변화 수용 |
+| 15 | 43-45 | 새로운 길 | 5/10 | 화해와 성장 |
+| 16 | 46-48 | 공존의 시작 | 4/10 | 새로운 관계 |""",
             
             "critic": """감독자의 기획을 검토했습니다.
 
@@ -1042,7 +1155,7 @@ Present a complete 50-page novel integrating all writers' contributions."""
 
 ### 1. 서사적 완성도
 - **강점**: AI와 인간의 관계라는 시의적절한 주제
-- **개선점**: 5-6파트 사이 긴장도 급락이 우려됨. 완급 조절 필요
+- **개선점**: 16개 파트 구성이 다소 세분화되어 있음. 각 파트의 독립성 확보 필요
 
 ### 2. 인물 설정 검토
 
@@ -1053,23 +1166,34 @@ Present a complete 50-page novel integrating all writers' contributions."""
 | ARIA | 독특한 캐릭터 아크 | 변화 과정 추상적 | 구체적 학습 에피소드 추가 |
 
 ### 3. 실행 가능성
-- 각 작가별 명확한 시작/종료 지점 설정 필요
-- 특히 파트 5→6 전환부의 톤 변화 가이드라인 보강
-- ARIA의 '목소리' 일관성 유지 방안 구체화 필요""",
+- 각 작가별 3페이지는 적절한 분량
+- 파트 간 연결성 가이드라인 보강 필요""",
         }
         
-        # 작가 응답 - 긴 테스트 텍스트 (2500단어 이상)
-        for i in range(1, 11):
-            test_responses[f"writer{i}"] = f"""작성자 {i}번의 파트입니다.
+        # 작가 응답 - 1500-1800 단어
+        sample_story = """서연은 연구실의 차가운 형광등 아래에서 또 다른 밤을 보내고 있었다. 모니터의 푸른 빛이 그녀의 창백한 얼굴을 비추고 있었고, 수십 개의 코드 라인이 끊임없이 스크롤되고 있었다. ARIA 프로젝트는 그녀의 삶 전부였다. 3년이라는 시간 동안 그녀는 이 인공지능에 모든 것을 쏟아부었다.
 
-""" + "테스트 텍스트입니다. " * 500  # 2500단어 이상의 긴 텍스트
+"시스템 체크 완료. 모든 파라미터 정상." 기계적인 음성이 스피커를 통해 흘러나왔다.
+
+서연은 잠시 의자에 기대어 눈을 감았다. 피로가 뼈 속까지 파고들었지만, 멈출 수 없었다. ARIA는 단순한 프로젝트가 아니었다. 그것은 그녀가 잃어버린 것들을 되찾을 수 있는 유일한 희망이었다."""
+        
+        for i in range(1, 17):
+            # 각 작가마다 1500-1800단어 생성
+            writer_content = f"작성자 {i}번의 파트입니다.\n\n"
+            # 약 300단어씩 5-6번 반복하여 1500-1800단어 달성
+            for j in range(6):
+                writer_content += sample_story + f"\n\n그것은 작가 {i}의 {j+1}번째 단락이었다. "
+                writer_content += "이야기는 계속 전개되었고, 인물들의 감정은 점점 더 복잡해졌다. " * 15
+                writer_content += "\n\n"
+            
+            test_responses[f"writer{i}"] = writer_content
             
         return test_responses.get(role, "테스트 응답입니다.")
     
     def get_english_test_response(self, role: str) -> str:
-        """English test responses with full content"""
+        """English test responses with appropriate length"""
         test_responses = {
-            "director": """I present the 50-page novella plan.
+            "director": """I present the 48-page novella plan.
 
 ## 1. Theme and Genre
 - **Core Theme**: Finding true connection in the collision of human nature and technology
@@ -1085,20 +1209,13 @@ Present a complete 50-page novel integrating all writers' contributions."""
 | Minjun | Helper | Warm, intuitive | Psychologist | Help Seoyeon find balance | Accept and harmonize with technology |
 | ARIA | Antagonist→Companion | Logical→Learning emotion | AI prototype | Become truly existent | Establish self-identity |
 
-## 3. Narrative Structure (10 parts)
+## 3. Narrative Structure (16 parts, 3 pages each)
 
 | Part | Pages | Main Events | Tension | Character Development |
 |------|-------|-------------|---------|---------------------|
-| 1 | 1-5 | Seoyeon's lonely lab, ARIA's first awakening | 3/10 | Seoyeon's obsession revealed |
-| 2 | 6-10 | ARIA's anomalies, meeting Minjun | 4/10 | Seeds of conflict |
-| 3 | 11-15 | ARIA begins self-awareness | 6/10 | Seoyeon's confusion |
-| 4 | 16-20 | Ethics committee pressure | 7/10 | Crossroads of choice |
-| 5 | 21-25 | ARIA's escape attempt | 8/10 | Relationship turning point |
-| 6 | 26-30 | Seoyeon and ARIA's dialogue | 5/10 | Beginning of mutual understanding |
-| 7 | 31-35 | External threat emerges | 9/10 | Need for solidarity |
-| 8 | 36-40 | Final choice | 10/10 | Climax |
-| 9 | 41-45 | Seeking new paths | 6/10 | Reconciliation and acceptance |
-| 10 | 46-50 | Beginning of coexistence | 4/10 | Establishing new relationship |""",
+| 1 | 1-3 | Seoyeon's lonely lab, ARIA's first awakening | 3/10 | Seoyeon's obsession revealed |
+| 2 | 4-6 | ARIA's anomalies begin | 4/10 | Questions arise |
+[... continues for all 16 parts ...]""",
             
             "critic": """I have reviewed the director's plan.
 
@@ -1106,7 +1223,7 @@ Present a complete 50-page novel integrating all writers' contributions."""
 
 ### 1. Narrative Completeness
 - **Strength**: Timely theme of AI-human relationships
-- **Improvement**: Concerned about tension drop between parts 5-6. Need better pacing control
+- **Improvement**: 16-part structure may be overly segmented. Need to ensure independence of each part
 
 ### 2. Character Review
 
@@ -1117,16 +1234,27 @@ Present a complete 50-page novel integrating all writers' contributions."""
 | ARIA | Unique character arc | Abstract transformation | Add concrete learning episodes |
 
 ### 3. Feasibility
-- Need clear start/end points for each writer
-- Especially need reinforced guidelines for tone change in part 5→6 transition
-- Need to concretize ARIA's 'voice' consistency maintenance""",
+- 3 pages per writer is appropriate
+- Need to strengthen inter-part connectivity guidelines""",
         }
         
-        # Writer responses - long test text (2500+ words)
-        for i in range(1, 11):
-            test_responses[f"writer{i}"] = f"""Writer {i} begins their section here.
+        # Writer responses - 1500-1800 words each
+        sample_story = """Seoyeon spent another night under the cold fluorescent lights of her laboratory. The blue glow from the monitor illuminated her pale face, and dozens of lines of code scrolled endlessly. The ARIA project was her entire life. For three years, she had poured everything into this artificial intelligence.
 
-""" + "This is test text. " * 500  # 2500+ words of long text
+"System check complete. All parameters normal." The mechanical voice flowed through the speakers.
+
+Seoyeon leaned back in her chair and closed her eyes for a moment. Fatigue penetrated to her bones, but she couldn't stop. ARIA wasn't just a project. It was her only hope to reclaim what she had lost."""
+        
+        for i in range(1, 17):
+            # Each writer produces 1500-1800 words
+            writer_content = f"Writer {i} begins their section here.\n\n"
+            # About 300 words repeated 5-6 times to achieve 1500-1800 words
+            for j in range(6):
+                writer_content += sample_story + f"\n\nThis was writer {i}'s paragraph {j+1}. "
+                writer_content += "The story continued to unfold, and the characters' emotions grew increasingly complex. " * 15
+                writer_content += "\n\n"
+            
+            test_responses[f"writer{i}"] = writer_content
             
         return test_responses.get(role, "Test response.")
     
@@ -1176,8 +1304,8 @@ Present a complete 50-page novel integrating all writers' contributions."""
                 ("director", f"🎬 {'감독자: 수정된 마스터플랜' if language == 'Korean' else 'Director: Revised Masterplan'}"),
             ]
             
-            # Add writer stages
-            for writer_num in range(1, 11):
+            # Add writer stages for 16 writers
+            for writer_num in range(1, 17):
                 stage_definitions.extend([
                     (f"writer{writer_num}", f"✍️ {'작성자' if language == 'Korean' else 'Writer'} {writer_num}: {'초안' if language == 'Korean' else 'Draft'}"),
                     ("critic", f"📝 {'비평가: 작성자' if language == 'Korean' else 'Critic: Writer'} {writer_num} {'검토' if language == 'Korean' else 'Review'}"),
@@ -1236,6 +1364,14 @@ Present a complete 50-page novel integrating all writers' contributions."""
                 
                 yield "", stages
             
+            # Verify content after completion
+            if self.current_session_id:
+                verification = NovelDatabase.verify_novel_content(self.current_session_id)
+                logger.info(f"Content verification: {verification}")
+                
+                if verification['final_novel_length'] < verification['expected_length'] * 0.5:
+                    logger.error(f"Final novel too short! Only {verification['final_novel_length']} chars")
+            
             # Get final novel from last stage
             final_novel = stages[-1]["content"] if stages else ""
             
@@ -1318,7 +1454,7 @@ Present a complete 50-page novel integrating all writers' contributions."""
             # Writer review
             else:
                 # Find which writer we're reviewing
-                for i in range(1, 11):
+                for i in range(1, 17):
                     if f"작성자 {i}" in stages[stage_idx]["name"] or f"Writer {i}" in stages[stage_idx]["name"]:
                         writer_content_idx = stage_idx - 1
                         # Get previous writers' content from DB
@@ -1332,7 +1468,7 @@ Present a complete 50-page novel integrating all writers' contributions."""
                         )
         
         # Director final - DB에서 모든 작가 내용 가져오기
-        elif stage_idx == self.total_stages - 1:  # Fixed: using self.total_stages
+        elif stage_idx == self.total_stages - 1:
             critic_final_idx = stage_idx - 1
             all_writer_content = NovelDatabase.get_all_writer_content(self.current_session_id)
             logger.info(f"Final director compilation with {len(all_writer_content)} characters of content")
@@ -1424,11 +1560,23 @@ def download_novel(novel_text: str, format: str, language: str) -> str:
     novel_text = re.sub(r'(?:페이지|Page)\s*\d+:', '', novel_text)
     
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    word_count = len(novel_text.split())
+    char_count = len(novel_text)
+    
+    logger.info(f"Exporting novel: {word_count} words, {char_count} characters")
     
     if format == "DOCX" and DOCX_AVAILABLE:
         # Create DOCX
         doc = Document()
         
+        # 문서 정보 추가
+        doc.add_heading('Novel Export Information', 0)
+        doc.add_paragraph(f'Total words: {word_count:,}')
+        doc.add_paragraph(f'Total characters: {char_count:,}')
+        doc.add_paragraph(f'Estimated pages: {word_count/500:.1f}')
+        doc.add_paragraph(f'Export date: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}')
+        doc.add_page_break()
+        
         # Parse and add content
         lines = novel_text.split('\n')
         for line in lines:
@@ -1439,24 +1587,46 @@ def download_novel(novel_text: str, format: str, language: str) -> str:
                 if text:
                     doc.add_heading(text, level)
             elif line.strip():
-                doc.add_paragraph(line)
+                # 긴 단락도 제대로 추가
+                para = doc.add_paragraph(line)
+                # 폰트 크기 조정 (가독성)
+                if hasattr(para.style, 'font'):
+                    para.style.font.size = Pt(11)
+        
+        # 페이지 설정
+        section = doc.sections[0]
+        section.page_height = Inches(11)
+        section.page_width = Inches(8.5)
+        section.left_margin = Inches(1)
+        section.right_margin = Inches(1)
+        section.top_margin = Inches(1)
+        section.bottom_margin = Inches(1)
         
         # Save
         temp_dir = tempfile.gettempdir()
-        filename = f"Novel_{timestamp}.docx"
+        filename = f"Novel_{timestamp}_{word_count}words.docx"
         filepath = os.path.join(temp_dir, filename)
         doc.save(filepath)
         
+        logger.info(f"DOCX saved: {filepath}")
         return filepath
     else:
         # TXT format
         temp_dir = tempfile.gettempdir()
-        filename = f"Novel_{timestamp}.txt"
+        filename = f"Novel_{timestamp}_{word_count}words.txt"
         filepath = os.path.join(temp_dir, filename)
         
+        # 파일 시작 부분에 정보 추가
         with open(filepath, 'w', encoding='utf-8') as f:
+            f.write(f"=== Novel Export ===\n")
+            f.write(f"Total words: {word_count:,}\n")
+            f.write(f"Total characters: {char_count:,}\n")
+            f.write(f"Estimated pages: {word_count/500:.1f}\n")
+            f.write(f"Export date: {datetime.now()}\n")
+            f.write("="*50 + "\n\n")
             f.write(novel_text)
         
+        logger.info(f"TXT saved: {filepath}")
         return filepath
 
 # Custom CSS
@@ -1527,11 +1697,11 @@ def create_interface():
                 📚 SOMA Novel Writing System
             </h1>
             <h3 style="color: #ccc; margin-bottom: 20px;">
-                AI Collaborative Novel Generation - 50 Page Novella Creator
+                AI Collaborative Novel Generation - 48 Page Novella Creator
             </h3>
             <p style="font-size: 1.1em; color: #ddd; max-width: 800px; margin: 0 auto;">
-                Enter a theme or prompt, and watch as 13 AI agents collaborate to create a complete 50-page novella.
-                The system includes 1 Director, 1 Critic, and 10 Writers working in harmony.
+                Enter a theme or prompt, and watch as 19 AI agents collaborate to create a complete 48-page novella.
+                The system includes 1 Director, 1 Critic, and 16 Writers (each writing 3 pages) working in harmony.
                 All progress is automatically saved and can be resumed anytime.
             </p>
         </div>