Update app-BACKUP.py

app-BACKUP.py

@@ -35,6 +35,8 @@ except ImportError:
     DOCX_AVAILABLE = False
     logger.warning("python-docx not installed. DOCX export will be disabled.")
 
+import io  # Add io import for DOCX export
+
 # --- Environment variables and constants ---
 FRIENDLI_TOKEN = os.getenv("FRIENDLI_TOKEN", "")
 BRAVE_SEARCH_API_KEY = os.getenv("BRAVE_SEARCH_API_KEY", "")
@@ -468,6 +470,12 @@ Provide detailed 40-episode plotline with key events for each episode."""
 **이전 내용 요약:**
 {previous_content[-1500:] if previous_content else "첫 화입니다"}
 
+**작성 형식:**
+반드시 다음 형식으로 시작하세요:
+{episode_num}화. [이번 화의 핵심을 담은 매력적인 소제목]
+
+(한 줄 띄우고 본문 시작)
+
 **작성 지침:**
 1. **구성**: 3-4개의 주요 장면으로 구성
    - 도입부: 이전 화 연결 및 현재 상황
@@ -493,6 +501,7 @@ Provide detailed 40-episode plotline with key events for each episode."""
 **참고 후크 예시:**
 {random.choice(hooks)}
 
+소제목은 이번 화의 핵심 사건이나 전환점을 암시하는 매력적인 문구로 작성하세요.
 {episode_num}화를 풍성하고 몰입감 있게 작성하세요. 반드시 400-600단어로 작성하세요.""",
 
             "English": f"""Write episode {episode_num} of the web novel.
@@ -506,6 +515,12 @@ Provide detailed 40-episode plotline with key events for each episode."""
 **Previous content:**
 {previous_content[-1500:] if previous_content else "First episode"}
 
+**Format:**
+Must start with:
+Episode {episode_num}. [Attractive subtitle that captures the essence of this episode]
+
+(blank line then start main text)
+
 **Guidelines:**
 1. **Structure**: 3-4 major scenes
    - Opening: Connect from previous, current situation
@@ -531,6 +546,7 @@ Provide detailed 40-episode plotline with key events for each episode."""
 **Hook example:**
 {random.choice(hooks)}
 
+Create an attractive subtitle that hints at key events or turning points.
 Write rich, immersive episode {episode_num}. Must be 400-600 words."""
         }
         
@@ -759,17 +775,32 @@ Provide specific, practical improvements."""
                     "writer", language
                 )
                 
+                # Extract episode title and content
+                lines = episode_content.strip().split('\n')
+                episode_title = ""
+                actual_content = episode_content
+                
+                # Check if first line contains episode number and title
+                if lines and (f"{episode_num}화." in lines[0] or f"Episode {episode_num}." in lines[0]):
+                    episode_title = lines[0]
+                    # Join the rest as content (excluding the title line and empty line after it)
+                    actual_content = '\n'.join(lines[2:] if len(lines) > 2 and lines[1].strip() == "" else lines[1:])
+                else:
+                    # If no title format found, generate a default title
+                    episode_title = f"{episode_num}화. 제{episode_num}화"
+                
                 # Extract hook (last sentence)
-                sentences = episode_content.split('.')
+                sentences = actual_content.split('.')
                 hook = sentences[-2] + '.' if len(sentences) > 1 else sentences[-1]
                 
-                # Save episode
+                # Save episode with title
                 WebNovelDatabase.save_episode(
                     self.current_session_id, episode_num, 
-                    episode_content, hook
+                    actual_content, hook
                 )
                 
-                accumulated_content += f"\n\n### {episode_num}화\n{episode_content}"
+                # Add to accumulated content with title
+                accumulated_content += f"\n\n### {episode_title}\n{actual_content}"
                 
                 # Quick critique every 5 episodes
                 if episode_num % 5 == 0:
@@ -791,7 +822,79 @@ Provide specific, practical improvements."""
             logger.error(f"Web novel generation error: {e}", exc_info=True)
             yield f"❌ 오류 발생: {e}", accumulated_content if 'accumulated_content' in locals() else "", "오류", self.current_session_id
 
-# --- Random theme generator ---
+# --- Export functions ---
+def export_to_txt(episodes: List[Dict], genre: str, title: str = "") -> str:
+    """Export web novel to TXT format"""
+    content = f"{'=' * 50}\n"
+    content += f"{title if title else genre + ' 웹소설'}\n"
+    content += f"{'=' * 50}\n\n"
+    content += f"총 {len(episodes)}화 완결\n"
+    content += f"총 단어 수: {sum(ep.get('word_count', 0) for ep in episodes):,}\n"
+    content += f"{'=' * 50}\n\n"
+    
+    for ep in episodes:
+        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"
+    
+    return content
+
+def export_to_docx(episodes: List[Dict], genre: str, title: str = "") -> bytes:
+    """Export web novel to DOCX format"""
+    if not DOCX_AVAILABLE:
+        raise Exception("python-docx is not installed")
+    
+    doc = Document()
+    
+    # Title
+    doc.add_heading(title if title else f"{genre} 웹소설", 0)
+    
+    # Stats
+    doc.add_paragraph(f"총 {len(episodes)}화 완결")
+    doc.add_paragraph(f"총 단어 수: {sum(ep.get('word_count', 0) for ep in episodes):,}")
+    doc.add_page_break()
+    
+    # Episodes
+    for ep in episodes:
+        ep_num = ep.get('episode_number', 0)
+        ep_content = ep.get('content', '')
+        
+        # Extract title if exists
+        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())
+        
+        if ep_num < len(episodes):
+            doc.add_page_break()
+    
+    # Save to bytes
+    import io
+    bytes_io = io.BytesIO()
+    doc.save(bytes_io)
+    bytes_io.seek(0)
+    return bytes_io.getvalue()
 def generate_random_webnovel_theme(genre: str, language: str) -> str:
     """Generate random web novel theme using novel_themes.json and LLM"""
     try:
@@ -896,7 +999,7 @@ def generate_random_webnovel_theme(genre: str, language: str) -> str:
         
         # Create prompt for LLM
         if language == "Korean":
-            prompt = f"""다음 요소들을 활용하여 {genre} 장르의 매력적인 웹소설 테마를 생성하세요:
+            prompt = f"""다음 요소들을 활용하여 {genre} 장르의 매력적인 웹소설을 기획하세요:
 
 【선택된 요소들】
 - 핵심 장르: {selected_genre_key}
@@ -914,19 +1017,24 @@ def generate_random_webnovel_theme(genre: str, language: str) -> str:
 【독자를 사로잡을 질문들】
 {chr(10).join(reader_questions[:2]) if reader_questions else ""}
 
-요구사항:
-1. 위 요소들을 자연스럽게 융합하여 하나의 매력적인 스토리로 만드세요
-2. 한국 웹소설 독자들이 좋아할 만한 설정을 추가하세요
-3. 제목이 떠오를 정도로 구체적이면서도 흥미로운 설정을 만드세요
-4. 200-300자 내외로 간결하게 작성하세요
+다음 형식으로 정확히 작성하세요:
+
+📖 **제목:**
+[매력적이고 기억에 남는 제목]
+
+🌍 **설정:**
+[세계관과 배경 설정을 3-4줄로 설명]
+
+👥 **주요 캐릭터:**
+• 주인공: [이름] - [간단한 설명]
+• 주요인물1: [이름] - [간단한 설명]
+• 주요인물2: [이름] - [간단한 설명]
 
-형식:
-한 줄의 강렬한 도입문으로 시작하세요.
-그 다음 주인공의 상황과 목표를 명확히 제시하세요.
-마지막으로 독자의 기대감을 자극하는 요소를 추가하세요."""
+📝 **작품소개:**
+[독자의 흥미를 끄는 3-4줄의 작품 소개. 주인공의 상황, 목표, 핵심 갈등을 포함]"""
 
         else:  # English
-            prompt = f"""Create an engaging web novel theme for {genre} genre using these elements:
+            prompt = f"""Create an engaging web novel for {genre} genre using these elements:
 
 【Selected Elements】
 - Core genre: {selected_genre_key}
@@ -941,16 +1049,21 @@ def generate_random_webnovel_theme(genre: str, language: str) -> str:
 【Reference Hook】
 {selected_hook}
 
-Requirements:
-1. Naturally blend these elements into one attractive story
-2. Add settings that Korean web novel readers would love
-3. Make it specific and intriguing enough to suggest a title
-4. Keep it concise, around 200-300 characters
+Format exactly as follows:
+
+📖 **Title:**
+[Attractive and memorable title]
+
+🌍 **Setting:**
+[World and background setting in 3-4 lines]
+
+👥 **Main Characters:**
+• Protagonist: [Name] - [Brief description]
+• Key Character 1: [Name] - [Brief description]
+• Key Character 2: [Name] - [Brief description]
 
-Format:
-Start with one powerful opening line.
-Then clearly present the protagonist's situation and goal.
-Finally add elements that spark reader anticipation."""
+📝 **Synopsis:**
+[3-4 lines that hook readers. Include protagonist's situation, goal, and core conflict]"""
 
         # Call LLM to generate theme
         messages = [{"role": "user", "content": prompt}]
@@ -967,44 +1080,205 @@ def generate_fallback_theme(genre: str, language: str) -> str:
     templates = {
         "로맨스": {
             "themes": [
-                "계약결혼 3개월, 이혼 직전 재벌 남편이 기억을 잃었다. '당신이 내 첫사랑이야.'",
-                "냉혈 검사와 미운정 가득한 이혼 전문 변호사. 법정에서 만날 때마다 불꽃이 튄다."
+                """📖 **제목:** 계약결혼 365일, 기억을 잃은 재벌 남편
+
+🌍 **설정:**
+현대 서울, 대기업 본사와 강남의 펜트하우스가 주 무대. 3개월 계약결혼 만료 직전, 남편이 교통사고로 기억을 잃고 아내를 첫사랑으로 착각하는 상황.
+
+👥 **주요 캐릭터:**
+• 주인공: 서연우(28) - 평범한 회사원, 부모님 병원비를 위해 계약결혼
+• 남주: 강준혁(32) - 냉혈 재벌 3세, 기억상실 후 순정남으로 변신
+• 조연: 한소영(30) - 준혁의 전 약혼녀, 복수를 계획 중
+
+📝 **작품소개:**
+"당신이 내 첫사랑이야." 이혼 서류에 도장을 찍으려던 순간, 교통사고를 당한 냉혈 재벌 남편이 나를 운명의 상대로 착각한다. 3개월간 연기했던 가짜 부부에서 진짜 사랑이 시작되는데...""",
+                
+                """📖 **제목:** 검사님, 이혼 소송은 제가 맡을게요
+
+🌍 **설정:**
+서울중앙지법과 검찰청이 주 무대. 냉혈 검사와 이혼 전문 변호사가 법정에서 대립하며 티격태격하는 법정 로맨스.
+
+👥 **주요 캐릭터:**
+• 주인공: 오지원(30) - 승률 100% 이혼 전문 변호사
+• 남주: 민시준(33) - 원칙주의 엘리트 검사
+• 조연: 박세진(35) - 지원의 전 남편이자 시준의 선배 검사
+
+📝 **작품소개:**
+"변호사님, 법정에서만 만나기로 했잖아요." 하필 전 남편의 불륜 소송을 맡은 날, 상대 검사가 나타났다. 법정에선 적, 밖에선 연인. 우리의 관계는 대체 뭘까?"""
             ]
         },
         "로판": {
             "themes": [
-                "악녀로 빙의했는데 이미 처형 직전. 살려면 북부 전쟁광 공작과 계약결혼해야 한다.",
-                "회귀한 황녀, 이번 생은 언니 대신 폐태자와 정략결혼하기로 했다."
+                """📖 **제목:** 악녀는 이번 생에서 도망친다
+
+🌍 **설정:**
+마법이 존재하는 제국, 1년 후 처형당할 운명의 악녀 공작 영애로 빙의. 북부 변방의 전쟁광 공작과의 계약결혼이 유일한 생존루트.
+
+👥 **주요 캐릭터:**
+• 주인공: 아델라이드(20) - 빙의한 악녀, 원작 지식 보유
+• 남주: 카시우스(25) - 북부의 전쟁광 공작, 숨겨진 순정남
+• 악역: 황태자 레온(23) - 여주에게 집착하는 얀데레
+
+📝 **작품소개:**
+소설 속 악녀로 빙의했는데 이미 처형 선고를 받은 상태? 살려면 원작에 없던 북부 공작과 계약결혼해야 한다. "1년만 함께해주세요. 그 후엔 자유를 드리겠습니다." 하지만 계약 기간이 끝나도 그가 날 놓아주지 않는다.""",
+                
+                """📖 **제목:** 회귀한 황녀는 버려진 왕자를 택한다
+
+🌍 **설정:**
+제국력 892년으로 회귀한 황녀. 전생에서 자신을 배신한 황태자 대신, 버려진 서자 왕자와 손을 잡고 제국을 뒤집으려 한다.
+
+👥 **주요 캐릭터:**
+• 주인공: 로젤린(22) - 회귀한 황녀, 미래를 아는 전략가
+• 남주: 다미안(24) - 버려진 서자 왕자, 숨겨진 흑막
+• 악역: 황태자 세바스찬(26) - 전생의 배신자
+
+📝 **작품소개:**
+독살당해 회귀한 황녀, 이번엔 다르게 살겠다. 모두가 무시하는 서자 왕자의 손을 잡았다. "저와 함께 제국을 뒤집으시겠습니까?" 하지만 그는 내가 아는 것보다 훨씬 위험한 남자였다."""
             ]
         },
         "판타지": {
             "themes": [
-                "F급 각성자, 하지만 내 스킬은 죽은 보스 몬스터를 부활시키는 SSS급 네크로맨서.",
-                "탑 등반 중 사망, 튜토리얼로 회귀했다. 이번엔 혼자 모든 층을 정복한다."
+                """📖 **제목:** F급 헌터, SSS급 네크로맨서가 되다
+
+🌍 **설정:**
+게이트와 던전이 출현한 지 10년 후의 한국. F급 헌터가 우연히 얻은 스킬로 죽은 보스 몬스터를 부활시켜 부리는 유일무이 네크로맨서가 된다.
+
+👥 **주요 캐릭터:**
+• 주인공: 김도현(24) - F급에서 SSS급 네크로맨서로 각성
+• 조력자: 리치 왕(???) - 첫 번째 언데드, 전설의 대마법사
+• 라이벌: 최강훈(26) - S급 길드 마스터, 주인공을 경계
+
+📝 **작품소개:**
+"F급 주제에 무슨 헛소리야?" 모두가 비웃었다. 하지만 첫 번째 보스를 쓰러뜨린 순간, 시스템 메시지가 떴다. [SSS급 히�� 클래스: 네크로맨서 각성] 이제 죽은 보스들이 내 부하가 된다.""",
+                
+                """📖 **제목:** 탑을 역주행하는 회귀자
+
+🌍 **설정:**
+100층 탑 정상에서 죽은 후 튜토리얼로 회귀. 하지만 이번엔 100층부터 거꾸로 내려가며 모든 층을 정복하는 역주행 시스템이 열렸다.
+
+👥 **주요 캐릭터:**
+• 주인공: 이성진(28) - 유일한 역주행 회귀자
+• 조력자: 관리자(???) - 탑의 시스템 AI, 주인공에게 호의적
+• 라이벌: 성하윤(25) - 이번 회차 최강 신인
+
+📝 **작품소개:**
+100층에서 죽었다. 눈을 떠보니 튜토리얼이었다. [역주행 시스템이 개방되었습니다] "뭐? 100층부터 시작한다고?" 최강자의 기억을 가진 채 정상에서부터 내려가는 전무후무한 공략이 시작된다."""
             ]
         },
         "현판": {
             "themes": [
-                "게이트 출현 10년 후, 무능력자였던 내게 시스템 창이 떴다. [유일무이 직업: 아이템 제작사]",
-                "헌터 사관학교 꼴찌, 졸업식 날 S급 게이트가 열렸다. 그리고 나만 살아남았다."
+                """📖 **제목:** 무능력자의 SSS급 아이템 제작
+
+🌍 **설정:**
+게이트 출현 10년, 전 국민의 70%가 각성한 한국. 무능력자로 살던 주인공에게 갑자기 아이템 제작 시스템이 열린다.
+
+👥 **주요 캐릭터:**
+• 주인공: 박준서(25) - 무능력자에서 유일무이 아이템 제작사로
+• 의뢰인: 강하늘(27) - S급 헌터, 첫 번째 고객
+• 라이벌: 대기업 '아르테미스' - 아이템 독점 기업
+
+📝 **작품소개:**
+"각성 등급: 없음" 10년째 무능력자로 살았다. 그런데 오늘, 이상한 시스템 창이 떴다. [SSS급 생산직: 아이템 크래프터] 이제 내가 만든 아이템이 세계를 바꾼다.""",
+                
+                """📖 **제목:** 헌터 사관학교의 숨겨진 최강자
+
+🌍 **설정:**
+한국 최고의 헌터 사관학교. 입학시험 꼴찌로 들어온 주인공이 사실은 능력을 숨기고 있는 특급 요원.
+
+👥 **주요 캐릭터:**
+• 주인공: 윤시우(20) - 꼴찌로 위장한 특급 헌터
+• 히로인: 차유진(20) - 학년 수석, 재벌가 영애
+• 교관: 한태성(35) - 전설의 헌터, 주인공의 정체를 의심
+
+📝 **작품소개:**
+"측정 불가? 그럼 F급이네." 일부러 힘을 숨기고 꼴찌로 입학했다. 하지만 S급 게이트가 학교에 열리면서 정체를 숨길 수 없게 됐다. "너... 대체 누구야?"라는 물음에 어떻게 답해야 할까."""
             ]
         },
         "무협": {
             "themes": [
-                "천하제일 문파의 폐급 막내, 우연히 마교 교주의 비급을 습득했다.",
-                "정파 최고 문파의 장문인으로 회귀. 이번 생은 마교와 손잡고 천하를 뒤집는다."
+                """📖 **제목:** 천하제일문 폐급제자의 마교 비급
+
+🌍 **설정:**
+정파 무림의 중원. 천하제일문의 폐급 막내제자가 우연히 마교 교주의 비급을 습득하고 정마를 아우르는 절대무공을 익힌다.
+
+👥 **주요 캐릭터:**
+• 주인공: 진천(18) - 폐급에서 절대고수로
+• 스승: 혈마노조(???) - 비급에 깃든 마교 전설
+• 라이벌: 남궁세가 소가주 - 정파 제일 천재
+
+📝 **작품소개:**
+"하찮은 것이 감히!" 모두가 무시하던 막내제자. 하지만 떨어진 절벽에서 발견한 것은 전설로만 전해지던 천마신공. "이제부터가 진짜 시작이다." 정파와 마교를 뒤흔들 폐급의 반란이 시작된다.""",
+                
+                """📖 **제목:** 화산파 장문인으로 회귀하다
+
+🌍 **설정:**
+100년 전 화산파가 최고 문파이던 시절로 회귀. 미래를 아는 장문인이 되어 문파를 지키고 무림을 재편한다.
+
+👥 **주요 캐릭터:**
+• 주인공: 청운진인(45→25) - 회귀한 화산파 장문인
+• 제자: 백무진(15) - 미래의 화산파 배신자
+• 맹우: 마교 성녀 - 전생의 적, 이생의 동료
+
+📝 **작품소개:**
+멸문 직전에 회귀했다. 이번엔 다르다. "앞으로 화산파는 정파의 규율을 벗어난다." 미래를 아는 장문인의 파격적인 결정. 마교와 손잡고 무림의 판도를 뒤집는다."""
             ]
         },
         "미스터리": {
             "themes": [
-                "폐쇄된 학교, 연쇄 실종 사건. 나만 범인의 패턴을 알고 있다.",
-                "완벽한 알리바이, 하지만 나는 그가 범인임을 안다. 타임루프에 갇혔으니까."
+                """📖 **제목:** 폐교에 갇힌 7명, 그리고 나
+
+🌍 **설정:**
+폐쇄된 산골 학교, 동창회를 위해 모인 8명이 갇힌다. 하나씩 사라지는 동창들. 범인은 이 안에 있다.
+
+👥 **주요 캐릭터:**
+• 주인공: 서민준(28) - 프로파일러 출신 교사
+• 용의자1: 김태희(28) - 실종된 친구의 전 연인
+• 용의자2: 박진우(28) - 10년 전 사건의 목격자
+
+📝 **작품소개:**
+"10년 전 그날처럼..." 폐교에서 열린 동창회, 하지만 출구는 봉쇄됐다. 한 명씩 사라지는 친구들. 10년 전 묻어둔 비밀이 되살아난다. 살인자는 우리 중 한 명이다.""",
+                
+                """📖 **제목:** 타임루프 속 연쇄살인마를 찾아라
+
+🌍 **설정:**
+같은 하루가 반복되는 타임루프. 매번 다른 방법으로 살인이 일어나지만 범인은 동일인. 루프를 깨려면 범인을 찾아야 한다.
+
+👥 **주요 캐릭터:**
+• 주인공: 강해인(30) - 타임루프에 갇힌 형사
+• 희생자: 이수연(25) - 매번 죽는 카페 알바생
+• 용의자들: 카페 단골 5명 - 각자의 비밀을 숨기고 있음
+
+📝 **작품소개:**
+"또 오늘이야..." 49번째 같은 아침. 오후 3시 33분, 카페에서 살인이 일어난다. 범인을 잡아야 내일이 온다. 하지만 범인은 매번 완벽한 알리바이를 만든다. 과연 50번째 오늘은 다를까?"""
             ]
         },
         "라이트노벨": {
             "themes": [
-                "게임 속 최종보스가 현실에 나타났다. '날 키운 건 너니까, 책임져.'",
-                "평범한 고등학생인 나, 어느 날 전학 온 미소녀가 말했다. '전 사신입니다.'"
+                """📖 **제목:** 내 여자친구가 사실은 마왕이었다
+
+🌍 **설정:**
+평범한 고등학교, 하지만 학생과 교사 중 일부는 이세계에서 온 존재들. 주인공만 모르는 학교의 비밀.
+
+👥 **주요 캐릭터:**
+• 주인공: 김태양(17) - 평범한 고등학생(?)
+• 히로인: 루시퍼(17) - 마왕이자 여자친구
+• 라이벌: 미카엘(17) - 천사이자 학생회장
+
+📝 **작품소개:**
+"선배, 사실 저... 마왕이에요!" 1년째 사귄 여자친구의 충격 고백. 근데 학생회장은 천사고, 담임은 드래곤이라고? 평범한 줄 알았던 우리 학교의 정체가 밝혀진다. "그래서... 우리 헤어져야 해?"라고 묻자 그녀가 울기 시작했다.""",
+                
+                """📖 **제목:** 게임 아이템이 현실에 떨어진다
+
+🌍 **설정:**
+모바일 게임과 현실이 연동되기 시작한 세계. 게임에서 얻은 아이템이 현실에 나타나면서 벌어지는 학원 코미디.
+
+👥 **주요 캐릭터:**
+• 주인공: 박도윤(18) - 게임 폐인 고등학생
+• 히로인: 최서연(18) - 전교 1등, 의외로 게임 고수
+• 친구: 장민혁(18) - 현질 전사, 개그 담당
+
+📝 **작품소개:**
+"어? 이거 내 SSR 무기잖아?" 핸드폰 게임에서 뽑은 아이템이 책상 위에 나타났다. 문제는 학교에 몬스터도 나타나기 시작했다는 것. "야, 수능보다 레이드가 더 중요해진 것 같은데?"라며 웃는 친구들과 함께하는 좌충우돌 학원 판타지."""
             ]
         }
     }
@@ -1012,11 +1286,7 @@ def generate_fallback_theme(genre: str, language: str) -> str:
     genre_themes = templates.get(genre, templates["로맨스"])
     selected = random.choice(genre_themes["themes"])
     
-    if language == "Korean":
-        return selected
-    else:
-        # Simple translation logic
-        return selected.replace("회귀", "regression").replace("빙의", "transmigration")
+    return selected
 
 def generate_theme_with_llm_only(genre: str, language: str) -> str:
     """Generate theme using only LLM when JSON is not available or has errors"""
@@ -1057,43 +1327,49 @@ def generate_theme_with_llm_only(genre: str, language: str) -> str:
     genre_info = genre_prompts.get(genre, genre_prompts["로맨스"])
     
     if language == "Korean":
-        prompt = f"""한국 웹소설 {genre} 장르의 중독성 있는 테마를 생성하세요.
+        prompt = f"""한국 웹소설 {genre} 장르의 중독성 있는 작품을 기획하세요.
 
 다음 인기 요소들을 참고하세요:
 - 핵심 요소: {', '.join(genre_info['elements'])}
 - 인기 훅: {', '.join(genre_info['hooks'])}
 
-요구사항:
-1. 200자 내외의 짧고 임팩트 있는 설정
-2. 첫 문장은 독자를 사로잡는 강렬한 훅
-3. 주인공의 특별한 능력이나 상황 명시
-4. 갈등과 목표가 명확히 드러나도록
-5. 제목이 떠오를 정도로 구체적이면서 흥미로운 설정
-
-예시 형식:
-"[강렬한 첫 문장]"
-[주인공 소개와 특별한 상황]
-[핵심 갈등과 목표]
-[독자의 기대감을 자극하는 마무리]"""
+다음 형식으로 정확히 작성하세요:
+
+📖 **제목:**
+[매력적이고 기억하기 쉬운 제목]
+
+🌍 **설정:**
+[세계관과 배경을 3-4줄로 설명. 시대, 장소, 핵심 설정 포함]
+
+👥 **주요 캐릭터:**
+• 주인공: [이름(나이)] - [직업/신분, 핵심 특징]
+• 주요인물1: [이름(나이)] - [관계/역할, 특��]
+• 주요인물2: [이름(나이)] - [관계/역할, 특징]
+
+📝 **작품소개:**
+[3-4줄로 작품의 핵심 갈등과 매력을 소개. 첫 문장은 강한 훅으로 시작하고, 주인공의 목표와 장애물을 명확히 제시]"""
     else:
-        prompt = f"""Generate an addictive Korean web novel theme for {genre} genre.
+        prompt = f"""Generate an addictive Korean web novel for {genre} genre.
 
 Reference these popular elements:
 - Core elements: {', '.join(genre_info['elements'])}
 - Popular hooks: {', '.join(genre_info['hooks'])}
 
-Requirements:
-1. Around 200 characters, short and impactful
-2. First sentence must be a powerful hook
-3. Clear special ability or situation for protagonist
-4. Clear conflict and goal
-5. Specific and intriguing enough to suggest a title
-
-Example format:
-"[Powerful opening sentence]"
-[Protagonist introduction and special situation]
-[Core conflict and goal]
-[Ending that sparks reader anticipation]"""
+Format exactly as follows:
+
+📖 **Title:**
+[Attractive and memorable title]
+
+🌍 **Setting:**
+[World and background in 3-4 lines. Include era, location, core settings]
+
+👥 **Main Characters:**
+• Protagonist: [Name(Age)] - [Job/Status, key traits]
+• Key Character 1: [Name(Age)] - [Relationship/Role, traits]
+• Key Character 2: [Name(Age)] - [Relationship/Role, traits]
+
+📝 **Synopsis:**
+[3-4 lines introducing core conflict and appeal. Start with strong hook, clearly present protagonist's goal and obstacles]"""
     
     messages = [{"role": "user", "content": prompt}]
     generated_theme = system.call_llm_sync(messages, "writer", language)
@@ -1156,8 +1432,18 @@ def format_webnovel_display(episodes: List[Dict], genre: str) -> str:
         ep_num = ep.get('episode_number', 0)
         content = ep.get('content', '')
         
-        formatted += f"## 제{ep_num}화\n\n"
-        formatted += f"{content}\n\n"
+        # 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
+            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
             formatted += "➡️ *다음 화에 계속...*\n\n"
@@ -1272,6 +1558,46 @@ def create_interface():
         def handle_random_theme(genre, language):
             return generate_random_webnovel_theme(genre, language)
         
+        def handle_download(download_format, session_id, genre):
+            """Handle download request"""
+            if not session_id:
+                return None
+            
+            try:
+                episodes = WebNovelDatabase.get_episodes(session_id)
+                if not episodes:
+                    return None
+                
+                # Get title from first episode or generate default
+                title = f"{genre} 웹소설"
+                
+                if download_format == "TXT":
+                    content = export_to_txt(episodes, genre, title)
+                    
+                    # Save to temporary file
+                    with tempfile.NamedTemporaryFile(mode='w', encoding='utf-8', 
+                                                   suffix='.txt', delete=False) as f:
+                        f.write(content)
+                        return f.name
+                
+                elif download_format == "DOCX":
+                    if not DOCX_AVAILABLE:
+                        gr.Warning("DOCX export requires python-docx library")
+                        return None
+                    
+                    content = export_to_docx(episodes, genre, title)
+                    
+                    # Save to temporary file
+                    with tempfile.NamedTemporaryFile(mode='wb', suffix='.docx', 
+                                                   delete=False) as f:
+                        f.write(content)
+                        return f.name
+                        
+            except Exception as e:
+                logger.error(f"Download error: {e}")
+                gr.Warning(f"다운로드 중 오류 발생: {str(e)}")
+                return None
+        
         # Connect events
         submit_btn.click(
             fn=process_query,
@@ -1285,6 +1611,16 @@ def create_interface():
             outputs=[query_input]
         )
         
+        download_btn.click(
+            fn=handle_download,
+            inputs=[download_format, current_session_id, genre_select],
+            outputs=[download_file]
+        ).then(
+            fn=lambda x: gr.update(visible=True) if x else gr.update(visible=False),
+            inputs=[download_file],
+            outputs=[download_file]
+        )
+        
         # Examples
         gr.Examples(
             examples=[