Update app.py

app.py

@@ -418,6 +418,52 @@ class NovelDatabase:
             conn.commit()
             logger.info(f"Updated final novel for session {session_id}, length: {len(final_novel)}")
     
+    @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, word_count
+                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
+            total_words = 0
+            
+            for row in cursor.fetchall():
+                results.append({
+                    'stage': row['stage_number'],
+                    'name': row['stage_name'],
+                    'length': row['content_length'] or 0,
+                    'words': row['word_count'] or 0
+                })
+                total_length += row['content_length'] or 0
+                total_words += row['word_count'] 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,
+                'total_words': total_words,
+                'final_novel_length': final_length,
+                'expected_words': 14500  # 10 작가 * 1450 평균
+            }
+    
     @staticmethod
     def get_active_sessions() -> List[Dict]:
         """Get all active sessions"""
@@ -1076,7 +1122,7 @@ Write a high-quality complete novel."""
             
             # 작성자들에게는 적절한 토큰 할당
             if role == "writer10" and stage_info and stage_info.get('test_mode'):
-                max_tokens = 20000  # 테스트 모드: 전체 소설 작성
+                max_tokens = 30000  # 테스트 모드: 전체 소설 작성 (충분한 토큰)
                 temperature = 0.8   
                 top_p = 0.95
             elif role.startswith("writer"):
@@ -1211,11 +1257,12 @@ Write a high-quality complete novel."""
             yield f"❌ Error occurred: {str(e)}"
     
     def get_system_prompts(self, language: str) -> Dict[str, str]:
-        """Get system prompts for all 10 writers"""
+        """Get system prompts for all writers including test mode"""
         if language == "Korean":
             prompts = {
                 "director": "당신은 30페이지 중편 소설을 기획하고 감독하는 문학 감독자입니다. 체계적이고 창의적인 스토리 구조를 만들어냅니다.",
-                "critic": "당신은 날카로운 통찰력을 가진 문학 비평가입니다. 건설적이고 구체적인 피드백을 제공합니다."
+                "critic": "당신은 날카로운 통찰력을 가진 문학 비평가입니다. 건설적이고 구체적인 피드백을 제공합니다.",
+                "writer10": "[테스트 모드] 당신은 전체 30페이지 소설을 한 번에 작성하는 특별 작가입니다. 10개 챕터로 구성된 완전한 소설을 작성하세요."
             }
             
             # 10명의 작가 프롬프트
@@ -1239,7 +1286,8 @@ Write a high-quality complete novel."""
         else:
             prompts = {
                 "director": "You are a literary director planning and supervising a 30-page novella. You create systematic and creative story structures.",
-                "critic": "You are a literary critic with sharp insights. You provide constructive and specific feedback."
+                "critic": "You are a literary critic with sharp insights. You provide constructive and specific feedback.",
+                "writer10": "[TEST MODE] You are a special writer creating the complete 30-page novel at once. Write a complete novel organized into 10 chapters."
             }
             
             # 10 writer prompts
@@ -1550,8 +1598,31 @@ Seoyeon leaned back in her chair and closed her eyes for a moment. Fatigue penet
                 # 테스트 모드에서 writer10 저장 시 특별 처리
                 if test_quick_mode and role == "writer10":
                     # 전체 내용을 10개 챕터로 분할하여 저장
-                    chapters = re.split(r'\[Chapter \d+\]', stage_content)
-                    chapters = [ch.strip() for ch in chapters if ch.strip()]
+                    # [Chapter 숫자] 패턴으로 분할
+                    chapter_pattern = r'\[Chapter\s+\d+\]'
+                    chapters_with_headers = re.split(f'({chapter_pattern})', stage_content)
+                    
+                    # 챕터 내용만 추출 (헤더와 내용을 쌍으로)
+                    chapters = []
+                    for i in range(1, len(chapters_with_headers), 2):
+                        if i+1 < len(chapters_with_headers):
+                            chapter_content = chapters_with_headers[i+1].strip()
+                            if chapter_content:
+                                chapters.append(chapter_content)
+                    
+                    # 챕터가 10개가 안 되면 균등 분할
+                    if len(chapters) < 10:
+                        logger.warning(f"Only {len(chapters)} chapters found, splitting evenly")
+                        # 전체 텍스트를 10개로 균등 분할
+                        total_text = stage_content.replace('[Chapter', '\n[Chapter')
+                        words = total_text.split()
+                        words_per_chapter = len(words) // 10
+                        chapters = []
+                        for i in range(10):
+                            start_idx = i * words_per_chapter
+                            end_idx = (i + 1) * words_per_chapter if i < 9 else len(words)
+                            chapter_words = words[start_idx:end_idx]
+                            chapters.append(' '.join(chapter_words))
                     
                     # 각 챕터를 개별 writer로 저장
                     for i, chapter_content in enumerate(chapters[:10], 1):
@@ -1564,7 +1635,8 @@ Seoyeon leaned back in her chair and closed her eyes for a moment. Fatigue penet
                             chapter_content,
                             "complete"
                         )
-                        logger.info(f"✅ Test mode: Saved chapter {i} as writer {i}")
+                        word_count = len(chapter_content.split())
+                        logger.info(f"✅ Test mode: Saved chapter {i} as writer {i} ({word_count} words)")
                 else:
                     NovelDatabase.save_stage(
                         self.current_session_id,
@@ -1585,9 +1657,12 @@ Seoyeon leaned back in her chair and closed her eyes for a moment. Fatigue penet
             # 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 test_quick_mode:
+                    logger.info(f"[TEST MODE] Content verification: {verification}")
+                else:
+                    logger.info(f"Content verification: {verification}")
                 
-                if verification['total_words'] < 12000:
+                if verification['total_words'] < 12000 and not test_quick_mode:
                     logger.error(f"Final novel too short! Only {verification['total_words']} words")
             
             # Get complete novel from DB
@@ -1737,30 +1812,30 @@ Write a high-quality complete novel."""
         return ""
 
 # Gradio Interface Functions
-def process_query(query: str, language: str, session_id: str = None, test_mode: bool = False) -> Generator[Tuple[str, str, str, str], None, None]:
-    """Process query and yield updates with recovery status"""
+def process_query(query: str, language: str, session_id: str = None, test_mode: bool = False) -> Generator[Tuple[str, str, str], None, None]:
+    """Process query and yield updates"""
     if not query.strip() and not session_id:
         if language == "Korean":
-            yield "", "", "❌ 소설 주제를 입력해주세요.", ""
+            yield "", "", "❌ 소설 주제를 입력해주세요."
         else:
-            yield "", "", "❌ Please enter a novel theme.", ""
+            yield "", "", "❌ Please enter a novel theme."
         return
     
     system = NovelWritingSystem()
     
     try:
-        # Recovery status message
-        recovery_status = ""
+        # Status message with mode indicator
+        mode_status = ""
         if session_id:
             if language == "Korean":
-                recovery_status = "♻️ 이전 세션을 복구하여 계속 진행합니다..."
+                mode_status = " [♻️ 복구 모드]"
             else:
-                recovery_status = "♻️ Recovering previous session and continuing..."
+                mode_status = " [♻️ Recovery Mode]"
         elif test_mode:
             if language == "Korean":
-                recovery_status = "🧪 테스트 모드: 빠른 생성 (1,2단계 + Writer 10만)"
+                mode_status = " [🧪 테스트 모드]"
             else:
-                recovery_status = "🧪 Test Mode: Quick generation (Stages 1,2 + Writer 10 only)"
+                mode_status = " [🧪 Test Mode]"
         
         for final_novel, stages in system.process_novel_stream(query, language, session_id, test_quick_mode=test_mode):
             # Format stages for display
@@ -1771,22 +1846,22 @@ def process_query(query: str, language: str, session_id: str = None, test_mode:
             total = len(stages)
             progress_percent = (completed / total * 100) if total > 0 else 0
             
-            if "✅ Novel complete!" in str(final_novel):
-                status = "✅ Complete! Ready to download."
+            if "✅ Novel complete!" in str(final_novel) or "✅ [TEST MODE] Novel complete!" in str(final_novel):
+                status = f"✅ Complete! Ready to download.{mode_status}"
             else:
                 if language == "Korean":
-                    status = f"🔄 진행중... ({completed}/{total} - {progress_percent:.1f}%)"
+                    status = f"🔄 진행중... ({completed}/{total} - {progress_percent:.1f}%){mode_status}"
                 else:
-                    status = f"🔄 Processing... ({completed}/{total} - {progress_percent:.1f}%)"
+                    status = f"🔄 Processing... ({completed}/{total} - {progress_percent:.1f}%){mode_status}"
             
-            yield stages_display, final_novel, status, recovery_status
+            yield stages_display, final_novel, status
                     
     except Exception as e:
         logger.error(f"Error in process_query: {str(e)}", exc_info=True)
         if language == "Korean":
-            yield "", "", f"❌ 오류 발생: {str(e)}", ""
+            yield "", "", f"❌ 오류 발생: {str(e)}"
         else:
-            yield "", "", f"❌ Error occurred: {str(e)}", ""
+            yield "", "", f"❌ Error occurred: {str(e)}"
 
 def format_stages_display(stages: List[Dict[str, str]], language: str) -> str:
     """Format stages into simple display with writer save status"""