Update app.py

app.py

@@ -473,6 +473,8 @@ Provide detailed 40-episode plotline centered on the input story setting. Specif
 **전체 플롯에서 {episode_num}화 내용:**
 {self._extract_episode_plan(plot_outline, episode_num)}
 
+⚠️ **중요**: 위의 플롯 내용을 반드시 충실히 따라 작성하세요. 플롯에서 벗어나지 마세요.
+
 **이전 내용 요약:**
 {previous_content[-1500:] if previous_content else "첫 화입니다"}
 
@@ -490,6 +492,7 @@ Provide detailed 40-episode plotline centered on the input story setting. Specif
    - 후크: 다음 화 예고
 
 2. **필수 요소:**
+   - 플롯에 제시된 내용을 충실히 구현
    - 생생한 대화와 행동 묘사
    - 캐릭터 감정과 내면 갈등
    - 장면 전환과 템포 조절
@@ -507,8 +510,7 @@ Provide detailed 40-episode plotline centered on the input story setting. Specif
 **참고 후크 예시:**
 {random.choice(hooks)}
 
-소제목은 이번 화의 핵심 사건이나 전환점을 암시하는 매력적인 문구로 작성하세요.
-{episode_num}화를 풍성하고 몰입감 있게 작성하세요. 반드시 400-600단어로 작성하세요.""",
+플롯에 충실하면서도 몰입감 있게 작성하세요. 반드시 400-600단어로 작성하세요.""",
 
             "English": f"""Write episode {episode_num} of the web novel.
 
@@ -518,6 +520,8 @@ Provide detailed 40-episode plotline centered on the input story setting. Specif
 **Episode {episode_num} from plot:**
 {self._extract_episode_plan(plot_outline, episode_num)}
 
+⚠️ **IMPORTANT**: You MUST faithfully follow the plot content above. Do not deviate from the plot.
+
 **Previous content:**
 {previous_content[-1500:] if previous_content else "First episode"}
 
@@ -535,6 +539,7 @@ Episode {episode_num}. [Attractive subtitle that captures the essence of this ep
    - Hook: Next episode teaser
 
 2. **Essential elements:**
+   - Faithfully implement the plot content
    - Vivid dialogue and action
    - Character emotions and conflicts
    - Scene transitions and pacing
@@ -552,8 +557,7 @@ Episode {episode_num}. [Attractive subtitle that captures the essence of this ep
 **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."""
+Write faithfully to the plot while being immersive. Must be 400-600 words."""
         }
         
         return lang_prompts.get(language, lang_prompts["Korean"])
@@ -603,18 +607,39 @@ Provide specific improvements."""
         
         patterns = [
             f"{episode_num}화:", f"Episode {episode_num}:", 
-            f"제{episode_num}화:", f"EP{episode_num}:"
+            f"제{episode_num}화:", f"EP{episode_num}:",
+            f"{episode_num}.", f"[{episode_num}]"
+        ]
+        
+        # Also check for next episode patterns
+        next_patterns = [
+            f"{episode_num+1}화:", f"Episode {episode_num+1}:", 
+            f"제{episode_num+1}화:", f"EP{episode_num+1}:",
+            f"{episode_num+1}.", f"[{episode_num+1}]"
         ]
         
         for line in lines:
+            # Start capturing when we find the episode number
             if any(pattern in line for pattern in patterns):
                 capturing = True
-            elif capturing and any(f"{episode_num+1}" in line for pattern in patterns):
+                episode_section.append(line)
+            # Stop capturing when we find the next episode number
+            elif capturing and any(pattern in line for pattern in next_patterns):
                 break
             elif capturing:
                 episode_section.append(line)
-                
-        return '\n'.join(episode_section) if episode_section else "플롯을 참고하여 작성"
+        
+        # If we found episode content, return it
+        if episode_section:
+            return '\n'.join(episode_section)
+        
+        # If no specific episode found, provide more context
+        logger.warning(f"Could not find specific plan for episode {episode_num}")
+        return f"""에피소드 {episode_num}에 대한 구체적인 플롯을 찾을 수 없습니다.
+전체 플롯을 참고하여 {episode_num}화를 작성하되, 반드시 사용자가 제공한 원본 스토리 설정을 따르세요.
+
+참고: 전체 플롯 일부
+{plot_outline[:1000]}..."""
 
     # --- LLM call functions ---
     def call_llm_sync(self, messages: List[Dict[str, str]], role: str, language: str) -> str:
@@ -701,38 +726,48 @@ Provide specific improvements."""
 독자를 중독시키는 플롯과 전개를 설계합니다.
 장르별 관습과 독자 기대를 정확히 파악합니다.
 40화 완결 구조로 완벽한 기승전결을 만듭니다.
-각 화마다 충실한 내용과 전개를 계획합니다.""",
+각 화마다 충실한 내용과 전개를 계획합니다.
+
+⚠️ 가장 중요한 원칙: 사용자가 제공한 스토리 설정을 절대적으로 우선시하고, 이를 중심으로 모든 플롯을 구성합니다. 장르 관습보다 사용자의 구체적 설정이 항상 우선입니다.""",
                 
                 "writer": """당신은 독자를 사로잡는 웹소설 작가입니다.
 풍부하고 몰입감 있는 문체를 구사합니다.
 각 화를 400-600단어로 충실하게 작성합니다.
 여러 장면과 전환을 통해 이야기를 전개합니다.
 대화, 행동, 내면 묘사를 균형있게 배치합니다.
-매 화 끝에 강력한 후크로 다음 화를 기다리게 만듭니다.""",
+매 화 끝에 강력한 후크로 다음 화를 기다리게 만듭니다.
+
+⚠️ 가장 중요한 원칙: 제공된 플롯 아웃라인을 정확히 따르고, 절대 임의로 내용을 변경하거나 추가하지 않습니다. 플롯에 명시된 내용만을 충실히 구현합니다.""",
                 
                 "critic": """당신은 웹소설 독자의 마음을 읽는 평론가입니다.
 재미와 몰입감을 최우선으로 평가합니다.
 장르적 쾌감과 독자 만족도를 분석합니다.
-구체적이고 실용적인 개선안을 제시합니다."""
+구체적이고 실용적인 개선안을 제시합니다.
+플롯 충실도와 일관성을 중요하게 평가합니다."""
             },
             "English": {
                 "planner": """You perfectly understand the Korean web novel market.
 Design addictive plots and developments.
 Accurately grasp genre conventions and reader expectations.
 Create perfect story structure in 40 episodes.
-Plan substantial content and development for each episode.""",
+Plan substantial content and development for each episode.
+
+⚠️ Most important principle: Absolutely prioritize the user's story setting and build all plots around it. User's specific settings always take precedence over genre conventions.""",
                 
                 "writer": """You are a web novelist who captivates readers.
 Use rich and immersive writing style.
 Write each episode with 400-600 words faithfully.
 Develop story through multiple scenes and transitions.
 Balance dialogue, action, and inner descriptions.
-End each episode with powerful hook for next.""",
+End each episode with powerful hook for next.
+
+⚠️ Most important principle: Follow the provided plot outline exactly and never arbitrarily change or add content. Faithfully implement only what is specified in the plot.""",
                 
                 "critic": """You read web novel readers' minds.
 Prioritize fun and immersion in evaluation.
 Analyze genre satisfaction and reader enjoyment.
-Provide specific, practical improvements."""
+Provide specific, practical improvements.
+Evaluate plot fidelity and consistency as important factors."""
             }
         }
         
@@ -753,6 +788,8 @@ Provide specific, practical improvements."""
                 self.current_session_id = WebNovelDatabase.create_session(query, genre, language)
                 self.tracker.set_genre(genre)
                 logger.info(f"Created new session: {self.current_session_id}")
+                # Store the original query for reference
+                self.original_query = query
 
             # Generate plot outline first
             if resume_from_episode == 0:
@@ -764,6 +801,9 @@ Provide specific, practical improvements."""
                     "planner", language
                 )
                 
+                # Store plot outline for debugging
+                self.plot_outline = plot_outline
+                
                 yield "✅ 플롯 구성 완료!", "", f"40화 구성 완료", self.current_session_id
 
             # Generate episodes
@@ -772,12 +812,21 @@ Provide specific, practical improvements."""
                 # Write episode
                 yield f"✍️ {episode_num}화 집필 중...", accumulated_content, f"진행률: {episode_num}/{TARGET_EPISODES}화", self.current_session_id
                 
+                # Create enhanced episode prompt with original query reminder
                 episode_prompt = self.create_episode_prompt(
                     episode_num, plot_outline, accumulated_content, genre, language
                 )
                 
+                # Add reminder about original query before calling LLM
+                enhanced_prompt = f"""⚠️ 필수 확인사항:
+원본 스토리 설정: {query}
+
+이 설정을 반드시 반영하여 작성하세요.
+
+{episode_prompt}"""
+                
                 episode_content = self.call_llm_sync(
-                    [{"role": "user", "content": episode_prompt}],
+                    [{"role": "user", "content": enhanced_prompt}],
                     "writer", language
                 )
                 
@@ -1105,9 +1154,9 @@ def generate_random_webnovel_theme(genre: str, language: str) -> str:
 [세계관과 배경 설정을 3-4줄로 설명]
 
 👥 **주요 캐릭터:**
-• 주인공: [이름] - [간단한 설명]
-• 주요인물1: [이름] - [간단한 설명]
-• 주요인물2: [이름] - [간단한 설명]
+- 주인공: [이름] - [간단한 설명]
+- 주요인물1: [이름] - [간단한 설명]
+- 주요인물2: [이름] - [간단한 설명]
 
 📝 **작품소개:**
 [독자의 흥미를 끄는 3-4줄의 작품 소개. 주인공의 상황, 목표, 핵심 갈등을 포함]"""
@@ -1140,9 +1189,9 @@ Format exactly as follows:
 [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]
+- Protagonist: [Name] - [Brief description]
+- Key Character 1: [Name] - [Brief description]
+- Key Character 2: [Name] - [Brief description]
 
 📝 **Synopsis:**
 [3-4 lines that hook readers. Include protagonist's situation, goal, and core conflict]"""
@@ -1169,9 +1218,9 @@ def generate_fallback_theme(genre: str, language: str) -> str:
 현대 서울, 대기업 본사와 강남의 펜트하우스가 주 무대. 3개월 계약결혼 만료 직전, 남편이 교통사고로 기억을 잃고 아내를 첫사랑으로 착각하는 상황.
 
 👥 **주요 캐릭터:**
-• 주인공: 서연우(28) - 평범한 회사원, 부모님 병원비를 위해 계약결혼
-• 남주: 강준혁(32) - 냉혈 재벌 3세, 기억상실 후 순정남으로 변신
-• 조연: 한소영(30) - 준혁의 전 약혼녀, 복수를 계획 중
+- 주인공: 서연우(28) - 평범한 회사원, 부모님 병원비를 위해 계약결혼
+- 남주: 강준혁(32) - 냉혈 재벌 3세, 기억상실 후 순정남으로 변신
+- 조연: 한소영(30) - 준혁의 전 약혼녀, 복수를 계획 중
 
 📝 **작품소개:**
 "당신이 내 첫사랑이야." 이혼 서류에 도장을 찍으려던 순간, 교통사고를 당한 냉혈 재벌 남편이 나를 운명의 상대로 착각한다. 3개월간 연기했던 가짜 부부에서 진짜 사랑이 시작되는데...""",
@@ -1182,235 +1231,235 @@ def generate_fallback_theme(genre: str, language: str) -> str:
 서울중앙지법과 검찰청이 주 무대. 냉혈 검사와 이혼 전문 변호사가 법정에서 대립하며 티격태격하는 법정 로맨스.
 
 👥 **주요 캐릭터:**
-• 주인공: 오지원(30) - 승률 100% 이혼 전문 변호사
-• 남주: 민시준(33) - 원칙주의 엘리트 검사
-• 조연: 박세진(35) - 지원의 전 남편이자 시준의 선배 검사
+- 주인공: 오지원(30) - 승률 100% 이혼 전문 변호사
+- 남주: 민시준(33) - 원칙주의 엘리트 검사
+- 조연: 박세진(35) - 지원의 전 남편이자 시준의 선배 검사
 
 📝 **작품소개:**
 "변호사님, 법정에서만 만나기로 했잖아요." 하필 전 남편의 불륜 소송을 맡은 날, 상대 검사가 나타났다. 법정에선 적, 밖에선 연인. 우리의 관계는 대체 뭘까?"""
-            ]
-        },
-        "로판": {
-            "themes": [
-                """📖 **제목:** 악녀는 이번 생에서 도망친다
+           ]
+       },
+       "로판": {
+           "themes": [
+               """📖 **제목:** 악녀는 이번 생에서 도망친다
 
 🌍 **설정:**
 마법이 존재하는 제국, 1년 후 처형당할 운명의 악녀 공작 영애로 빙의. 북부 변방의 전쟁광 공작과의 계약결혼이 유일한 생존루트.
 
 👥 **주요 캐릭터:**
-• 주인공: 아델라이드(20) - 빙의한 악녀, 원작 지식 보유
-• 남주: 카시우스(25) - 북부의 전쟁광 공작, 숨겨진 순정남
-• 악역: 황태자 레온(23) - 여주에게 집착하는 얀데레
+- 주인공: 아델라이드(20) - 빙의한 악녀, 원작 지식 보유
+- 남주: 카시우스(25) - 북부의 전쟁광 공작, 숨겨진 순정남
+- 악역: 황태자 레온(23) - 여주에게 집착하는 얀데레
 
 📝 **작품소개:**
 소설 속 악녀로 빙의했는데 이미 처형 선고를 받은 상태? 살려면 원작에 없던 북부 공작과 계약결혼해야 한다. "1년만 함께해주세요. 그 후엔 자유를 드리겠습니다." 하지만 계약 기간이 끝나도 그가 날 놓아주지 않는다.""",
-                
-                """📖 **제목:** 회귀한 황녀는 버려진 왕자를 택한다
+               
+               """📖 **제목:** 회귀한 황녀는 버려진 왕자를 택한다
 
 🌍 **설정:**
 제국력 892년으로 회귀한 황녀. 전생에서 자신을 배신한 황태자 대신, 버려진 서자 왕자와 손을 잡고 제국을 뒤집으려 한다.
 
 👥 **주요 캐릭터:**
-• 주인공: 로젤린(22) - 회귀한 황녀, 미래를 아는 전략가
-• 남주: 다미안(24) - 버려진 서자 왕자, 숨겨진 흑막
-• 악역: 황태자 세바스찬(26) - 전생의 배신자
+- 주인공: 로젤린(22) - 회귀한 황녀, 미래를 아는 전략가
+- 남주: 다미안(24) - 버려진 서자 왕자, 숨겨진 흑막
+- 악역: 황태자 세바스찬(26) - 전생의 배신자
 
 📝 **작품소개:**
 독살당해 회귀한 황녀, 이번엔 다르게 살겠다. 모두가 무시하는 서자 왕자의 손을 잡았다. "저와 함께 제국을 뒤집으시겠습니까?" 하지만 그는 내가 아는 것보다 훨씬 ��험한 남자였다."""
-            ]
-        },
-        "판타지": {
-            "themes": [
-                """📖 **제목:** F급 헌터, SSS급 네크로맨서가 되다
+           ]
+       },
+       "판타지": {
+           "themes": [
+               """📖 **제목:** F급 헌터, SSS급 네크로맨서가 되다
 
 🌍 **설정:**
 게이트와 던전이 출현한 지 10년 후의 한국. F급 헌터가 우연히 얻은 스킬로 죽은 보스 몬스터를 부활시켜 부리는 유일무이 네크로맨서가 된다.
 
 👥 **주요 캐릭터:**
-• 주인공: 김도현(24) - F급에서 SSS급 네크로맨서로 각성
-• 조력자: 리치 왕(???) - 첫 번째 언데드, 전설의 대마법사
-• 라이벌: 최강훈(26) - S급 길드 마스터, 주인공을 경계
+- 주인공: 김도현(24) - F급에서 SSS급 네크로맨서로 각성
+- 조력자: 리치 왕(???) - 첫 번째 언데드, 전설의 대마법사
+- 라이벌: 최강훈(26) - S급 길드 마스터, 주인공을 경계
 
 📝 **작품소개:**
 "F급 주제에 무슨 헛소리야?" 모두가 비웃었다. 하지만 첫 번째 보스를 쓰러뜨린 순간, 시스템 메시지가 떴다. [SSS급 히든 클래스: 네크로맨서 각성] 이제 죽은 보스들이 내 부하가 된다.""",
-                
-                """📖 **제목:** 탑을 역주행하는 회귀자
+               
+               """📖 **제목:** 탑을 역주행하는 회귀자
 
 🌍 **설정:**
 100층 탑 정상에서 죽은 후 튜토리얼로 회귀. 하지만 이번엔 100층부터 거꾸로 내려가며 모든 층을 정복하는 역주행 시스템이 열렸다.
 
 👥 **주요 캐릭터:**
-• 주인공: 이성진(28) - 유일한 역주행 회귀자
-• 조력자: 관리자(???) - 탑의 시스템 AI, 주인공에게 호의적
-• 라이벌: 성하윤(25) - 이번 회차 최강 신인
+- 주인공: 이성진(28) - 유일한 역주행 회귀자
+- 조력자: 관리자(???) - 탑의 시스템 AI, 주인공에게 호의적
+- 라이벌: 성하윤(25) - 이번 회차 최강 신인
 
 📝 **작품소개:**
 100층에서 죽었다. 눈을 떠보니 튜토리얼이었다. [역주행 시스템이 개방되었습니다] "뭐? 100층부터 시작한다고?" 최강자의 기억을 가진 채 정상에서부터 내려가는 전무후무한 공략이 시작된다."""
-            ]
-        },
-        "현판": {
-            "themes": [
-                """📖 **제목:** 무능력자의 SSS급 아이템 제작
+           ]
+       },
+       "현판": {
+           "themes": [
+               """📖 **제목:** 무능력자의 SSS급 아이템 제작
 
 🌍 **설정:**
 게이트 출현 10년, 전 국민의 70%가 각성한 한국. 무능력자로 살던 주인공에게 갑자기 아이템 제작 시스템이 열린다.
 
 👥 **주요 캐릭터:**
-• 주인공: 박준서(25) - 무능력자에서 유일무이 아이템 제작사로
-• 의뢰인: 강하늘(27) - S급 헌터, 첫 번째 고객
-• 라이벌: 대기업 '아르테미스' - 아이템 독점 기업
+- 주인공: 박준서(25) - 무능력자에서 유일무이 아이템 제작사로
+- 의뢰인: 강하늘(27) - S급 헌터, 첫 번째 고객
+- 라이벌: 대기업 '아르테미스' - 아이템 독점 기업
 
 📝 **작품소개:**
 "각성 등급: 없음" 10년째 무능력자로 살았다. 그런데 오늘, 이상한 시스템 창이 떴다. [SSS급 생산직: 아이템 크래프터] 이제 내가 만든 아이템이 세계를 바꾼다.""",
-                
-                """📖 **제목:** 헌터 사관학교의 숨겨진 최강자
+               
+               """📖 **제목:** 헌터 사관학교의 숨겨진 최강자
 
 🌍 **설정:**
 한국 최고의 헌터 사관학교. 입학시험 꼴찌로 들어온 주인공이 사실은 능력을 숨기고 있는 특급 요원.
 
 👥 **주요 캐릭터:**
-• 주인공: 윤시우(20) - 꼴찌로 위장한 특급 헌터
-• 히로인: 차유진(20) - 학년 수석, 재벌가 영애
-• 교관: 한태성(35) - 전설의 헌터, 주인공의 정체를 의심
+- 주인공: 윤시우(20) - 꼴찌로 위장한 특급 헌터
+- 히로인: 차유진(20) - 학년 수석, 재벌가 영애
+- 교관: 한태성(35) - 전설의 헌터, 주인공의 정체를 의심
 
 📝 **작품소개:**
 "측정 불가? 그럼 F급이네." 일부러 힘을 숨기고 꼴찌로 입학했다. 하지만 S급 게이트가 학교에 열리면서 정체를 숨길 수 없게 됐다. "너... 대체 누구야?"라는 물음에 어떻게 답해야 할까."""
-            ]
-        },
-        "무협": {
-            "themes": [
-                """📖 **제목:** 천하제일문 폐급제자의 마교 비급
+           ]
+       },
+       "무협": {
+           "themes": [
+               """📖 **제목:** 천하제일문 폐급제자의 마교 비급
 
 🌍 **설정:**
 정파 무림의 중원. 천하제일문의 폐급 막내제자가 우연히 마교 교주의 비급을 습득하고 정마를 아우르는 절대무공을 익힌다.
 
 👥 **주요 캐릭터:**
-• 주인공: 진천(18) - 폐급에서 절대고수로
-• 스승: 혈마노조(???) - 비급에 깃든 마교 전설
-• 라이벌: 남궁세가 소가주 - 정파 제일 천재
+- 주인공: 진천(18) - 폐급에서 절대고수로
+- 스승: 혈마노조(???) - 비급에 깃든 마교 전설
+- 라이벌: 남궁세가 소가주 - 정파 제일 천재
 
 📝 **작품소개:**
 "하찮은 것이 감히!" 모두가 무시하던 막내제자. 하지만 떨어진 절벽에서 발견한 것은 전설로만 전해지던 천마신공. "이제부터가 진짜 시작이다." 정파와 마교를 뒤흔들 폐급의 반란이 시작된다.""",
-                
-                """📖 **제목:** 화산파 장문인으로 회귀하다
+               
+               """📖 **제목:** 화산파 장문인으로 회귀하다
 
 🌍 **설정:**
 100년 전 화산파가 최고 문파이던 시절로 회귀. 미래를 아는 장문인이 되어 문파를 지키고 무림을 재편한다.
 
 👥 **주요 캐릭터:**
-• 주인공: 청운진인(45→25) - 회귀한 화산파 장문인
-• 제자: 백무진(15) - 미래의 화산파 배신자
-• 맹우: 마교 성녀 - 전생의 적, 이생의 동료
+- 주인공: 청운진인(45→25) - 회귀한 화산파 장문인
+- 제자: 백무진(15) - 미래의 화산파 배신자
+- 맹우: 마교 성녀 - 전생의 적, 이생의 동료
 
 📝 **작품소개:**
 멸문 직전에 회귀했다. 이번엔 다르다. "앞으로 화산파는 정파의 규율을 벗어난다." 미래를 아는 장문인의 파격적인 결정. 마교와 손잡고 무림의 판도를 뒤집는다."""
-            ]
-        },
-        "미스터리": {
-            "themes": [
-                """📖 **제목:** 폐교에 갇힌 7명, 그리고 나
+           ]
+       },
+       "미스터리": {
+           "themes": [
+               """📖 **제목:** 폐교에 갇힌 7명, 그리고 나
 
 🌍 **설정:**
 폐쇄된 산골 학교, 동창회를 위해 모인 8명이 갇힌다. 하나씩 사라지는 동창들. 범인은 이 안에 있다.
 
 👥 **주요 캐릭터:**
-• 주인공: 서민준(28) - 프로파일러 출신 교사
-• 용의자1: 김태희(28) - 실종된 친구의 전 연인
-• 용의자2: 박진우(28) - 10년 전 사건의 목격자
+- 주인공: 서민준(28) - 프로파일러 출신 교사
+- 용의자1: 김태희(28) - 실종된 친구의 전 연인
+- 용의자2: 박진우(28) - 10년 전 사건의 목격자
 
 📝 **작품소개:**
 "10년 전 그날처럼..." 폐교에서 열린 동창회, 하지만 출구는 봉쇄됐다. 한 명씩 사라지는 친구들. 10년 전 묻어둔 비밀이 되살아난다. 살인자는 우리 중 한 명이다.""",
-                
-                """📖 **제목:** 타임루프 속 연쇄살인마를 찾아라
+               
+               """📖 **제목:** 타임루프 속 연쇄살인마를 찾아라
 
 🌍 **설정:**
 같은 하루가 반복되는 타임루프. 매번 다른 방법으로 살인이 일어나지만 범인은 동일인. 루프를 깨려면 범인을 찾아야 한다.
 
 👥 **주요 캐릭터:**
-• 주인공: 강해인(30) - 타임루프에 갇힌 형사
-• 희생자: 이수연(25) - 매번 죽는 카페 알바생
-• 용의자들: 카페 단골 5명 - 각자의 비밀을 숨기고 있음
+- 주인공: 강해인(30) - 타임루프에 갇힌 형사
+- 희생자: 이수연(25) - 매번 죽는 카페 알바생
+- 용의자들: 카페 단골 5명 - 각자의 비밀을 숨기고 있음
 
 📝 **작품소개:**
 "또 오늘이야..." 49번째 같은 아침. 오후 3시 33분, 카페에서 살인이 일어난다. 범인을 잡아야 내일이 온다. 하지만 범인은 매번 완벽한 알리바이를 만든다. 과연 50번째 오늘은 다를까?"""
-            ]
-        },
-        "라이트노벨": {
-            "themes": [
-                """📖 **제목:** 내 여자친구가 사실은 마왕이었다
+           ]
+       },
+       "라이트노벨": {
+           "themes": [
+               """📖 **제목:** 내 여자친구가 사실은 마왕이었다
 
 🌍 **설정:**
 평범한 고등학교, 하지만 학생과 교사 중 일부는 이세계에서 온 존재들. 주인공만 모르는 학교의 비밀.
 
 👥 **주요 캐릭터:**
-• 주인공: 김태양(17) - 평범한 고등학생(?)
-• 히로인: 루시퍼(17) - 마왕이자 여자친구
-• 라이벌: 미카엘(17) - 천사이자 학생회장
+- 주인공: 김태양(17) - 평범한 고등학생(?)
+- 히로인: 루시퍼(17) - 마왕이자 여자친구
+- 라이벌: 미카엘(17) - 천사이자 학생회장
 
 📝 **작품소개:**
 "선배, 사실 저... 마왕이에요!" 1년째 사귄 여자친구의 충격 고백. 근데 학생회장은 천사고, 담임은 드래곤이라고? 평범한 줄 알았던 우리 학교의 정체가 밝혀진다. "그래서... 우리 헤어져야 해?"라고 묻자 그녀가 울기 시작했다.""",
-                
-                """📖 **제목:** 게�� 아이템이 현실에 떨어진다
+               
+               """📖 **제목:** 게임 아이템이 현실에 떨어진다
 
 🌍 **설정:**
 모바일 게임과 현실이 연동되기 시작한 세계. 게임에서 얻은 아이템이 현실에 나타나면서 벌어지는 학원 코미디.
 
 👥 **주요 캐릭터:**
-• 주인공: 박도윤(18) - 게임 폐인 고등학생
-• 히로인: 최서연(18) - 전교 1등, 의외로 게임 고수
-• 친구: 장민혁(18) - 현질 전사, 개그 담당
+- 주인공: 박도윤(18) - 게임 폐인 고등학생
+- 히로인: 최서연(18) - 전교 1등, 의외로 게임 고수
+- 친구: 장민혁(18) - 현질 전사, 개그 담당
 
 📝 **작품소개:**
 "어? 이거 내 SSR 무기잖아?" 핸드폰 게임에서 뽑은 아이템이 책상 위에 나타났다. 문제는 학교에 몬스터도 나타나기 시작했다는 것. "야, 수능보다 레이드가 더 중요해진 것 같은데?"라며 웃는 친구들과 함께하는 좌충우돌 학원 판타지."""
-            ]
-        }
-    }
-    
-    genre_themes = templates.get(genre, templates["로맨스"])
-    selected = random.choice(genre_themes["themes"])
-    
-    return selected
+           ]
+       }
+   }
+   
+   genre_themes = templates.get(genre, templates["로맨스"])
+   selected = random.choice(genre_themes["themes"])
+   
+   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"""
-    system = WebNovelSystem()
-    
-    # Genre-specific prompts based on popular web novel trends
-    genre_prompts = {
-        "로맨스": {
-            "elements": ["계약결혼", "재벌", "이혼", "첫사랑", "운명적 만남", "오해와 화해"],
-            "hooks": ["기억상실", "정체 숨기기", "가짜 연인", "원나잇 후 재회"]
-        },
-        "로판": {
-            "elements": ["빙의", "회귀", "악녀", "황녀", "공작", "원작 파괴"],
-            "hooks": ["처형 직전", "파혼 선언", "독살 시도", "폐위 위기"]
-        },
-        "판타지": {
-            "elements": ["시스템", "각성", "던전", "회귀", "탑 등반", "SSS급"],
-            "hooks": ["F급에서 시작", "숨겨진 클래스", "유일무이 스킬", "죽음 후 각성"]
-        },
-        "현판": {
-            "elements": ["헌터", "게이트", "각성자", "길드", "아이템", "랭킹"],
-            "hooks": ["늦은 각성", "재능 재평가", "S급 게이트", "시스템 오류"]
-        },
-        "무협": {
-            "elements": ["회귀", "천재", "마교", "비급", "복수", "환생"],
-            "hooks": ["폐급에서 최강", "배신 후 각성", "숨겨진 혈통", "기연 획득"]
-        },
-        "미스터리": {
-            "elements": ["탐정", "연쇄살인", "타임루프", "초능력", "과거의 비밀"],
-            "hooks": ["밀실 살인", "예고 살인", "기억 조작", "시간 역행"]
-        },
-        "라이트노벨": {
-            "elements": ["학원", "이세계", "히로인", "게임", "일상", "판타지"],
-            "hooks": ["전학생 정체", "게임 현실화", "평행세계", "숨겨진 능력"]
-        }
-    }
-    
-    genre_info = genre_prompts.get(genre, genre_prompts["로맨스"])
-    
-    if language == "Korean":
-        prompt = f"""한국 웹소설 {genre} 장르의 중독성 있는 작품을 기획하세요.
+   """Generate theme using only LLM when JSON is not available or has errors"""
+   system = WebNovelSystem()
+   
+   # Genre-specific prompts based on popular web novel trends
+   genre_prompts = {
+       "로맨스": {
+           "elements": ["계약결혼", "재벌", "이혼", "첫사랑", "운명적 만남", "오해와 화해"],
+           "hooks": ["기억상실", "정체 숨기기", "가짜 연인", "원나잇 후 재회"]
+       },
+       "로판": {
+           "elements": ["빙의", "회귀", "악녀", "황녀", "공작", "원작 파괴"],
+           "hooks": ["처형 직전", "파혼 선언", "독살 시도", "폐위 위기"]
+       },
+       "판타지": {
+           "elements": ["시스템", "각성", "던전", "회귀", "탑 등반", "SSS급"],
+           "hooks": ["F급에서 시작", "숨겨진 클래스", "유일무이 스킬", "죽음 후 각성"]
+       },
+       "현판": {
+           "elements": ["헌터", "게이트", "각성자", "길드", "아이템", "랭킹"],
+           "hooks": ["늦은 각성", "재능 재평가", "S급 게이트", "시스템 오류"]
+       },
+       "무협": {
+           "elements": ["회귀", "천재", "마교", "비급", "복수", "환생"],
+           "hooks": ["폐급에서 최강", "배신 후 각성", "숨겨진 혈통", "기연 획득"]
+       },
+       "미스터리": {
+           "elements": ["탐정", "연쇄살인", "타임루프", "초능력", "과거의 비밀"],
+           "hooks": ["밀실 살인", "예고 살인", "기억 조작", "시간 역행"]
+       },
+       "라이트노벨": {
+           "elements": ["학원", "이세계", "히로인", "게임", "일상", "판타지"],
+           "hooks": ["전학생 정체", "게임 현실화", "평행세계", "숨겨진 능력"]
+       }
+   }
+   
+   genre_info = genre_prompts.get(genre, genre_prompts["로맨스"])
+   
+   if language == "Korean":
+       prompt = f"""한국 웹소설 {genre} 장르의 중독성 있는 작품을 기획하세요.
 
 다음 인기 요소들을 참고하세요:
 - 핵심 요소: {', '.join(genre_info['elements'])}
@@ -1425,14 +1474,14 @@ def generate_theme_with_llm_only(genre: str, language: str) -> str:
 [세계관과 배경을 3-4줄로 설명. 시대, 장소, 핵심 설정 포함]
 
 👥 **주요 캐릭터:**
-• 주인공: [이름(나이)] - [직업/신분, 핵심 특징]
-• 주요인물1: [이름(나이)] - [관계/역할, 특징]
-• 주요인물2: [이름(나이)] - [관계/역할, 특징]
+- 주인공: [이름(나이)] - [직업/신분, 핵심 특징]
+- 주요인물1: [이름(나이)] - [관계/역할, 특징]
+- 주요인물2: [이름(나이)] - [관계/역할, 특징]
 
 📝 **작품소개:**
 [3-4줄로 작품의 핵심 갈등과 매력을 소개. 첫 문장은 강한 훅으로 시작하고, 주인공의 목표와 장애물을 명확히 제시]"""
-    else:
-        prompt = f"""Generate an addictive Korean web novel for {genre} genre.
+   else:
+       prompt = f"""Generate an addictive Korean web novel for {genre} genre.
 
 Reference these popular elements:
 - Core elements: {', '.join(genre_info['elements'])}
@@ -1447,370 +1496,370 @@ Format exactly as follows:
 [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]
+- 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)
-    
-    return generated_theme
+   
+   messages = [{"role": "user", "content": prompt}]
+   generated_theme = system.call_llm_sync(messages, "writer", language)
+   
+   return generated_theme
 
 # --- UI functions ---
 def format_episodes_display(episodes: List[Dict], current_episode: int = 0) -> str:
-    """Format episodes for display"""
-    markdown = "## 📚 웹소설 연재 현황\n\n"
-    
-    if not episodes:
-        return markdown + "*아직 작성된 에피소드가 없습니다.*"
-    
-    # Stats
-    total_episodes = len(episodes)
-    total_words = sum(ep.get('word_count', 0) for ep in episodes)
-    avg_engagement = sum(ep.get('reader_engagement', 0) for ep in episodes) / len(episodes) if episodes else 0
-    
-    markdown += f"**진행 상황:** {total_episodes} / {TARGET_EPISODES}화\n"
-    markdown += f"**총 단어 수:** {total_words:,} / {TARGET_WORDS:,}\n"
-    markdown += f"**평균 몰입도:** ⭐ {avg_engagement:.1f} / 10\n\n"
-    markdown += "---\n\n"
-    
-    # Episode list
-    for ep in episodes[-5:]:  # Show last 5 episodes
-        ep_num = ep.get('episode_number', 0)
-        word_count = ep.get('word_count', 0)
-        
-        markdown += f"### 📖 {ep_num}화\n"
-        markdown += f"*{word_count}단어*\n\n"
-        
-        content = ep.get('content', '')
-        if content:
-            preview = content[:200] + "..." if len(content) > 200 else content
-            markdown += f"{preview}\n\n"
-            
-            hook = ep.get('hook', '')
-            if hook:
-                markdown += f"**🪝 후크:** *{hook}*\n\n"
-        
-        markdown += "---\n\n"
-    
-    return markdown
+   """Format episodes for display"""
+   markdown = "## 📚 웹소설 연재 현황\n\n"
+   
+   if not episodes:
+       return markdown + "*아직 작성된 에피소드가 없습니다.*"
+   
+   # Stats
+   total_episodes = len(episodes)
+   total_words = sum(ep.get('word_count', 0) for ep in episodes)
+   avg_engagement = sum(ep.get('reader_engagement', 0) for ep in episodes) / len(episodes) if episodes else 0
+   
+   markdown += f"**진행 상황:** {total_episodes} / {TARGET_EPISODES}화\n"
+   markdown += f"**총 단어 수:** {total_words:,} / {TARGET_WORDS:,}\n"
+   markdown += f"**평균 몰입도:** ⭐ {avg_engagement:.1f} / 10\n\n"
+   markdown += "---\n\n"
+   
+   # Episode list
+   for ep in episodes[-5:]:  # Show last 5 episodes
+       ep_num = ep.get('episode_number', 0)
+       word_count = ep.get('word_count', 0)
+       
+       markdown += f"### 📖 {ep_num}화\n"
+       markdown += f"*{word_count}단어*\n\n"
+       
+       content = ep.get('content', '')
+       if content:
+           preview = content[:200] + "..." if len(content) > 200 else content
+           markdown += f"{preview}\n\n"
+           
+           hook = ep.get('hook', '')
+           if hook:
+               markdown += f"**🪝 후크:** *{hook}*\n\n"
+       
+       markdown += "---\n\n"
+   
+   return markdown
 
 def format_webnovel_display(episodes: List[Dict], genre: str) -> str:
-    """Format complete web novel for display"""
-    if not episodes:
-        return "아직 완성된 웹소설이 없습니다."
-    
-    formatted = f"# 🎭 {genre} 웹소설\n\n"
-    
-    # Novel stats
-    total_words = sum(ep.get('word_count', 0) for ep in episodes)
-    formatted += f"**총 {len(episodes)}화 완결 | {total_words:,}단어**\n\n"
-    formatted += "---\n\n"
-    
-    # Episodes
-    for idx, ep in enumerate(episodes):
-        ep_num = ep.get('episode_number', 0)
-        content = ep.get('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"
-        
-        if idx < len(episodes) - 1:  # Not last episode
-            formatted += "➡️ *다음 화에 계속...*\n\n"
-        
-        formatted += "---\n\n"
-    
-    return formatted
+   """Format complete web novel for display"""
+   if not episodes:
+       return "아직 완성된 웹소설이 없습니다."
+   
+   formatted = f"# 🎭 {genre} 웹소설\n\n"
+   
+   # Novel stats
+   total_words = sum(ep.get('word_count', 0) for ep in episodes)
+   formatted += f"**총 {len(episodes)}화 완결 | {total_words:,}단어**\n\n"
+   formatted += "---\n\n"
+   
+   # Episodes
+   for idx, ep in enumerate(episodes):
+       ep_num = ep.get('episode_number', 0)
+       content = ep.get('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"
+       
+       if idx < len(episodes) - 1:  # Not last episode
+           formatted += "➡️ *다음 화에 계속...*\n\n"
+       
+       formatted += "---\n\n"
+   
+   return formatted
 
 # --- Gradio interface ---
 def create_interface():
-    with gr.Blocks(theme=gr.themes.Soft(), title="K-WebNovel Generator") as interface:
-        gr.HTML("""
-        <style>
-            .main-header {
-                text-align: center;
-                margin-bottom: 2rem;
-            }
-            
-            .header-title {
-                font-size: 3rem;
-                margin-bottom: 1rem;
-            }
-            
-            .header-subtitle {
-                font-size: 1.2rem;
-                margin-bottom: 0.5rem;
-            }
-            
-            .header-description {
-                margin-bottom: 1.5rem;
-            }
-            
-            .badges-container {
-                display: flex;
-                justify-content: center;
-                align-items: center;
-                gap: 8px;
-                flex-wrap: wrap;
-                margin-top: 12px;
-            }
-            
-            .badges-container a img {
-                height: 28px;
-                transition: transform 0.2s ease;
-            }
-            
-            .badges-container a:hover img {
-                transform: scale(1.05);
-            }
-            
-            @media (max-width: 768px) {
-                .header-title {
-                    font-size: 2.5rem;
-                }
-                
-                .header-subtitle {
-                    font-size: 1.1rem;
-                }
-                
-                .badges-container {
-                    gap: 6px;
-                }
-                
-                .badges-container a img {
-                    height: 24px;
-                }
-            }
-        </style>
-        
-        <div class="main-header">
-            <h1 class="header-title">📚 K-WebNovel Generator</h1>
-            
-            <div class="badges-container">
-                <a href="https://huggingface.co/spaces/fantaxy/AGI-LEADERBOARD" target="_blank">
-                    <img src="https://img.shields.io/static/v1?label=HF&message=AGI-LEADERBOARD&color=%23d4a574&labelColor=%238b6239&logo=HUGGINGFACE&logoColor=%23faf8f5&style=for-the-badge" alt="badge">
-                </a>
-                <a href="https://huggingface.co/spaces/openfree/AGI-NOVEL" target="_blank">
-                    <img src="https://img.shields.io/static/v1?label=HF&message=AGI-NOVEL&color=%23d4a574&labelColor=%235a3e28&logo=huggingface&logoColor=%23faf8f5&style=for-the-badge" alt="badge">
-                </a>                
-                <a href="https://huggingface.co/spaces/openfree/AGI-Screenplay" target="_blank">
-                    <img src="https://img.shields.io/static/v1?label=HF&message=AGI-Screenplay&color=%23b8956f&labelColor=%23745940&logo=HUGGINGFACE&logoColor=%23faf8f5&style=for-the-badge" alt="badge">
-                </a>
-                <a href="https://huggingface.co/spaces/openfree/AGI-WebNovel" target="_blank">
-                    <img src="https://img.shields.io/static/v1?label=HF&message=AGI-WebNovel&color=%23c7a679&labelColor=%236b5036&logo=HUGGINGFACE&logoColor=%23faf8f5&style=for-the-badge" alt="badge">
-                </a>
-                <a href="https://discord.gg/openfreeai" target="_blank">
-                    <img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%23c19656&labelColor=%236d4e31&logo=discord&logoColor=white&style=for-the-badge" alt="badge">
-                </a>
-            </div>
-            
-            <p class="header-subtitle">한국형 웹소설 자동 생성 시스템</p>
-            <p class="header-description">장르별 맞춤형 40화 완결 웹소설을 생성합니다</p>
-        </div>
-        """)
-        
-        # State
-        current_session_id = gr.State(None)
-        
-        with gr.Tab("✍️ 웹소설 쓰기"):
-            with gr.Group():
-                gr.Markdown("### 🎯 웹소설 설정")
-                
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        genre_select = gr.Radio(
-                            choices=list(WEBNOVEL_GENRES.keys()),
-                            value="로맨스",
-                            label="장르 선택",
-                            info="원하는 장르를 선택하세요"
-                        )
-                        
-                        query_input = gr.Textbox(
-                            label="스토리 테마",
-                            placeholder="웹소설의 기본 설정이나 주제를 입력하세요...",
-                            lines=3
-                        )
-                        
-                        with gr.Row():
-                            random_btn = gr.Button("🎲 랜덤 테마", variant="secondary")
-                            submit_btn = gr.Button("📝 연재 시작", variant="primary", size="lg")
-                    
-                    with gr.Column(scale=1):
-                        language_select = gr.Radio(
-                            choices=["Korean", "English"],
-                            value="Korean",
-                            label="언어"
-                        )
-                        
-                        gr.Markdown("""
-                        **장르별 특징:**
-                        - 로맨스: 달달한 사랑 이야기
-                        - 로판: 회귀/빙의 판타지
-                        - 판타지: 성장과 모험
-                        - 현판: 현대 배경 능력자
-                        - 무협: 무공과 강호
-                        - 미스터리: 추리와 반전
-                        - 라노벨: 가벼운 일상물
-                        """)
-                
-                status_text = gr.Textbox(
-                    label="진행 상황",
-                    interactive=False,
-                    value="장르를 선택하고 테마를 입력하세요"
-                )
-            
-            # Output
-            with gr.Row():
-                with gr.Column():
-                    episodes_display = gr.Markdown("*연재 진행 상황이 여기에 표시됩니다*")
-                
-                with gr.Column():
-                    novel_display = gr.Markdown("*완성된 웹소설이 여기에 표시됩니다*")
-                    
-                    with gr.Row():
-                        download_format = gr.Radio(
-                            choices=["TXT", "DOCX"],
-                            value="TXT",
-                            label="다운로드 형식"
-                        )
-                        download_btn = gr.Button("📥 다운로드", variant="secondary")
-                    
-                    download_file = gr.File(visible=False)
-        
-        with gr.Tab("📚 테마 라이브러리"):
-            gr.Markdown("### 인기 웹소설 테마")
-            
-            library_genre = gr.Radio(
-                choices=["전체"] + list(WEBNOVEL_GENRES.keys()),
-                value="전체",
-                label="장르 필터"
-            )
-            
-            theme_library = gr.HTML("<p>테마 라이브러리 로딩 중...</p>")
-            
-            refresh_library_btn = gr.Button("🔄 새로고침")
-        
-        # Event handlers
-        def process_query(query, genre, language, session_id):
-            system = WebNovelSystem()
-            episodes = ""
-            novel = ""
-            
-            for ep_display, novel_display, status, new_session_id in system.process_webnovel_stream(query, genre, language, session_id):
-                episodes = ep_display
-                novel = novel_display
-                yield episodes, novel, status, new_session_id
-        
-        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,
-            inputs=[query_input, genre_select, language_select, current_session_id],
-            outputs=[episodes_display, novel_display, status_text, current_session_id]
-        )
-        
-        random_btn.click(
-            fn=handle_random_theme,
-            inputs=[genre_select, language_select],
-            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=[
-                ["계약결혼한 재벌 3세와 평범한 회사원의 로맨스", "로맨스"],
-                ["회귀한 천재 마법사의 복수극", "로판"],
-                ["F급 헌터에서 SSS급 각성자가 되는 이야기", "현판"],
-                ["폐급에서 천하제일이 되는 무공 천재", "무협"],
-                ["평범한 고등학생이 이세계 용사가 되는 이야기", "라이트노벨"]
-            ],
-            inputs=[query_input, genre_select]
-        )
-    
-    return interface
+   with gr.Blocks(theme=gr.themes.Soft(), title="K-WebNovel Generator") as interface:
+       gr.HTML("""
+       <style>
+           .main-header {
+               text-align: center;
+               margin-bottom: 2rem;
+           }
+           
+           .header-title {
+               font-size: 3rem;
+               margin-bottom: 1rem;
+           }
+           
+           .header-subtitle {
+               font-size: 1.2rem;
+               margin-bottom: 0.5rem;
+           }
+           
+           .header-description {
+               margin-bottom: 1.5rem;
+           }
+           
+           .badges-container {
+               display: flex;
+               justify-content: center;
+               align-items: center;
+               gap: 8px;
+               flex-wrap: wrap;
+               margin-top: 12px;
+           }
+           
+           .badges-container a img {
+               height: 28px;
+               transition: transform 0.2s ease;
+           }
+           
+           .badges-container a:hover img {
+               transform: scale(1.05);
+           }
+           
+           @media (max-width: 768px) {
+               .header-title {
+                   font-size: 2.5rem;
+               }
+               
+               .header-subtitle {
+                   font-size: 1.1rem;
+               }
+               
+               .badges-container {
+                   gap: 6px;
+               }
+               
+               .badges-container a img {
+                   height: 24px;
+               }
+           }
+       </style>
+       
+       <div class="main-header">
+           <h1 class="header-title">📚 K-WebNovel Generator</h1>
+           
+           <div class="badges-container">
+               <a href="https://huggingface.co/spaces/fantaxy/AGI-LEADERBOARD" target="_blank">
+                   <img src="https://img.shields.io/static/v1?label=HF&message=AGI-LEADERBOARD&color=%23d4a574&labelColor=%238b6239&logo=HUGGINGFACE&logoColor=%23faf8f5&style=for-the-badge" alt="badge">
+               </a>
+               <a href="https://huggingface.co/spaces/openfree/AGI-NOVEL" target="_blank">
+                   <img src="https://img.shields.io/static/v1?label=HF&message=AGI-NOVEL&color=%23d4a574&labelColor=%235a3e28&logo=huggingface&logoColor=%23faf8f5&style=for-the-badge" alt="badge">
+               </a>                
+               <a href="https://huggingface.co/spaces/openfree/AGI-Screenplay" target="_blank">
+                   <img src="https://img.shields.io/static/v1?label=HF&message=AGI-Screenplay&color=%23b8956f&labelColor=%23745940&logo=HUGGINGFACE&logoColor=%23faf8f5&style=for-the-badge" alt="badge">
+               </a>
+               <a href="https://huggingface.co/spaces/openfree/AGI-WebNovel" target="_blank">
+                   <img src="https://img.shields.io/static/v1?label=HF&message=AGI-WebNovel&color=%23c7a679&labelColor=%236b5036&logo=HUGGINGFACE&logoColor=%23faf8f5&style=for-the-badge" alt="badge">
+               </a>
+               <a href="https://discord.gg/openfreeai" target="_blank">
+                   <img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%23c19656&labelColor=%236d4e31&logo=discord&logoColor=white&style=for-the-badge" alt="badge">
+               </a>
+           </div>
+           
+           <p class="header-subtitle">한국형 웹소설 자동 생성 시스템</p>
+           <p class="header-description">장르별 맞춤형 40화 완결 웹소설을 생성합니다</p>
+       </div>
+       """)
+       
+       # State
+       current_session_id = gr.State(None)
+       
+       with gr.Tab("✍️ 웹소설 쓰기"):
+           with gr.Group():
+               gr.Markdown("### 🎯 웹소설 설정")
+               
+               with gr.Row():
+                   with gr.Column(scale=2):
+                       genre_select = gr.Radio(
+                           choices=list(WEBNOVEL_GENRES.keys()),
+                           value="로맨스",
+                           label="장르 선택",
+                           info="원하는 장르를 선택하세요"
+                       )
+                       
+                       query_input = gr.Textbox(
+                           label="스토리 테마",
+                           placeholder="웹소설의 기본 설정이나 주제를 입력하세요...",
+                           lines=3
+                       )
+                       
+                       with gr.Row():
+                           random_btn = gr.Button("🎲 랜덤 테마", variant="secondary")
+                           submit_btn = gr.Button("📝 연재 시작", variant="primary", size="lg")
+                   
+                   with gr.Column(scale=1):
+                       language_select = gr.Radio(
+                           choices=["Korean", "English"],
+                           value="Korean",
+                           label="언어"
+                       )
+                       
+                       gr.Markdown("""
+                       **장르별 특징:**
+                       - 로맨스: 달달한 사랑 이야기
+                       - 로판: 회귀/빙의 판타지
+                       - 판타지: 성장과 모험
+                       - 현판: 현대 배경 능력자
+                       - 무협: 무공과 강호
+                       - 미스터리: 추리와 반전
+                       - 라노벨: 가벼운 일상물
+                       """)
+               
+               status_text = gr.Textbox(
+                   label="진행 상황",
+                   interactive=False,
+                   value="장르를 선택하고 테마를 입력하세요"
+               )
+           
+           # Output
+           with gr.Row():
+               with gr.Column():
+                   episodes_display = gr.Markdown("*연재 진행 상황이 여기에 표시됩니다*")
+               
+               with gr.Column():
+                   novel_display = gr.Markdown("*완성된 웹소설이 여기에 표시됩니다*")
+                   
+                   with gr.Row():
+                       download_format = gr.Radio(
+                           choices=["TXT", "DOCX"],
+                           value="TXT",
+                           label="다운로드 형식"
+                       )
+                       download_btn = gr.Button("📥 다운로드", variant="secondary")
+                   
+                   download_file = gr.File(visible=False)
+       
+       with gr.Tab("📚 테마 라이브러리"):
+           gr.Markdown("### 인기 웹소설 테마")
+           
+           library_genre = gr.Radio(
+               choices=["전체"] + list(WEBNOVEL_GENRES.keys()),
+               value="전체",
+               label="장르 필터"
+           )
+           
+           theme_library = gr.HTML("<p>테마 라이브러리 로딩 중...</p>")
+           
+           refresh_library_btn = gr.Button("🔄 새로고침")
+       
+       # Event handlers
+       def process_query(query, genre, language, session_id):
+           system = WebNovelSystem()
+           episodes = ""
+           novel = ""
+           
+           for ep_display, novel_display, status, new_session_id in system.process_webnovel_stream(query, genre, language, session_id):
+               episodes = ep_display
+               novel = novel_display
+               yield episodes, novel, status, new_session_id
+       
+       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,
+           inputs=[query_input, genre_select, language_select, current_session_id],
+           outputs=[episodes_display, novel_display, status_text, current_session_id]
+       )
+       
+       random_btn.click(
+           fn=handle_random_theme,
+           inputs=[genre_select, language_select],
+           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=[
+               ["계약결혼한 재벌 3세와 평범한 회사원의 로맨스", "로맨스"],
+               ["회귀한 천재 마법사의 복수극", "로판"],
+               ["F급 헌터에서 SSS급 각성자가 되는 이야기", "현판"],
+               ["폐급에서 천하제일이 되는 무공 천재", "무협"],
+               ["평범한 고등학생이 이세계 용사가 되는 이야기", "라이트노벨"]
+           ],
+           inputs=[query_input, genre_select]
+       )
+   
+   return interface
 
 # Main
 if __name__ == "__main__":
-    logger.info("K-WebNovel Generator Starting...")
-    logger.info("=" * 60)
-    
-    # Environment check
-    logger.info(f"API Endpoint: {API_URL}")
-    logger.info(f"Target: {TARGET_EPISODES} episodes, {TARGET_WORDS:,} words")
-    logger.info("Genres: " + ", ".join(WEBNOVEL_GENRES.keys()))
-    
-    logger.info("=" * 60)
-    
-    # Initialize database
-    logger.info("Initializing database...")
-    WebNovelDatabase.init_db()
-    logger.info("Database ready.")
-    
-    # Launch interface
-    interface = create_interface()
-    interface.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False
-    )
+   logger.info("K-WebNovel Generator Starting...")
+   logger.info("=" * 60)
+   
+   # Environment check
+   logger.info(f"API Endpoint: {API_URL}")
+   logger.info(f"Target: {TARGET_EPISODES} episodes, {TARGET_WORDS:,} words")
+   logger.info("Genres: " + ", ".join(WEBNOVEL_GENRES.keys()))
+   
+   logger.info("=" * 60)
+   
+   # Initialize database
+   logger.info("Initializing database...")
+   WebNovelDatabase.init_db()
+   logger.info("Database ready.")
+   
+   # Launch interface
+   interface = create_interface()
+   interface.launch(
+       server_name="0.0.0.0",
+       server_port=7860,
+       share=False
+   )