Update app.py

app.py

@@ -793,10 +793,11 @@ Provide specific, practical improvements."""
                 sentences = actual_content.split('.')
                 hook = sentences[-2] + '.' if len(sentences) > 1 else sentences[-1]
                 
-                # Save episode with title
+                # Save episode with full content including title
+                full_episode_content = f"{episode_title}\n\n{actual_content}"
                 WebNovelDatabase.save_episode(
                     self.current_session_id, episode_num, 
-                    actual_content, hook
+                    full_episode_content, hook
                 )
                 
                 # Add to accumulated content with title
@@ -836,24 +837,14 @@ def export_to_txt(episodes: List[Dict], genre: str, title: str = "") -> str:
         ep_num = ep.get('episode_number', 0)
         ep_content = ep.get('content', '')
         
-        # Extract title if exists in content
-        lines = ep_content.strip().split('\n')
-        if lines and (f"{ep_num}화." in lines[0] or f"Episode {ep_num}." in lines[0]):
-            content += f"\n{lines[0]}\n"
-            content += f"{'-' * 40}\n\n"
-            actual_content = '\n'.join(lines[2:] if len(lines) > 2 and lines[1].strip() == "" else lines[1:])
-            content += actual_content
-        else:
-            content += f"\n{ep_num}화\n"
-            content += f"{'-' * 40}\n\n"
-            content += ep_content
-        
-        content += f"\n\n{'=' * 50}\n"
+        # Content already includes title, so just add it
+        content += f"\n{ep_content}\n"
+        content += f"\n{'=' * 50}\n"
     
     return content
 
 def export_to_docx(episodes: List[Dict], genre: str, title: str = "") -> bytes:
-    """Export web novel to DOCX format"""
+    """Export web novel to DOCX format - matches screen display exactly"""
     if not DOCX_AVAILABLE:
         raise Exception("python-docx is not installed")
     
@@ -868,29 +859,37 @@ def export_to_docx(episodes: List[Dict], genre: str, title: str = "") -> bytes:
     doc.add_page_break()
     
     # Episodes
-    for ep in episodes:
+    for idx, ep in enumerate(episodes):
         ep_num = ep.get('episode_number', 0)
         ep_content = ep.get('content', '')
         
-        # Extract title if exists
+        # Split content into lines
         lines = ep_content.strip().split('\n')
-        if lines and (f"{ep_num}화." in lines[0] or f"Episode {ep_num}." in lines[0]):
-            doc.add_heading(lines[0], 1)
-            actual_content = '\n'.join(lines[2:] if len(lines) > 2 and lines[1].strip() == "" else lines[1:])
-        else:
-            doc.add_heading(f"{ep_num}화", 1)
-            actual_content = ep_content
         
-        # Add content paragraphs
-        for paragraph in actual_content.split('\n'):
-            if paragraph.strip():
-                doc.add_paragraph(paragraph.strip())
+        # First line should be the title (e.g., "1화. 제목")
+        if lines:
+            # Add episode title as heading
+            doc.add_heading(lines[0], 1)
+            
+            # Add the rest of the content
+            content_lines = lines[1:] if len(lines) > 1 else []
+            
+            # Skip empty lines at the beginning
+            while content_lines and not content_lines[0].strip():
+                content_lines.pop(0)
+            
+            # Add content paragraphs
+            for line in content_lines:
+                if line.strip():  # Only add non-empty lines
+                    doc.add_paragraph(line.strip())
+                elif len(doc.paragraphs) > 0:  # Add spacing between paragraphs
+                    doc.add_paragraph()
         
-        if ep_num < len(episodes):
+        # Add page break except for the last episode
+        if idx < len(episodes) - 1:
             doc.add_page_break()
     
     # Save to bytes
-    import io
     bytes_io = io.BytesIO()
     doc.save(bytes_io)
     bytes_io.seek(0)
@@ -1428,24 +1427,20 @@ def format_webnovel_display(episodes: List[Dict], genre: str) -> str:
     formatted += "---\n\n"
     
     # Episodes
-    for ep in episodes:
+    for idx, ep in enumerate(episodes):
         ep_num = ep.get('episode_number', 0)
         content = ep.get('content', '')
         
-        # Check if content already has episode title
-        lines = content.strip().split('\n')
-        if lines and (f"{ep_num}화." in lines[0] or f"Episode {ep_num}." in lines[0]):
-            # Use the existing title
-            formatted += f"## {lines[0]}\n\n"
-            # Use the rest as content
+        # Content already includes the title, so display as is
+        formatted += f"## {content.split(chr(10))[0] if content else f'{ep_num}화'}\n\n"
+        
+        # Get the actual content (skip title and empty line)
+        lines = content.split('\n')
+        if len(lines) > 1:
             actual_content = '\n'.join(lines[2:] if len(lines) > 2 and lines[1].strip() == "" else lines[1:])
             formatted += f"{actual_content}\n\n"
-        else:
-            # No title found, use default
-            formatted += f"## 제{ep_num}화\n\n"
-            formatted += f"{content}\n\n"
         
-        if ep_num < len(episodes):  # Not last episode
+        if idx < len(episodes) - 1:  # Not last episode
             formatted += "➡️ *다음 화에 계속...*\n\n"
         
         formatted += "---\n\n"