Update app.py

app.py

@@ -2040,7 +2040,7 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
         if is_test_mode:
             # 테스트 모드: writer1 revision + writer10 내용 처리
             for stage in stages:
-                logger.debug(f"Checking stage: role={stage['role']}, name={stage['stage_name']}, status={stage['status']}")
+                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']:
@@ -2086,19 +2086,20 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                                 logger.info(f"Added Chapter {chapter_num}: {word_count} words")
         else:
             # 일반 모드: 모든 작가 수정본 처리
+            logger.info("[NORMAL MODE] Processing all writer revisions...")
+            
             for stage in stages:
-                logger.debug(f"Checking stage: role={stage['role']}, name={stage['stage_name']}, status={stage['status']}")
+                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']:
                     # 작가 번호 추출
                     try:
-                        writer_num_from_role = int(stage['role'].replace('writer', ''))
-                        logger.debug(f"Found writer {writer_num_from_role} revision")
+                        writer_num = int(stage['role'].replace('writer', ''))
+                        logger.info(f"Found writer {writer_num} revision")
                     except:
                         logger.warning(f"Could not extract writer number from role: {stage['role']}")
                         continue
-                        
-                    writer_count += 1
+                    
                     content = stage['content'] or ""
                     
                     # 페이지 마크 제거
@@ -2110,11 +2111,17 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                         word_count = stage['word_count'] or len(content.split())
                         total_words += word_count
                         writer_contents.append({
-                            'writer_num': writer_count,
+                            'writer_num': writer_num,  # ✅ 올바른 작가 번호 사용!
                             'content': content,
                             'word_count': word_count
                         })
-                        logger.info(f"Added writer {writer_count}: {word_count} words")
+                        writer_count += 1  # 실제 작가 수 카운트
+                        logger.info(f"Added writer {writer_num}: {word_count} words, content length: {len(content)}")
+                    else:
+                        logger.warning(f"Writer {writer_num} has empty content after cleaning")
+        
+        logger.info(f"Total writers collected: {writer_count}, Total words: {total_words}")
+        logger.info(f"Writer contents: {len(writer_contents)} entries")
         
         # 통계 페이지
         doc.add_heading('Novel Statistics', 1)
@@ -2157,6 +2164,9 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                 
                 if idx < len(sorted_contents) - 1:  # 마지막 챕터 후에는 페이지 구분 없음
                     doc.add_page_break()
+        else:
+            logger.warning("No writer contents found! Creating empty document.")
+            doc.add_paragraph("No content found. Please check if the novel generation completed successfully.")
         
         # 페이지 설정
         for section in doc.sections:
@@ -2178,7 +2188,7 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
         logger.info(f"DOCX saved successfully: {filepath} ({total_words} words, {writer_count} writers)")
         return filepath
     else:
-        # TXT format
+        # TXT format - 동일한 수정 적용
         temp_dir = tempfile.gettempdir()
         safe_filename = re.sub(r'[^\w\s-]', '', session['user_query'][:30]).strip()
         mode_suffix = "_TestMode" if is_test_mode else "_Complete"
@@ -2244,7 +2254,7 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                                     f.write(chapter_content)
                                     f.write("\n\n")
             else:
-                # 일반 모드
+                # 일반 모드 - 수정된 로직
                 for stage in stages:
                     if stage['role'] and stage['role'].startswith('writer') and stage['stage_name'] and 'Revision' in stage['stage_name']:
                         # 작가 번호 추출
@@ -2252,8 +2262,7 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                             writer_num = int(stage['role'].replace('writer', ''))
                         except:
                             continue
-                            
-                        writer_count += 1
+                        
                         content = stage['content'] or ""
                         # 페이지 마크 제거
                         content = re.sub(r'\[(?:페이지|Page|page)\s*\d+\]', '', content)
@@ -2263,9 +2272,10 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                         if content:
                             word_count = stage['word_count'] or len(content.split())
                             total_words += word_count
+                            writer_count += 1
                             
                             f.write(f"\n{'='*40}\n")
-                            f.write(f"CHAPTER {writer_count} (Pages {(writer_count-1)*3+1}-{writer_count*3})\n")
+                            f.write(f"CHAPTER {writer_num} (Pages {(writer_num-1)*3+1}-{writer_num*3})\n")
                             f.write(f"Word Count: {word_count:,}\n")
                             f.write(f"{'='*40}\n\n")
                             f.write(content)