Update app.py

app.py

@@ -1945,14 +1945,17 @@ 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")
     
-    # 테스트 모드 감지 - Writer 2까지만 있고 전체 stage가 9개면 테스트 모드
+    # 테스트 모드 감지 - writer10이 있으면 테스트 모드
     is_test_mode = False
-    has_writer2 = any(stage['role'] == 'writer2' for stage in stages)
+    has_writer10 = any(stage['role'] == 'writer10' for stage in stages)
+    has_writer1 = any(stage['role'] == 'writer1' for stage in stages)
     has_writer3 = any(stage['role'] == 'writer3' for stage in stages)
-    if has_writer2 and not has_writer3 and len(stages) <= 10:
+    
+    if has_writer10 and has_writer1 and not has_writer3:
         is_test_mode = True
-        logger.info("Test mode detected - will export 2 chapters only")
+        logger.info("Test mode detected - writer1 + writer10 mode")
     
+
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     
     if format == "DOCX" and DOCX_AVAILABLE: