Update app-backup.py

app-backup.py

@@ -1976,6 +1976,40 @@ def auto_recover_session(language: str) -> Tuple[str, str]:
 
 def download_novel(novel_text: str, format: str, language: str, session_id: str = None) -> str:
     """Download novel - DB에서 직접 작가 내용을 가져와서 통합"""
+    
+    def extract_chapter_title(content: str, chapter_num: int, language: str) -> str:
+        """챕터 내용에서 제목 추출"""
+        # 내용을 줄 단위로 분리
+        lines = content.strip().split('\n')
+        
+        # 첫 번째 의미있는 문장 찾기 (빈 줄 제외)
+        for line in lines[:5]:  # 처음 5줄 내에서 찾기
+            line = line.strip()
+            if len(line) > 10:  # 10자 이상의 의미있는 문장
+                # 대화문이면 스킵 (따옴표로 시작하는 경우)
+                if line.startswith('"') or line.startswith("'") or line.startswith('"') or line.startswith("'"):
+                    continue
+                    
+                # 마침표, 느낌표, 물음표로 끝나는 첫 문장만 추출
+                import re
+                match = re.match(r'^[^.!?]+[.!?]', line)
+                if match:
+                    title = match.group(0).strip()
+                else:
+                    title = line
+                    
+                # 너무 길면 자르기
+                if len(title) > 50:
+                    title = title[:47] + "..."
+                    
+                return title
+        
+        # 적절한 제목을 찾지 못한 경우 기본 제목
+        if language == "Korean":
+            return f"제{chapter_num}장"
+        else:
+            return f"Part {chapter_num}"
+    
     if not session_id:
         logger.error("No session_id provided for download")
         return None
@@ -2022,8 +2056,12 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
         
         # 제목 페이지
         title_para = doc.add_paragraph()
-        title_run = title_para.add_run('AI Collaborative Novel' + (' - Test Mode' if is_test_mode else ''))
-        title_run.font.size = Pt(24)
+        if actual_language == 'Korean':
+            main_title = 'AI 협업 소설' + (' - 테스트 모드' if is_test_mode else '')
+        else:
+            main_title = 'AI Collaborative Novel' + (' - Test Mode' if is_test_mode else '')
+        title_run = title_para.add_run(main_title)
+        title_run.font.size = Pt(20)  # 신국판에 맞게 크기 조정
         title_run.font.bold = True
         title_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
         
@@ -2031,15 +2069,17 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
         doc.add_paragraph()
         
         theme_para = doc.add_paragraph()
-        theme_run = theme_para.add_run(f'Theme: {session["user_query"]}')
-        theme_run.font.size = Pt(14)
+        theme_label = '주제: ' if actual_language == 'Korean' else 'Theme: '
+        theme_run = theme_para.add_run(f'{theme_label}{session["user_query"]}')
+        theme_run.font.size = Pt(12)  # 크기 조정
         theme_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
         
         doc.add_paragraph()
         
         date_para = doc.add_paragraph()
-        date_run = date_para.add_run(f'Created: {datetime.now().strftime("%Y-%m-%d")}')
-        date_run.font.size = Pt(12)
+        date_label = '생성일: ' if actual_language == 'Korean' else 'Created: '
+        date_run = date_para.add_run(f'{date_label}{datetime.now().strftime("%Y-%m-%d")}')
+        date_run.font.size = Pt(10)  # 크기 조정
         date_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
         
         doc.add_page_break()
@@ -2148,23 +2188,42 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
         logger.info(f"Writer contents: {len(writer_contents)} entries")
         
         # 통계 페이지
-        doc.add_heading('Novel Statistics', 1)
-        doc.add_paragraph(f'Mode: {"Test Mode (Actual chapters written)" if is_test_mode else "Full Mode (10 chapters)"}')
-        doc.add_paragraph(f'Total Chapters: {writer_count}')
-        doc.add_paragraph(f'Total Words: {total_words:,}')
-        doc.add_paragraph(f'Estimated Pages: ~{total_words/500:.0f}')
-        doc.add_paragraph(f'Language: {actual_language}')
+        stats_heading = '소설 통계' if actual_language == 'Korean' else 'Novel Statistics'
+        doc.add_heading(stats_heading, 1)
+        if actual_language == 'Korean':
+            doc.add_paragraph(f'모드: {"테스트 모드 (실제 작성된 챕터)" if is_test_mode else "전체 모드 (10개 챕터)"}')
+            doc.add_paragraph(f'총 챕터 수: {writer_count}')
+            doc.add_paragraph(f'총 단어 수: {total_words:,}')
+            doc.add_paragraph(f'예상 페이지 수: 약 {total_words/300:.0f}')  # 한국어는 페이지당 300단어 기준
+            doc.add_paragraph(f'언어: 한국어')
+        else:
+            doc.add_paragraph(f'Mode: {"Test Mode (Actual chapters written)" if is_test_mode else "Full Mode (10 chapters)"}')
+            doc.add_paragraph(f'Total Chapters: {writer_count}')
+            doc.add_paragraph(f'Total Words: {total_words:,}')
+            doc.add_paragraph(f'Estimated Pages: ~{total_words/250:.0f}')  # 영어는 페이지당 250단어 기준
+            doc.add_paragraph(f'Language: English')
         doc.add_page_break()
         
         # 목차
         if writer_contents:
-            doc.add_heading('Table of Contents', 1)
+            toc_heading = '목차' if actual_language == 'Korean' else 'Table of Contents'
+            doc.add_heading(toc_heading, 1)
             # writer_contents를 writer_num으로 정렬
             sorted_contents = sorted(writer_contents, key=lambda x: x['writer_num'])
             
             for item in sorted_contents:
                 chapter_num = item['writer_num']
-                doc.add_paragraph(f'Chapter {chapter_num}: Pages {(chapter_num-1)*3+1}-{chapter_num*3}')
+                content = item['content']
+                
+                # 챕터 제목 추출 - 첫 번째 의미있는 문장 또는 단락
+                chapter_title = extract_chapter_title(content, chapter_num, actual_language)
+                
+                # 목차 항목 추가 (페이지 번호 없이)
+                if actual_language == 'Korean':
+                    toc_entry = f"제{chapter_num}장: {chapter_title}"
+                else:
+                    toc_entry = f"Chapter {chapter_num}: {chapter_title}"
+                doc.add_paragraph(toc_entry)
             doc.add_page_break()
             
             # 각 작가의 내용 추가 (정렬된 순서대로)
@@ -2174,9 +2233,18 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                 word_count = writer_data['word_count']
                 
                 # 챕터 헤더
-                doc.add_heading(f'Chapter {writer_num}', 1)
-                doc.add_paragraph(f'Pages {(writer_num-1)*3+1}-{writer_num*3}')
-                doc.add_paragraph(f'Word Count: {word_count:,}')
+                chapter_header = f'제{writer_num}장' if actual_language == 'Korean' else f'Chapter {writer_num}'
+                doc.add_heading(chapter_header, 1)
+                
+                # 챕터 제목 추가
+                chapter_title = extract_chapter_title(content, writer_num, actual_language)
+                title_para = doc.add_paragraph(chapter_title)
+                title_para.style.font.size = Pt(14)
+                title_para.style.font.bold = True
+                title_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
+                
+                word_count_label = f'단어 수: {word_count:,}' if actual_language == 'Korean' else f'Word Count: {word_count:,}'
+                doc.add_paragraph(word_count_label)
                 doc.add_paragraph()
                 
                 # 작가 내용 추가
@@ -2184,22 +2252,25 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                 for para_text in paragraphs:
                     if para_text.strip():
                         para = doc.add_paragraph(para_text.strip())
-                        para.style.font.size = Pt(11)
+                        para.style.font.size = Pt(10.5)  # 한국 소설 표준 크기
                 
                 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.")
+            if actual_language == 'Korean':
+                doc.add_paragraph("내용을 찾을 수 없습니다. 소설 생성이 정상적으로 완료되었는지 확인해주세요.")
+            else:
+                doc.add_paragraph("No content found. Please check if the novel generation completed successfully.")
         
-        # 페이지 설정
+        # 페이지 설정 - 한국 신국판 (152 × 225mm)
         for section in doc.sections:
-            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)
+            section.page_width = Inches(5.98)   # 152mm
+            section.page_height = Inches(8.86)  # 225mm
+            section.left_margin = Inches(0.79)  # 20mm
+            section.right_margin = Inches(0.79) # 20mm
+            section.top_margin = Inches(0.79)   # 20mm
+            section.bottom_margin = Inches(0.79) # 20mm
         
         # Save
         temp_dir = tempfile.gettempdir()
@@ -2221,12 +2292,22 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
         
         with open(filepath, 'w', encoding='utf-8') as f:
             f.write("="*60 + "\n")
-            f.write(f"AI COLLABORATIVE NOVEL - {'TEST MODE' if is_test_mode else 'COMPLETE VERSION'}\n")
+            if actual_language == 'Korean':
+                f.write(f"AI 협업 소설 - {'테스트 모드' if is_test_mode else '완성본'}\n")
+            else:
+                f.write(f"AI COLLABORATIVE NOVEL - {'TEST MODE' if is_test_mode else 'COMPLETE VERSION'}\n")
             f.write("="*60 + "\n")
-            f.write(f"Theme: {session['user_query']}\n")
-            f.write(f"Language: {actual_language}\n")
-            f.write(f"Created: {datetime.now()}\n")
-            f.write(f"Mode: {'Test Mode (Actual chapters)' if is_test_mode else 'Full Mode (10 chapters)'}\n")
+            
+            if actual_language == 'Korean':
+                f.write(f"주제: {session['user_query']}\n")
+                f.write(f"언어: 한국어\n")
+                f.write(f"생성일: {datetime.now()}\n")
+                f.write(f"모드: {'테스트 모드 (실제 작성된 챕터)' if is_test_mode else '전체 모드 (10개 챕터)'}\n")
+            else:
+                f.write(f"Theme: {session['user_query']}\n")
+                f.write(f"Language: English\n")
+                f.write(f"Created: {datetime.now()}\n")
+                f.write(f"Mode: {'Test Mode (Actual chapters)' if is_test_mode else 'Full Mode (10 chapters)'}\n")
             f.write("="*60 + "\n\n")
             
             total_words = 0
@@ -2251,8 +2332,13 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                             writer_count = 1
                             
                             f.write(f"\n{'='*40}\n")
-                            f.write(f"CHAPTER 1 (Pages 1-3)\n")
-                            f.write(f"Word Count: {word_count:,}\n")
+                            chapter_label = f"제1장" if actual_language == 'Korean' else "CHAPTER 1"
+                            f.write(f"{chapter_label}\n")
+                            # 챕터 제목 추출 및 출력
+                            chapter_title = extract_chapter_title(content, 1, actual_language)
+                            f.write(f"{chapter_title}\n")
+                            word_count_label = f"단어 수: {word_count:,}" if actual_language == 'Korean' else f"Word Count: {word_count:,}"
+                            f.write(f"{word_count_label}\n")
                             f.write(f"{'='*40}\n\n")
                             f.write(content)
                             f.write("\n\n")
@@ -2273,8 +2359,13 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                                     writer_count = max(writer_count, chapter_num)
                                     
                                     f.write(f"\n{'='*40}\n")
-                                    f.write(f"CHAPTER {chapter_num} (Pages {(chapter_num-1)*3+1}-{chapter_num*3})\n")
-                                    f.write(f"Word Count: {word_count:,}\n")
+                                    chapter_label = f"제{chapter_num}장" if actual_language == 'Korean' else f"CHAPTER {chapter_num}"
+                                    f.write(f"{chapter_label}\n")
+                                    # 챕터 제목 추출 및 출력
+                                    chapter_title = extract_chapter_title(chapter_content, chapter_num, actual_language)
+                                    f.write(f"{chapter_title}\n")
+                                    word_count_label = f"단어 수: {word_count:,}" if actual_language == 'Korean' else f"Word Count: {word_count:,}"
+                                    f.write(f"{word_count_label}\n")
                                     f.write(f"{'='*40}\n\n")
                                     f.write(chapter_content)
                                     f.write("\n\n")
@@ -2307,14 +2398,22 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
                             writer_count += 1
                             
                             f.write(f"\n{'='*40}\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")
+                            chapter_label = f"제{writer_num}장" if actual_language == 'Korean' else f"CHAPTER {writer_num}"
+                            f.write(f"{chapter_label}\n")
+                            # 챕터 제목 추출 및 출력
+                            chapter_title = extract_chapter_title(content, writer_num, actual_language)
+                            f.write(f"{chapter_title}\n")
+                            word_count_label = f"단어 수: {word_count:,}" if actual_language == 'Korean' else f"Word Count: {word_count:,}"
+                            f.write(f"{word_count_label}\n")
                             f.write(f"{'='*40}\n\n")
                             f.write(content)
                             f.write("\n\n")
             
             f.write(f"\n{'='*60}\n")
-            f.write(f"TOTAL: {writer_count} chapters, {total_words:,} words\n")
+            if actual_language == 'Korean':
+                f.write(f"총계: {writer_count}개 챕터, {total_words:,} 단어\n")
+            else:
+                f.write(f"TOTAL: {writer_count} chapters, {total_words:,} words\n")
             f.write(f"{'='*60}\n")
         
         logger.info(f"TXT saved successfully: {filepath} ({total_words} words)")