Update app.py

app.py

@@ -1990,6 +1990,11 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
     stages = NovelDatabase.get_stages(session_id)
     logger.info(f"Found {len(stages)} stages in database")
     
+    # 디버깅: 모든 stage 정보 출력
+    for stage in stages:
+        logger.debug(f"Stage: role={stage.get('role')}, stage_name={stage.get('stage_name')}, "
+                    f"content_length={len(stage.get('content', ''))}, status={stage.get('status')}")
+    
     # 테스트 모드 감지 - writer10이 있으면 테스트 모드
     is_test_mode = False
     has_writer10 = any(stage['role'] == 'writer10' for stage in stages)
@@ -2043,7 +2048,7 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                 logger.info(f"[TEST MODE] Checking stage: role={stage['role']}, name={stage['stage_name']}, status={stage['status']}")
                 
                 # Writer 1 수정본
-                if stage['role'] == 'writer1' and stage['stage_name'] and 'Revision' in stage['stage_name']:
+                if stage['role'] == 'writer1' and stage['stage_name'] and ('Revision' in stage['stage_name'] or '수정본' in stage['stage_name']):
                     content = stage['content'] or ""
                     content = re.sub(r'\[(?:페이지|Page|page)\s*\d+\]', '', content)
                     content = re.sub(r'(?:페이지|Page)\s*\d+:', '', content)
@@ -2091,11 +2096,16 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
             for stage in stages:
                 logger.info(f"[NORMAL MODE] Checking stage: role={stage['role']}, name={stage['stage_name']}, status={stage['status']}")
                 
-                if stage['role'] and stage['role'].startswith('writer') and stage['stage_name'] and 'Revision' in stage['stage_name']:
+                # 언어에 상관없이 작가 수정본 찾기
+                is_revision = (stage['role'] and stage['role'].startswith('writer') and 
+                             stage['stage_name'] and 
+                             ('Revision' in stage['stage_name'] or '수정본' in stage['stage_name']))
+                
+                if is_revision:
                     # 작가 번호 추출
                     try:
                         writer_num = int(stage['role'].replace('writer', ''))
-                        logger.info(f"Found writer {writer_num} revision")
+                        logger.info(f"Found writer {writer_num} revision - stage_name: {stage['stage_name']}")
                     except:
                         logger.warning(f"Could not extract writer number from role: {stage['role']}")
                         continue
@@ -2212,7 +2222,7 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                 # 테스트 모드: writer1 + writer10 처리
                 for stage in stages:
                     # Writer 1 수정본
-                    if stage['role'] == 'writer1' and stage['stage_name'] and 'Revision' in stage['stage_name']:
+                    if stage['role'] == 'writer1' and stage['stage_name'] and ('Revision' in stage['stage_name'] or '수정본' in stage['stage_name']):
                         content = stage['content'] or ""
                         content = re.sub(r'\[(?:페이지|Page|page)\s*\d+\]', '', content)
                         content = re.sub(r'(?:페이지|Page)\s*\d+:', '', content)
@@ -2256,7 +2266,12 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
             else:
                 # 일반 모드 - 수정된 로직
                 for stage in stages:
-                    if stage['role'] and stage['role'].startswith('writer') and stage['stage_name'] and 'Revision' in stage['stage_name']:
+                    # 언어에 상관없이 작가 수정본 찾기
+                    is_revision = (stage['role'] and stage['role'].startswith('writer') and 
+                                 stage['stage_name'] and 
+                                 ('Revision' in stage['stage_name'] or '수정본' in stage['stage_name']))
+                    
+                    if is_revision:
                         # 작가 번호 추출
                         try:
                             writer_num = int(stage['role'].replace('writer', ''))