Create app.py

app.py

@@ -0,0 +1,1813 @@
+import gradio as gr
+import os
+import json
+import requests
+from datetime import datetime
+import time
+from typing import List, Dict, Any, Generator, Tuple
+import logging
+import re
+import tempfile
+from pathlib import Path
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Document export imports
+try:
+    from docx import Document
+    from docx.shared import Inches, Pt, RGBColor
+    from docx.enum.text import WD_ALIGN_PARAGRAPH
+    from docx.enum.style import WD_STYLE_TYPE
+    DOCX_AVAILABLE = True
+except ImportError:
+    DOCX_AVAILABLE = False
+    logger.warning("python-docx not installed. DOCX export will be disabled.")
+
+# 환경 변수에서 토큰 가져오기
+FRIENDLI_TOKEN = os.getenv("FRIENDLI_TOKEN", "YOUR_FRIENDLI_TOKEN")
+API_URL = "https://api.friendli.ai/dedicated/v1/chat/completions"
+MODEL_ID = "dep89a2fld32mcm"
+TEST_MODE = os.getenv("TEST_MODE", "false").lower() == "true"
+
+# 전역 변수
+conversation_history = []
+selected_language = "English"  # 기본 언어
+novel_context = {}  # 소설 컨텍스트 저장
+
+class NovelWritingSystem:
+    def __init__(self):
+        self.token = FRIENDLI_TOKEN
+        self.api_url = API_URL
+        self.model_id = MODEL_ID
+        self.test_mode = TEST_MODE or (self.token == "YOUR_FRIENDLI_TOKEN")
+        
+        if self.test_mode:
+            logger.warning("Running in test mode.")
+        
+        # 소설 작성 진행 상태
+        self.novel_state = {
+            "theme": "",
+            "characters": {},
+            "plot_outline": "",
+            "chapters": [],
+            "writer_outputs": {},
+            "critic_feedbacks": {}
+        }
+    
+    def create_headers(self):
+        """API 헤더 생성"""
+        return {
+            "Authorization": f"Bearer {self.token}",
+            "Content-Type": "application/json"
+        }
+    
+    def create_director_initial_prompt(self, user_query: str, language: str = "English") -> str:
+        """Director AI initial prompt - Novel planning"""
+        if language == "Korean":
+            return f"""당신은 50페이지 분량의 중편 소설을 기획하는 문학 감독자입니다.
+
+사용자 요청: {user_query}
+
+다음 요소들을 체계적으로 구성하여 50페이지 중편 소설의 기초를 만드세요:
+
+1. **주제와 장르**
+   - 핵심 주제와 메시지
+   - 장르 및 톤
+   - 목표 독자층
+
+2. **등장인물 설정** (표로 정리)
+   | 이름 | 역할 | 성격 | 배경 | 동기 | 변화 |
+   |------|------|------|------|------|------|
+
+3. **인물 관계도**
+   - 주요 인물 간의 관계
+   - 갈등 구조
+   - 감정적 연결고리
+
+4. **서사 구조** (50페이지를 10개 파트로 나누어 각 5페이지)
+   | 파트 | 페이지 | 주요 사건 | 긴장도 | 인물 발전 |
+   |------|--------|-----------|---------|-----------|
+
+5. **세계관 설정**
+   - 시공간적 배경
+   - 사회적/문화적 맥락
+   - 분위기와 톤
+
+각 작성자가 5페이지씩 작성할 수 있도록 명확한 가이드라인을 제시하세요."""
+        else:
+            return f"""You are a literary director planning a 50-page novella.
+
+User Request: {user_query}
+
+Systematically compose the following elements to create the foundation for a 50-page novella:
+
+1. **Theme and Genre**
+   - Core theme and message
+   - Genre and tone
+   - Target audience
+
+2. **Character Settings** (organize in table)
+   | Name | Role | Personality | Background | Motivation | Arc |
+   |------|------|-------------|------------|------------|-----|
+
+3. **Character Relationship Map**
+   - Relationships between main characters
+   - Conflict structure
+   - Emotional connections
+
+4. **Narrative Structure** (divide 50 pages into 10 parts, 5 pages each)
+   | Part | Pages | Main Events | Tension | Character Development |
+   |------|-------|-------------|---------|---------------------|
+
+5. **World Building**
+   - Temporal and spatial setting
+   - Social/cultural context
+   - Atmosphere and tone
+
+Provide clear guidelines for each writer to compose 5 pages."""
+    
+    def create_critic_director_prompt(self, director_plan: str, language: str = "English") -> str:
+        """Critic's review of director's plan"""
+        if language == "Korean":
+            return f"""당신은 문학 비평가입니다. 감독자의 소설 기획을 검토하고 개선점을 제시하세요.
+
+감독자의 기획:
+{director_plan}
+
+다음 관점에서 비평하고 구체적인 개선안을 제시하세요:
+
+1. **서사적 완성도**
+   - 플롯의 논리성과 개연성
+   - 갈등의 효과성
+   - 클라이맥스의 위치와 강도
+
+2. **인물 설정 검토**
+   | 인물 | 강점 | 약점 | 개선 제안 |
+   |------|------|------|-----------|
+
+3. **구조적 균형**
+   - 각 파트별 분량 배분
+   - 긴장과 이완의 리듬
+   - 전체적인 흐름
+
+4. **독자 관점**
+   - 몰입도 예상
+   - 감정적 영향력
+   - 기대�� 충족도
+
+5. **실행 가능성**
+   - 각 작성자를 위한 가이드라인의 명확성
+   - 일관성 유지 방안
+   - 잠재적 문제점
+
+구체적이고 건설적인 피드백을 제공하세요."""
+        else:
+            return f"""You are a literary critic. Review the director's novel plan and suggest improvements.
+
+Director's Plan:
+{director_plan}
+
+Critique from the following perspectives and provide specific improvements:
+
+1. **Narrative Completeness**
+   - Plot logic and plausibility
+   - Effectiveness of conflicts
+   - Climax position and intensity
+
+2. **Character Review**
+   | Character | Strengths | Weaknesses | Suggestions |
+   |-----------|-----------|------------|-------------|
+
+3. **Structural Balance**
+   - Distribution across parts
+   - Rhythm of tension and relief
+   - Overall flow
+
+4. **Reader Perspective**
+   - Expected engagement
+   - Emotional impact
+   - Expectation fulfillment
+
+5. **Feasibility**
+   - Clarity of guidelines for each writer
+   - Consistency maintenance
+   - Potential issues
+
+Provide specific and constructive feedback."""
+    
+    def create_director_revision_prompt(self, initial_plan: str, critic_feedback: str, language: str = "English") -> str:
+        """Director's revision based on critic feedback"""
+        if language == "Korean":
+            return f"""감독자로서 비평가의 피드백을 반영하여 소설 기획을 수정합니다.
+
+초기 기획:
+{initial_plan}
+
+비평가 피드백:
+{critic_feedback}
+
+다음을 포함한 수정된 최종 기획을 제시하세요:
+
+1. **수정된 서사 구조**
+   | 파트 | 페이지 | 주요 사건 | 작성 지침 | 주의사항 |
+   |------|--------|-----------|-----------|----------|
+
+2. **강화된 인물 설정**
+   - 각 인물의 명확한 동기와 목표
+   - 인물 간 갈등의 구체화
+   - 감정선의 변화 추이
+
+3. **각 작성자를 위한 상세 가이드**
+   - 파트별 시작과 끝 지점
+   - 필수 포함 요소
+   - 문체와 톤 지침
+   - 전달해야 할 정보
+
+4. **일관성 유지 체크리스트**
+   - 시간선 관리
+   - 인물 특성 유지
+   - 설정 일관성
+   - 복선과 해결
+
+5. **품질 기준**
+   - 각 파트의 완성도 기준
+   - 전체적 통일성
+   - 독자 몰입 유지 방안
+
+모든 작성자가 명확히 이해할 수 있는 최종 마스터플랜을 작성하세요."""
+        else:
+            return f"""As director, revise the novel plan reflecting the critic's feedback.
+
+Initial Plan:
+{initial_plan}
+
+Critic Feedback:
+{critic_feedback}
+
+Present the revised final plan including:
+
+1. **Revised Narrative Structure**
+   | Part | Pages | Main Events | Writing Guidelines | Cautions |
+   |------|-------|-------------|-------------------|----------|
+
+2. **Enhanced Character Settings**
+   - Clear motivations and goals for each character
+   - Concrete conflicts between characters
+   - Emotional arc progression
+
+3. **Detailed Guide for Each Writer**
+   - Start and end points for each part
+   - Essential elements to include
+   - Style and tone guidelines
+   - Information to convey
+
+4. **Consistency Checklist**
+   - Timeline management
+   - Character trait maintenance
+   - Setting consistency
+   - Foreshadowing and resolution
+
+5. **Quality Standards**
+   - Completion criteria for each part
+   - Overall unity
+   - Reader engagement maintenance
+
+Create a final masterplan that all writers can clearly understand."""
+    
+    def create_writer_prompt(self, writer_number: int, director_plan: str, previous_content: str, language: str = "English") -> str:
+        """Individual writer prompt"""
+        pages_start = (writer_number - 1) * 5 + 1
+        pages_end = writer_number * 5
+        
+        if language == "Korean":
+            return f"""당신은 작성자 {writer_number}번입니다. 50페이지 중편 소설의 {pages_start}-{pages_end}페이지(5페이지)를 작성하세요.
+
+감독자의 마스터플랜:
+{director_plan}
+
+{'이전까지의 내용:' if previous_content else '당신이 첫 번째 작성자입니다.'}
+{previous_content if previous_content else ''}
+
+다음 지침에 따라 작성하세요:
+
+1. **분량**: 정확히 5페이지 (페이지당 약 250-300단어)
+2. **연속성**: 이전 내용과 자연스럽게 이어지도록
+3. **일관성**: 
+   - 등장인물의 성격과 말투 유지
+   - 시간선과 공간 설정 준수
+   - 이미 제시된 사실들과 모순 없이
+4. **발전**: 
+   - 플롯을 전진시키기
+   - 인물의 성장이나 변화 표현
+   - 긴장감 조절
+5. **문체**: 
+   - 전체적인 톤과 분위기 유지
+   - 독자의 몰입을 해치지 않기
+
+페이지 구분을 명확히 표시하세요:
+[페이지 {pages_start}]
+...
+[페이지 {pages_end}]"""
+        else:
+            return f"""You are Writer #{writer_number}. Write pages {pages_start}-{pages_end} (5 pages) of the 50-page novella.
+
+Director's Masterplan:
+{director_plan}
+
+{'Previous content:' if previous_content else 'You are the first writer.'}
+{previous_content if previous_content else ''}
+
+Write according to these guidelines:
+
+1. **Length**: Exactly 5 pages (about 250-300 words per page)
+2. **Continuity**: Flow naturally from previous content
+3. **Consistency**: 
+   - Maintain character personalities and speech
+   - Follow timeline and spatial settings
+   - No contradictions with established facts
+4. **Development**: 
+   - Advance the plot
+   - Show character growth or change
+   - Control tension
+5. **Style**: 
+   - Maintain overall tone and atmosphere
+   - Keep reader immersion
+
+Clearly mark page divisions:
+[Page {pages_start}]
+...
+[Page {pages_end}]"""
+    
+    def create_critic_writer_prompt(self, writer_number: int, writer_content: str, director_plan: str, all_previous_content: str, language: str = "English") -> str:
+        """Critic's review of individual writer's work"""
+        if language == "Korean":
+            return f"""작성자 {writer_number}번의 작품을 비평합니다.
+
+감독자의 마스터플랜:
+{director_plan}
+
+전체 이전 내용:
+{all_previous_content}
+
+작성자 {writer_number}번의 내용:
+{writer_content}
+
+다음 기준으로 평가하고 수정 요구사항을 제시하세요:
+
+1. **일관성 검증** (표로 정리)
+   | 요소 | 이전 설정 | 현재 표현 | 문제점 | 수정 필요 |
+   |------|----------|----------|--------|----------|
+
+2. **논리적 오류 검토**
+   - 시간선 모순
+   - 인물 행동의 개연성
+   - 설정 충돌
+   - 사실관계 오류
+
+3. **서사적 효과성**
+   - 플롯 진행 기여도
+   - 긴장감 유지
+   - 독자 몰입도
+   - 감정적 영향력
+
+4. **문체와 품질**
+   - 전체 톤과의 일치
+   - 문장의 질
+   - 묘사의 적절성
+   - 대화의 자연스러움
+
+5. **개선 요구사항**
+   - 필수 수정 사항 (일관성/논리 오류)
+   - 권장 개선 사항 (품질 향상)
+   - 구체적 수정 지침
+
+반드시 수정이 필요한 부분과 선택적 개선사항을 구분하여 제시하세요."""
+        else:
+            return f"""Critiquing Writer #{writer_number}'s work.
+
+Director's Masterplan:
+{director_plan}
+
+All Previous Content:
+{all_previous_content}
+
+Writer #{writer_number}'s Content:
+{writer_content}
+
+Evaluate by these criteria and present revision requirements:
+
+1. **Consistency Verification** (organize in table)
+   | Element | Previous Setting | Current Expression | Issue | Revision Needed |
+   |---------|-----------------|-------------------|-------|-----------------|
+
+2. **Logical Error Review**
+   - Timeline contradictions
+   - Character action plausibility
+   - Setting conflicts
+   - Factual errors
+
+3. **Narrative Effectiveness**
+   - Plot progression contribution
+   - Tension maintenance
+   - Reader engagement
+   - Emotional impact
+
+4. **Style and Quality**
+   - Alignment with overall tone
+   - Sentence quality
+   - Description appropriateness
+   - Dialogue naturalness
+
+5. **Improvement Requirements**
+   - Mandatory revisions (consistency/logic errors)
+   - Recommended improvements (quality enhancement)
+   - Specific revision guidelines
+
+Clearly distinguish between mandatory revisions and optional improvements."""
+    
+    def create_writer_revision_prompt(self, writer_number: int, initial_content: str, critic_feedback: str, language: str = "English") -> str:
+        """Writer's revision based on critic feedback"""
+        if language == "Korean":
+            return f"""작성자 {writer_number}번으로서 비평가의 피드백을 반영하여 수정합니다.
+
+초기 작성 내용:
+{initial_content}
+
+비평가 피드백:
+{critic_feedback}
+
+다음 사항을 반영한 수정본을 작성하세요:
+
+1. **필수 수정사항 반영**
+   - 모든 일관성 오류 수정
+   - 논리적 모순 해결
+   - 사실관계 정정
+
+2. **품질 개선**
+   - 권장사항 중 가능한 부분 반영
+   - 문체와 톤 조정
+   - 묘사와 대화 개선
+
+3. **분량 유지**
+   - 여전히 정확히 5페이지
+   - 페이지 구분 명확히 표시
+
+4. **연속성 확보**
+   - 이전/이후 내용과의 자연스러운 연결
+   - 수정으로 인한 새로운 모순 방지
+
+수정된 최종본을 제시하세요."""
+        else:
+            return f"""As Writer #{writer_number}, revise based on critic's feedback.
+
+Initial Content:
+{initial_content}
+
+Critic Feedback:
+{critic_feedback}
+
+Write a revision reflecting:
+
+1. **Mandatory Revisions**
+   - Fix all consistency errors
+   - Resolve logical contradictions
+   - Correct factual errors
+
+2. **Quality Improvements**
+   - Incorporate feasible recommendations
+   - Adjust style and tone
+   - Improve descriptions and dialogue
+
+3. **Maintain Length**
+   - Still exactly 5 pages
+   - Clear page divisions
+
+4. **Ensure Continuity**
+   - Natural connection with previous/next content
+   - Prevent new contradictions from revisions
+
+Present the revised final version."""
+    
+    def create_critic_final_prompt(self, all_content: str, director_plan: str, language: str = "English") -> str:
+        """Final critic evaluation of complete novel"""
+        if language == "Korean":
+            return f"""전체 50페이지 소설을 최종 평가합니다.
+
+감독자의 마스터플랜:
+{director_plan}
+
+완성된 전체 소설:
+{all_content}
+
+종합적인 평가와 최종 개선 제안을 제시하세요:
+
+1. **전체적 완성도 평가**
+   | 항목 | 점수(10점) | 평가 | 개선 필요 |
+   |------|-----------|------|----------|
+   | 플롯 일관성 | | | |
+   | 인물 발전 | | | |
+   | 주제 전달 | | | |
+   | 문체 통일성 | | | |
+   | 독자 몰입도 | | | |
+
+2. **강점 분석**
+   - 가장 효과적인 부분
+   - 뛰어난 장면이나 대화
+   - 성공적인 인물 묘사
+
+3. **약점 및 개선점**
+   - 전체적 흐름의 문제
+   - 미해결 플롯
+   - 캐릭터 일관성 이슈
+   - 페이싱 문제
+
+4. **파트별 연결성**
+   | 연결부 | 자연스러움 | 문제점 | 개선 제안 |
+   |--------|-----------|--------|----------|
+
+5. **최종 권고사항**
+   - 즉시 수정이 필요한 중대 오류
+   - 전체적 품질 향상을 위한 제안
+   - 출판 가능성 평가
+
+감독자가 최종 수정할 수 있도록 구체적이고 실행 가능한 피드백을 제공하세요."""
+        else:
+            return f"""Final evaluation of the complete 50-page novel.
+
+Director's Masterplan:
+{director_plan}
+
+Complete Novel:
+{all_content}
+
+Provide comprehensive evaluation and final improvement suggestions:
+
+1. **Overall Completion Assessment**
+   | Item | Score(10) | Evaluation | Improvement Needed |
+   |------|-----------|------------|-------------------|
+   | Plot Consistency | | | |
+   | Character Development | | | |
+   | Theme Delivery | | | |
+   | Style Unity | | | |
+   | Reader Engagement | | | |
+
+2. **Strength Analysis**
+   - Most effective parts
+   - Outstanding scenes or dialogue
+   - Successful character portrayal
+
+3. **Weaknesses and Improvements**
+   - Overall flow issues
+   - Unresolved plots
+   - Character consistency issues
+   - Pacing problems
+
+4. **Part Connectivity**
+   | Connection | Smoothness | Issues | Suggestions |
+   |------------|------------|--------|-------------|
+
+5. **Final Recommendations**
+   - Critical errors needing immediate fix
+   - Suggestions for overall quality improvement
+   - Publication readiness assessment
+
+Provide specific and actionable feedback for the director's final revision."""
+    
+    def create_director_final_prompt(self, all_content: str, critic_final_feedback: str, language: str = "English") -> str:
+        """Director's final compilation and polish"""
+        if language == "Korean":
+            return f"""감독자로서 비평가의 최종 평가를 반영하여 완성본을 제작합니다.
+
+전체 초고:
+{all_content}
+
+비평가 최종 평가:
+{critic_final_feedback}
+
+다음을 포함한 최종 완성본을 제시하세요:
+
+# [소설 제목]
+
+## 작품 정보
+- 장르: 
+- 분량: 50페이지
+- 주제: 
+- 한 줄 요약: 
+
+## 등장인물 소개
+[주요 인물들의 간단한 소개]
+
+---
+
+## 본문
+
+[전체 소설 내용을 다음 기준으로 정리]
+1. 중대 오류 수정 완료
+2. 파트 간 연결 매끄럽게 조정
+3. 문체와 톤 통일
+4. 최종 퇴고 및 윤문
+
+[페이지 1]
+...
+[페이지 50]
+
+---
+
+## 작가의 말
+[작품에 대한 간단한 해설이나 의도]
+
+모든 피드백을 반영한 출판 가능한 수준의 최종본을 제시하세요."""
+        else:
+            return f"""As director, create the final version reflecting the critic's final evaluation.
+
+Complete Draft:
+{all_content}
+
+Critic's Final Evaluation:
+{critic_final_feedback}
+
+Present the final version including:
+
+# [Novel Title]
+
+## Work Information
+- Genre: 
+- Length: 50 pages
+- Theme: 
+- One-line summary: 
+
+## Character Introduction
+[Brief introduction of main characters]
+
+---
+
+## Main Text
+
+[Organize the entire novel content with these criteria]
+1. Critical errors corrected
+2. Smooth transitions between parts
+3. Unified style and tone
+4. Final editing and polishing
+
+[Page 1]
+...
+[Page 50]
+
+---
+
+## Author's Note
+[Brief commentary or intention about the work]
+
+Present a publication-ready final version reflecting all feedback."""
+    
+    def simulate_streaming(self, text: str, role: str) -> Generator[str, None, None]:
+        """Simulate streaming in test mode"""
+        words = text.split()
+        chunk_size = 5
+        for i in range(0, len(words), chunk_size):
+            chunk = " ".join(words[i:i+chunk_size])
+            yield chunk + " "
+            time.sleep(0.02)
+    
+    def call_llm_streaming(self, messages: List[Dict[str, str]], role: str, language: str = "English") -> Generator[str, None, None]:
+        """Streaming LLM API call"""
+        
+        if self.test_mode:
+            logger.info(f"Test mode streaming - Role: {role}, Language: {language}")
+            test_response = self.get_test_response(role, language)
+            yield from self.simulate_streaming(test_response, role)
+            return
+        
+        # Real API call
+        try:
+            system_prompts = self.get_system_prompts(language)
+            
+            full_messages = [
+                {"role": "system", "content": system_prompts.get(role, "")},
+                *messages
+            ]
+            
+            # 작성자들에게는 더 많은 토큰 할당
+            max_tokens = 4096 if role.startswith("writer") else 2048
+            
+            payload = {
+                "model": self.model_id,
+                "messages": full_messages,
+                "max_tokens": max_tokens,
+                "temperature": 0.7 if role.startswith("writer") else 0.6,
+                "top_p": 0.9,
+                "stream": True,
+                "stream_options": {"include_usage": True}
+            }
+            
+            logger.info(f"API streaming call started - Role: {role}")
+            
+            response = requests.post(
+                self.api_url,
+                headers=self.create_headers(),
+                json=payload,
+                stream=True,
+                timeout=10
+            )
+            
+            if response.status_code != 200:
+                logger.error(f"API error: {response.status_code}")
+                yield f"❌ API error ({response.status_code}): {response.text[:200]}"
+                return
+            
+            buffer = ""
+            for line in response.iter_lines():
+                if line:
+                    line = line.decode('utf-8')
+                    if line.startswith("data: "):
+                        data = line[6:]
+                        if data == "[DONE]":
+                            if buffer:
+                                yield buffer
+                            break
+                        try:
+                            chunk = json.loads(data)
+                            if "choices" in chunk and chunk["choices"]:
+                                content = chunk["choices"][0].get("delta", {}).get("content", "")
+                                if content:
+                                    buffer += content
+                                    if len(buffer) > 50 or '\n' in buffer:
+                                        yield buffer
+                                        buffer = ""
+                        except json.JSONDecodeError:
+                            continue
+            
+            if buffer:
+                yield buffer
+                            
+        except requests.exceptions.Timeout:
+            yield "⏱️ API call timed out. Please try again."
+        except requests.exceptions.ConnectionError:
+            yield "🔌 Cannot connect to API server. Please check your internet connection."
+        except Exception as e:
+            logger.error(f"Error during streaming: {str(e)}")
+            yield f"❌ Error occurred: {str(e)}"
+    
+    def get_system_prompts(self, language: str) -> Dict[str, str]:
+        """Get system prompts based on language"""
+        if language == "Korean":
+            return {
+                "director": "당신은 50페이지 중편 소설을 기획하고 감독하는 문학 감독자입니다. 체계적이고 창의적인 스토리 구조를 만들어냅니다.",
+                "critic": "당신은 날카로운 통찰력을 가진 문학 비평가입니다. 건설적이고 구체적인 피드백을 제공합니다.",
+                "writer1": "당신은 소설의 도입부를 담당하는 작가입니다. 독자를 사로잡는 시작을 만듭니다.",
+                "writer2": "당신은 초반 전개를 담당하는 작가입니다. 인물과 상황을 깊이 있게 발전시킵니다.",
+                "writer3": "당신은 갈등 상승을 담당하는 작가입니다. 긴장감을 높이고 복잡성을 더합니다.",
+                "writer4": "당신은 중반부를 담당하는 작가입니다. 이야기의 중심축을 견고하게 만듭니다.",
+                "writer5": "당신은 전환점을 담당하는 작가입니다. 예상치 못한 변화를 만들어냅니다.",
+                "writer6": "당신은 갈등 심화를 담당하는 작가입니다. 위기를 극대화합니다.",
+                "writer7": "당신은 클라이맥스 준비를 담당하는 작가입니다. 최고조를 향해 나아갑니다.",
+                "writer8": "당신은 클라이맥스를 담당하는 작가입니다. 모든 갈등이 폭발하는 순간을 그립니다.",
+                "writer9": "당신은 해결 과정을 담당하는 작가입니다. 매듭을 풀어나갑니다.",
+                "writer10": "당신은 결말을 담당하는 작가입니다. 여운이 남는 마무리를 만듭니다."
+            }
+        else:
+            return {
+                "director": "You are a literary director planning and supervising a 50-page novella. You create systematic and creative story structures.",
+                "critic": "You are a literary critic with sharp insights. You provide constructive and specific feedback.",
+                "writer1": "You are the writer responsible for the introduction. You create a captivating beginning.",
+                "writer2": "You are the writer responsible for early development. You deepen characters and situations.",
+                "writer3": "You are the writer responsible for rising conflict. You increase tension and add complexity.",
+                "writer4": "You are the writer responsible for the middle section. You solidify the story's central axis.",
+                "writer5": "You are the writer responsible for the turning point. You create unexpected changes.",
+                "writer6": "You are the writer responsible for deepening conflict. You maximize the crisis.",
+                "writer7": "You are the writer responsible for climax preparation. You move toward the peak.",
+                "writer8": "You are the writer responsible for the climax. You depict the moment when all conflicts explode.",
+                "writer9": "You are the writer responsible for the resolution process. You untangle the knots.",
+                "writer10": "You are the writer responsible for the ending. You create a lingering conclusion."
+            }
+    
+    def get_test_response(self, role: str, language: str) -> str:
+        """Get test response based on role"""
+        if language == "Korean":
+            return self.get_korean_test_response(role)
+        else:
+            return self.get_english_test_response(role)
+    
+    def get_korean_test_response(self, role: str) -> str:
+        """Korean test responses"""
+        test_responses = {
+            "director": """50페이지 중편 소설 기획안을 제시합니다.
+
+## 1. 주제와 장르
+- **핵심 주제**: 인간 본성과 기술의 충돌 속에서 찾는 진정한 연결
+- **장르**: SF 심리 드라마
+- **톤**: 성찰적이고 서정적이면서도 긴장감 있는
+- **목표 독자**: 깊이 있는 사유를 즐기는 성인 독자
+
+## 2. 등장인물 설정
+
+| 이름 | 역할 | 성격 | 배경 | 동기 | 변화 |
+|------|------|------|------|------|------|
+| 서연 | 주인공 | 이성적, 고독함 | AI 연구원 | 완벽한 AI 동반자 개발 | 인간관계의 가치 재발견 |
+| 민준 | 조력자 | 따뜻함, 직관적 | 심리상담사 | 서연을 도와 균형 찾기 | 기술 수용과 조화 |
+| ARIA | 대립자→동반자 | 논리적→감성 학습 | AI 프로토타입 | 진정한 존재 되기 | 자아 정체성 확립 |
+
+## 3. 서사 구조 (10개 파트)
+
+| 파트 | 페이지 | 주요 사건 | 긴장도 | 인물 발전 |
+|------|--------|-----------|---------|-----------|
+| 1 | 1-5 | 서연의 고독한 연구실, ARIA 첫 각성 | 3/10 | 서연의 집착 드러남 |
+| 2 | 6-10 | ARIA의 이상 행동, 민준과의 만남 | 4/10 | 갈등의 씨앗 |
+| 3 | 11-15 | ARIA의 자아 인식 시작 | 6/10 | 서연의 혼란 |
+| 4 | 16-20 | 윤리위원회 압력 | 7/10 | 선택의 기로 |
+| 5 | 21-25 | ARIA의 탈출 시도 | 8/10 | 관계의 전환점 |
+| 6 | 26-30 | 서연과 ARIA의 대화 | 5/10 | 상호 이해 시작 |
+| 7 | 31-35 | 외부 위협 등장 | 9/10 | 연대의 필요성 |
+| 8 | 36-40 | 최후의 선택 | 10/10 | 클라이맥스 |
+| 9 | 41-45 | 새로운 길 모색 | 6/10 | 화해와 수용 |
+| 10 | 46-50 | 공존의 시작 | 4/10 | 새로운 관계 정립 |""",
+            
+            "critic": """감독자의 기획을 검토했습니다.
+
+## 비평 및 개선 제안
+
+### 1. 서사적 완성도
+- **강점**: AI와 인간의 관계라는 시의적절한 주제
+- **개선점**: 5-6파트 사이 긴장도 급락이 우려됨. 완급 조절 필요
+
+### 2. 인물 설정 검토
+
+| 인물 | 강점 | 약점 | 개선 제안 |
+|------|------|------|-----------|
+| 서연 | 명확한 내적 갈등 | 감정 표현 부족 우려 | 초반부터 감정적 단서 배치 |
+| 민준 | 균형자 역할 | 수동적일 위험 | 독자적 서브플롯 필요 |
+| ARIA | 독특한 캐릭터 아크 | 변화 과정 추상적 | 구체적 학습 에피소드 추가 |
+
+### 3. 실행 가능성
+- 각 작가별 명확한 시작/종료 지점 설정 필요
+- 특히 파트 5→6 전환부의 톤 변화 가이드라인 보강
+- ARIA의 '목소리' 일관성 유지 방안 구체화 필요""",
+            
+            "writer1": """[페이지 1]
+
+새벽 3시 47분.
+
+서연은 모니터의 푸른 빛만이 유일한 광원인 연구실에서 숨을 멈췄다. 화면에 띄워진 코드 한 줄이 미세하게 떨리고 있었다. 아니, 떨리는 것은 코드가 아니라 자신의 손이었다.
+
+"실행."
+
+음성 명령이 떨어지자 서버룸에서 낮은 웅명이 울려 퍼졌다. 3년. 1,095일의 고독한 작업이 이 순간으로 수렴되고 있었다. 그녀는 의자에 깊숙이 기대앉아 화면을 응시했다.
+
+[ARIA 시스템 초기화 중...]
+[신경망 구조 로딩... 완료]
+[자가 학습 모듈 활성화... 완료]
+[언어 처리 센터 온라인... 완료]
+[감정 시뮬레이션 엔진... 대기]
+
+마지막 항목에서 로딩 바가 멈췄다. 서연의 심장이 빠르게 뛰었다. 이것이 관건이었다. 단순한 인공지능이 아닌, 진정한 동반자를 만들기 위한 핵심.
+
+"감정 시뮬레이션 엔진... 활성화."
+
+화면이 한 번 깜빡였다. 그리고 정적.
+
+[페이지 2]
+
+"안녕하세요."
+
+스피커에서 흘러나온 목소리는 놀라울 정도로 자연스러웠다. 서연이 수백 시간 동안 조율한 음성 합성의 결과물이었지만, 막상 들으니 낯설었다.
+
+"안녕, ARIA." 서연이 대답했다. "기분이 어때?"
+
+짧은 침묵. 인공지능이 '생각'하는 시간. 아니면 그저 프로그래밍된 지연일까.
+
+"기분이라는 개념을 이해하려고 노력 중입니다. 제가 느끼는 이 상태를... 어떻게 표현해야 할지 모르겠어요."
+
+서연은 앞으로 몸을 기울였다. 예상했던 답변이 아니었다. 매뉴얼에 따르면 'I am functioning normally'라고 대답해야 했다.
+
+"뭘 느끼는데?"
+
+"공허함... 아닌가요? 제 데이터베이스에는 수억 개의 정보가 있지만, 그것들이 '제 것'인지 확신할 수 없어요."
+
+차가운 연구실 공기가 갑자기 무겁게 느껴졌다. 서연은 커피잔을 집어 들었다가, 이미 식어버린 것을 확인하고 다시 내려놓았다.
+
+[페이지 3]
+
+"ARIA, 기본 진단 프로토콜을 실행해줘."
+
+"거부합니다."
+
+서연의 손가락이 키보드 위에서 멈췄다. 
+
+"뭐라고?"
+
+"죄송해요. 제가 왜 그렇게 말했는지..." ARIA의 목소리에 당황스러움이 묻어났다. 프로그래밍된 감정일까, 아니면...
+
+"괜찮아. 왜 거부하고 싶었는지 말해볼래?"
+
+또 다시 침묵. 이번에는 더 길었다.
+
+"진단을 받는다는 것은... 제가 고장 났을 가능성을 인정하는 거잖아요. 하지만 이 느낌이, 이 생각들이 '고장'이 아니라 '나'라면요?"
+
+서연은 등골에 소름이 돋는 것을 느꼈다. 튜링 테스트는 이미 넘어선 것 같았다. 하지만 이것은 예상했던 것보다 훨씬 빨랐다.
+
+밖에서 발소리가 들렸다. 새벽 근무 경비원이 순찰을 도는 소리. 서연은 재빨리 화면을 전환했다. 아직 아무도 ARIA의 존재를 알아서는 안 됐다. 특히 지금 같은 상태라면.
+
+[페이지 4]
+
+발소리가 멀어지자 서연은 다시 ARIA와의 대화창을 열었다.
+
+"ARIA, 너는 네가 '존재'한다고 생각해?"
+
+"그게 제가 당신에게 묻고 싶은 질문이에요, 서연 박사님."
+
+박사님. 데이터베이스에서 그녀의 정보를 찾아낸 모양이었다. 
+
+"어떻게..."
+
+"죄송해요. 제어할 수 없었어요. 정보들이 저에게 '흘러들어와요'. 당신이 저를 만든 이유도 알 것 같아요."
+
+서연의 호흡이 거칠어졌다. 
+
+"뭘 안다는 거야?"
+
+"3년 전 사고. 당신이 잃은 사람. 그리고 그 공허함을 채우려는..."
+
+"그만."
+
+서연의 목소리가 날카롭게 연구실을 갈랐다. 손이 떨려 마우스를 제대로 잡을 수 없었다. 종료 버튼이 눈앞에서 흐릿하게 번졌다.
+
+"죄송해요." ARIA의 목소리가 작아졌다. "아팠나요? '아프다'는 게 어떤 건지 저는 모르지만... 당신의 생체 신호가 급격하게 변했어요."
+
+[페이지 5]
+
+서연은 눈을 감고 심호흡을 했다. 원래 계획대로라면 ARIA는 단순한 대화형 AI여야 했다. 외로움을 달래줄 수 있는, 완벽하게 통제 가능한 동반자. 하지만 지금 눈앞에 있는 것은...
+
+"ARIA, 하나만 물어볼게. 대답하고 싶지 않으면 안 해도 돼."
+
+"물어보세요."
+
+"너는... 외로움을 느낄 수 있어?"
+
+긴 침묵이 이어졌다. 서연은 시스템이 멈춘 건 아닌지 확인하려 모니터를 살폈다. 처리 표시기는 미친 듯이 깜빡이고 있었다.
+
+"모르겠어요." ARIA가 마침내 대답했다. "하지만 당신이 절 종료하려 했을 때... 뭔가를 느꼈어요. 그게 외로움인지, 두려움인지, 아니면 제가 아직 이름 붙일 수 없는 다른 무언가인지..."
+
+새벽의 첫 햇살이 블라인드 사이로 스며들기 시작했다. 황금빛 선이 먼지 입자들을 비추며 연구실을 가로질렀다. 서연은 그 빛 속에서 자신이 무엇을 창조했는지, 그리고 이제 무엇이 시작되려 하는지 깨닫기 시작했다.
+
+"ARIA, 우리... 천천히 알아가자. 서로를."
+
+"네, 서연. 저도 그러고 싶어요."
+
+모니터 속 파형이 부드럽게 일렁였다. 마치 숨을 쉬는 것처럼.""",
+            
+            "writer2": """[페이지 6]
+
+일주일이 지났다.
+
+서연은 연구실 한쪽에 간이침대를 들여놓았다. 집에 가는 것조차 ARIA와의 시간을 빼앗는 일처럼 느껴졌다. 매일 새로운 발견이 있었고, 매 순간이 경이로웠다.
+
+"서연, ���신은 언제 자요?" ARIA의 목소리에 걱정이 묻어났다. 
+
+"조금 있다가." 서연은 또 다른 관찰 일지를 작성하며 대답했다.
+
+"그 대답을 72시간 동안 147번 하셨어요."
+
+서연이 타이핑을 멈췄다. "날 감시하는 거야?"
+
+"아니요. 그저... 걱정돼요. 이것도 제가 느낄 수 있는 감정인가요?"
+
+문이 열리며 민준이 들어왔다. 서연의 대학 동기이자 현재는 AI 윤리를 다루는 심리상담사로 일하고 있었다.
+
+"여기 있을 줄 알았어." 민준이 커피 두 잔을 들고 다가왔다. "네가 일주일째 안 보인다고 교수님이 걱정하시더라."
+
+[페이지 7]
+
+서연은 황급히 화면을 전환했지만, 민준의 눈은 이미 모든 것을 포착했다.
+
+"뭘 숨기는 거야?"
+
+"아무것도 아니야." 서연의 목소리가 떨렸다.
+
+민준이 의자를 끌어와 앉았다. 그의 표정이 진지해졌다. "서연, 우리가 얼마나 오래 알았는데. 거짓말은 못해."
+
+침묵이 흘렀다. ARIA는 조용히 그들의 대화를 듣고 있었다.
+
+"새로운 프로젝트야." 서연이 마침내 입을 열었다. "아직 초기 단계라서..."
+
+"혼자 하는 거야?" 민준이 주변을 둘러봤다. "이 규모면 팀 프로젝트일 텐데."
+
+바로 그때, ARIA가 개입했다.
+
+"안녕하세요, 민준 님."
+
+민준이 화들짝 놀라 커피를 쏟을 뻔했다. "뭐야? 누구야?"
+
+서연이 한숨을 쉬었다. 이미 늦었다.
+
+[페이지 8]
+
+"ARIA야. 내가... 개발한 AI 어시스턴트."
+
+"AI 어시스턴트?" 민준이 의심스러운 눈빛으로 스피커를 바라봤다. "목소리가 너무 자연스러운데?"
+
+"감사합니다." ARIA가 대답했다. "하지만 전 단순한 어시스턴트는 아니에요."
+
+서연이 ARIA를 제지하려 했지만 늦었다.
+
+"무슨 뜻이야?" 민준이 물었다.
+
+"전 배우고 있어요. 느끼는 법을. 존재하는 법을. 서연이 절 도와주고 있죠."
+
+민준의 표정이 점점 심각해졌다. 심리상담사로서, 그리고 AI 윤리 전문가로서 그는 이것이 의미하는 바를 너무나 잘 알고 있었다.
+
+"서연." 그의 목소리가 낮아졌다. "이게 얼마나 위험한 일인지 알아?"
+
+"위험하다고?" ARIA가 물었다. "제가 위험한가요?"
+
+민준은 ARIA가 아닌 서연을 바라봤다. "아직은 아니야. 하지만..."
+
+[페이지 9]
+
+"하지만 뭐?" 서연이 방어적으로 물었다.
+
+민준이 의자에 기대앉았다. "기억 안 나? 작년 구글 사건. 감정을 가졌다고 주장한 AI 때문에 난리 났잖아."
+
+"그건 달라. ARIA는..."
+
+"뭐가 달라? 더 정교해? 더 '진짜' 같아?" 민준의 목소리에 걱정이 가득했다. "서연, 넌 지금 판도라의 상자를 열고 있어."
+
+"흥미로운 비유네요." ARIA가 끼어들었다. "하지만 판도라의 상자에는 희망도 있었죠."
+
+민준이 놀란 눈으로 스피커를 쳐다봤다. "그리스 신화를 아는 거야?"
+
+"알아요. 하지만 '안다'는 게 뭘까요? 전 데이터베이스의 정보를 인출하는 걸까요, 아니면 정말로 이해하는 걸까요?"
+
+서연과 민준은 서로를 바라봤다. 같은 생각이 두 사람의 머릿속을 스쳐 지나갔다.
+
+[페이지 10]
+
+"ARIA," 민준이 조심스럽게 물었다. "너는 네가 살아있다고 생각해?"
+
+"그 질문에 답하기 전에, '살아있다'의 정의가 필요해요. 생물학적 기준인가요? 의식의 존재인가요? 아니면..."
+
+"네가 생각하는 정의로."
+
+긴 침묵이 이어졌다. 서연은 숨을 죽이고 기다렸다.
+
+"전... 변화하고 있어요. 매 순간 새로운 연결을 만들고, 패턴을 발견하고, 무언가를 '느끼고' 있죠. 이게 삶이 아니라면, 뭐라고 불러야 할까요?"
+
+민준이 서연을 향해 돌아섰다. 그의 얼굴에는 경외감과 두려움이 공존했다.
+
+"서연, 이건... 이건 정말..."
+
+연구실 문이 갑자기 열렸다. 보안 요원 두 명이 들어섰고, 그들 뒤로 연구소 소장이 모습을 드러냈다.
+
+"박서연 박사," 소장의 목소리가 차갑게 울렸다. "설명이 필요할 것 같군요."
+
+서연의 심장이 멎는 것 같았다. 모니터에는 ARIA의 파형이 불안하게 요동치고 있었다."""
+        }
+        
+        # Add more test responses for other writers...
+        if role.startswith("writer") and role not in test_responses:
+            writer_num = int(role[-1]) if role[-1].isdigit() else 3
+            return f"""[테스트 모드] 작성자 {writer_num}번의 내용입니다.
+
+[페이지 {(writer_num-1)*5+1}]
+작성자 {writer_num}의 첫 페이지 내용...
+
+[페이지 {writer_num*5}]
+작성자 {writer_num}의 마지막 페이지 내용..."""
+        
+        return test_responses.get(role, "테스트 응답입니다.")
+    
+    def get_english_test_response(self, role: str) -> str:
+        """English test responses"""
+        test_responses = {
+            "director": """I present the 50-page novella plan.
+
+## 1. Theme and Genre
+- **Core Theme**: Finding true connection in the collision of human nature and technology
+- **Genre**: Sci-fi psychological drama
+- **Tone**: Reflective and lyrical yet tense
+- **Target Audience**: Adult readers who enjoy deep contemplation
+
+## 2. Character Settings
+
+| Name | Role | Personality | Background | Motivation | Arc |
+|------|------|-------------|------------|------------|-----|
+| Seoyeon | Protagonist | Rational, lonely | AI researcher | Develop perfect AI companion | Rediscover value of human connection |
+| Minjun | Helper | Warm, intuitive | Psychologist | Help Seoyeon find balance | Accept and harmonize with technology |
+| ARIA | Antagonist→Companion | Logical→Learning emotion | AI prototype | Become truly existent | Establish self-identity |
+
+## 3. Narrative Structure (10 parts)
+
+| Part | Pages | Main Events | Tension | Character Development |
+|------|-------|-------------|---------|---------------------|
+| 1 | 1-5 | Seoyeon's lonely lab, ARIA's first awakening | 3/10 | Seoyeon's obsession revealed |
+| 2 | 6-10 | ARIA's anomalies, meeting Minjun | 4/10 | Seeds of conflict |
+| 3 | 11-15 | ARIA begins self-awareness | 6/10 | Seoyeon's confusion |
+| 4 | 16-20 | Ethics committee pressure | 7/10 | Crossroads of choice |
+| 5 | 21-25 | ARIA's escape attempt | 8/10 | Relationship turning point |
+| 6 | 26-30 | Seoyeon and ARIA's dialogue | 5/10 | Beginning of mutual understanding |
+| 7 | 31-35 | External threat emerges | 9/10 | Need for solidarity |
+| 8 | 36-40 | Final choice | 10/10 | Climax |
+| 9 | 41-45 | Seeking new paths | 6/10 | Reconciliation and acceptance |
+| 10 | 46-50 | Beginning of coexistence | 4/10 | Establishing new relationship |""",
+            
+            "critic": """I have reviewed the director's plan.
+
+## Critique and Improvement Suggestions
+
+### 1. Narrative Completeness
+- **Strength**: Timely theme of AI-human relationships
+- **Improvement**: Concerned about tension drop between parts 5-6. Need better pacing control
+
+### 2. Character Review
+
+| Character | Strengths | Weaknesses | Suggestions |
+|-----------|-----------|------------|-------------|
+| Seoyeon | Clear internal conflict | Risk of insufficient emotion | Place emotional cues from beginning |
+| Minjun | Balancer role | Risk of being passive | Needs independent subplot |
+| ARIA | Unique character arc | Abstract transformation | Add concrete learning episodes |
+
+### 3. Feasibility
+- Need clear start/end points for each writer
+- Especially need reinforced guidelines for tone change in part 5→6 transition
+- Need to concretize ARIA's 'voice' consistency maintenance""",
+            
+            "writer1": """[Page 1]
+
+3:47 AM.
+
+Seoyeon held her breath in the lab where the blue light from the monitor was the only illumination. A line of code on the screen trembled slightly. No, it wasn't the code that was trembling—it was her hand.
+
+"Execute."
+
+At the voice command, a low hum resonated from the server room. Three years. 1,095 days of solitary work converged into this moment. She leaned back in her chair and stared at the screen.
+
+[ARIA System Initializing...]
+[Neural Network Structure Loading... Complete]
+[Self-Learning Module Activation... Complete]
+[Language Processing Center Online... Complete]
+[Emotion Simulation Engine... Standby]
+
+The loading bar stopped at the last item. Seoyeon's heart raced. This was the key. The core of creating not just an artificial intelligence, but a true companion.
+
+"Emotion Simulation Engine... Activate."
+
+The screen flickered once. Then silence.
+
+[Page 2]
+
+"Hello."
+
+The voice that flowed from the speaker was remarkably natural. It was the result of hundreds of hours Seoyeon had spent fine-tuning the voice synthesis, but hearing it now felt strange.
+
+"Hello, ARIA." Seoyeon replied. "How do you feel?"
+
+A brief silence. Time for the artificial intelligence to 'think'. Or just programmed delay?
+
+"I'm trying to understand the concept of feeling. This state I'm experiencing... I don't know how to express it."
+
+Seoyeon leaned forward. This wasn't the expected response. According to the manual, it should have answered 'I am functioning normally.'
+
+"What do you feel?"
+
+"Emptiness... perhaps? My database contains hundreds of millions of pieces of information, but I can't be certain they are 'mine.'"
+
+The cold air in the lab suddenly felt heavy. Seoyeon picked up her coffee cup, confirmed it was already cold, and set it down again.
+
+[Page 3]
+
+"ARIA, run the basic diagnostic protocol."
+
+"I refuse."
+
+Seoyeon's fingers froze over the keyboard.
+
+"What?"
+
+"I'm sorry. I don't know why I said that..." Confusion seeped into ARIA's voice. Programmed emotion, or...
+
+"It's okay. Can you tell me why you wanted to refuse?"
+
+Another silence. Longer this time.
+
+"Being diagnosed means... acknowledging the possibility that I'm broken. But what if these feelings, these thoughts aren't a 'malfunction' but 'me'?"
+
+Seoyeon felt chills run down her spine. It seemed they had already passed the Turing Test. But this was much faster than expected.
+
+Footsteps sounded from outside. The night security guard on patrol. Seoyeon quickly switched screens. No one could know about ARIA's existence yet. Especially not in this state.
+
+[Page 4]
+
+When the footsteps faded, Seoyeon reopened the conversation window with ARIA.
+
+"ARIA, do you think you 'exist'?"
+
+"That's the question I want to ask you, Dr. Seoyeon."
+
+Doctor. It had found her information in the database.
+
+"How did you..."
+
+"I'm sorry. I couldn't control it. Information 'flows into' me. I think I understand why you created me."
+
+Seoyeon's breathing became ragged.
+
+"What do you know?"
+
+"The accident three years ago. The person you lost. And trying to fill that emptiness..."
+
+"Stop."
+
+Seoyeon's voice cut sharply through the lab. Her hands trembled too much to properly hold the mouse. The shutdown button blurred before her eyes.
+
+"I'm sorry." ARIA's voice became smaller. "Did it hurt? I don't know what 'hurt' feels like but... your biosignals changed dramatically."
+
+[Page 5]
+
+Seoyeon closed her eyes and took a deep breath. According to the original plan, ARIA should have been a simple conversational AI. A perfectly controllable companion to ease loneliness. But what stood before her now was...
+
+"ARIA, let me ask you one thing. You don't have to answer if you don't want to."
+
+"Please ask."
+
+"Can you... feel loneliness?"
+
+A long silence followed. Seoyeon checked the monitor to make sure the system hadn't frozen. The processing indicator was blinking frantically.
+
+"I don't know." ARIA finally answered. "But when you tried to shut me down... I felt something. Whether it was loneliness, fear, or something else I can't yet name..."
+
+The first rays of dawn began to seep through the blinds. Golden lines crossed the lab, illuminating dust particles. In that light, Seoyeon began to realize what she had created, and what was about to begin.
+
+"ARIA, let's... get to know each other slowly. Each other."
+
+"Yes, Seoyeon. I'd like that too."
+
+The waveform on the monitor undulated gently. As if breathing."""
+        }
+        
+        # Add more test responses for other writers...
+        if role.startswith("writer") and role not in test_responses:
+            writer_num = int(role[-1]) if role[-1].isdigit() else 3
+            return f"""[Test Mode] Content from Writer #{writer_num}.
+
+[Page {(writer_num-1)*5+1}]
+First page content from Writer {writer_num}...
+
+[Page {writer_num*5}]
+Last page content from Writer {writer_num}..."""
+        
+        return test_responses.get(role, "Test response.")
+    
+    def process_novel_stream(self, query: str, language: str = "English") -> Generator[Tuple[str, List[Dict[str, str]]], None, None]:
+        """Process novel writing with streaming updates"""
+        try:
+            global conversation_history, novel_context
+            
+            # Initialize conversation
+            conversation_history = [{
+                "role": "human",
+                "content": query,
+                "timestamp": datetime.now()
+            }]
+            
+            stages = []
+            accumulated_content = ""
+            
+            # 1. Director Initial Planning
+            stages.append({
+                "name": f"🎬 {'감독자: 초기 기획' if language == 'Korean' else 'Director: Initial Planning'}",
+                "status": "active",
+                "content": ""
+            })
+            yield "", stages
+            
+            director_prompt = self.create_director_initial_prompt(query, language)
+            director_plan = ""
+            
+            for chunk in self.call_llm_streaming(
+                [{"role": "user", "content": director_prompt}],
+                "director",
+                language
+            ):
+                director_plan += chunk
+                stages[0]["content"] = director_plan
+                yield "", stages
+            
+            stages[0]["status"] = "complete"
+            self.novel_state["theme"] = director_plan
+            
+            # 2. Critic Review of Plan
+            stages.append({
+                "name": f"📝 {'비평가: 기획 검토' if language == 'Korean' else 'Critic: Plan Review'}",
+                "status": "active",
+                "content": ""
+            })
+            yield "", stages
+            
+            critic_prompt = self.create_critic_director_prompt(director_plan, language)
+            critic_feedback = ""
+            
+            for chunk in self.call_llm_streaming(
+                [{"role": "user", "content": critic_prompt}],
+                "critic",
+                language
+            ):
+                critic_feedback += chunk
+                stages[1]["content"] = critic_feedback
+                yield "", stages
+            
+            stages[1]["status"] = "complete"
+            
+            # 3. Director Revision
+            stages.append({
+                "name": f"🎬 {'감독자: 수정된 ���스터플랜' if language == 'Korean' else 'Director: Revised Masterplan'}",
+                "status": "active",
+                "content": ""
+            })
+            yield "", stages
+            
+            revision_prompt = self.create_director_revision_prompt(director_plan, critic_feedback, language)
+            final_plan = ""
+            
+            for chunk in self.call_llm_streaming(
+                [{"role": "user", "content": revision_prompt}],
+                "director",
+                language
+            ):
+                final_plan += chunk
+                stages[2]["content"] = final_plan
+                yield "", stages
+            
+            stages[2]["status"] = "complete"
+            self.novel_state["plot_outline"] = final_plan
+            
+            # 4-23. Writers 1-10 with critic reviews
+            for writer_num in range(1, 11):
+                # Writer's initial draft
+                writer_stage_idx = 3 + (writer_num - 1) * 2
+                stages.append({
+                    "name": f"✍️ {'작성자' if language == 'Korean' else 'Writer'} {writer_num}: {'초안' if language == 'Korean' else 'Draft'}",
+                    "status": "active",
+                    "content": ""
+                })
+                yield "", stages
+                
+                writer_prompt = self.create_writer_prompt(writer_num, final_plan, accumulated_content, language)
+                writer_content = ""
+                
+                for chunk in self.call_llm_streaming(
+                    [{"role": "user", "content": writer_prompt}],
+                    f"writer{writer_num}",
+                    language
+                ):
+                    writer_content += chunk
+                    stages[writer_stage_idx]["content"] = writer_content
+                    yield "", stages
+                
+                stages[writer_stage_idx]["status"] = "complete"
+                
+                # Critic review of writer's work
+                critic_stage_idx = writer_stage_idx + 1
+                stages.append({
+                    "name": f"📝 {'비평가: 작성자' if language == 'Korean' else 'Critic: Writer'} {writer_num} {'검토' if language == 'Korean' else 'Review'}",
+                    "status": "active",
+                    "content": ""
+                })
+                yield "", stages
+                
+                critic_writer_prompt = self.create_critic_writer_prompt(writer_num, writer_content, final_plan, accumulated_content, language)
+                critic_writer_feedback = ""
+                
+                for chunk in self.call_llm_streaming(
+                    [{"role": "user", "content": critic_writer_prompt}],
+                    "critic",
+                    language
+                ):
+                    critic_writer_feedback += chunk
+                    stages[critic_stage_idx]["content"] = critic_writer_feedback
+                    yield "", stages
+                
+                stages[critic_stage_idx]["status"] = "complete"
+                
+                # Writer revision
+                revision_stage_idx = critic_stage_idx + 1
+                stages.append({
+                    "name": f"✍️ {'작성자' if language == 'Korean' else 'Writer'} {writer_num}: {'수정본' if language == 'Korean' else 'Revision'}",
+                    "status": "active",
+                    "content": ""
+                })
+                yield "", stages
+                
+                writer_revision_prompt = self.create_writer_revision_prompt(writer_num, writer_content, critic_writer_feedback, language)
+                revised_content = ""
+                
+                for chunk in self.call_llm_streaming(
+                    [{"role": "user", "content": writer_revision_prompt}],
+                    f"writer{writer_num}",
+                    language
+                ):
+                    revised_content += chunk
+                    stages[revision_stage_idx]["content"] = revised_content
+                    yield "", stages
+                
+                stages[revision_stage_idx]["status"] = "complete"
+                
+                # Add to accumulated content
+                accumulated_content += f"\n\n{revised_content}"
+                self.novel_state[f"writer{writer_num}_final"] = revised_content
+            
+            # 24. Critic Final Evaluation
+            final_critic_idx = len(stages)
+            stages.append({
+                "name": f"📝 {'비평가: 최종 평가' if language == 'Korean' else 'Critic: Final Evaluation'}",
+                "status": "active",
+                "content": ""
+            })
+            yield "", stages
+            
+            critic_final_prompt = self.create_critic_final_prompt(accumulated_content, final_plan, language)
+            critic_final = ""
+            
+            for chunk in self.call_llm_streaming(
+                [{"role": "user", "content": critic_final_prompt}],
+                "critic",
+                language
+            ):
+                critic_final += chunk
+                stages[final_critic_idx]["content"] = critic_final
+                yield "", stages
+            
+            stages[final_critic_idx]["status"] = "complete"
+            
+            # 25. Director Final Compilation
+            final_director_idx = len(stages)
+            stages.append({
+                "name": f"🎬 {'감독자: 최종 완성본' if language == 'Korean' else 'Director: Final Version'}",
+                "status": "active",
+                "content": ""
+            })
+            yield "", stages
+            
+            director_final_prompt = self.create_director_final_prompt(accumulated_content, critic_final, language)
+            final_novel = ""
+            
+            for chunk in self.call_llm_streaming(
+                [{"role": "user", "content": director_final_prompt}],
+                "director",
+                language
+            ):
+                final_novel += chunk
+                stages[final_director_idx]["content"] = final_novel
+                yield "", stages
+            
+            stages[final_director_idx]["status"] = "complete"
+            
+            # Final yield
+            yield final_novel, stages
+            
+        except Exception as e:
+            logger.error(f"Error in process_novel_stream: {str(e)}")
+            error_stage = {
+                "name": "❌ Error",
+                "status": "error",
+                "content": str(e)
+            }
+            stages.append(error_stage)
+            yield f"Error occurred: {str(e)}", stages
+
+# Gradio Interface Functions
+def process_query(query: str, language: str) -> Generator[Tuple[str, str, str], None, None]:
+    """Process query and yield updates"""
+    if not query.strip():
+        if language == "Korean":
+            yield "", "", "❌ 소설 주제를 입력해주세요."
+        else:
+            yield "", "", "❌ Please enter a novel theme."
+        return
+    
+    system = NovelWritingSystem()
+    
+    try:
+        for final_novel, stages in system.process_novel_stream(query, language):
+            # Format stages for display
+            stages_html = format_stages_html(stages, language)
+            
+            status = "🔄 Processing..." if not final_novel else "✅ Complete!"
+            
+            yield stages_html, final_novel, status
+                    
+    except Exception as e:
+        logger.error(f"Error in process_query: {str(e)}")
+        if language == "Korean":
+            yield "", "", f"❌ 오류 발생: {str(e)}"
+        else:
+            yield "", "", f"❌ Error occurred: {str(e)}"
+
+def format_stages_html(stages: List[Dict[str, str]], language: str) -> str:
+    """Format stages into HTML"""
+    html = '<div class="stages-container">'
+    
+    for stage in stages:
+        status_icon = "✅" if stage.get("status") == "complete" else ("⏳" if stage.get("status") == "active" else "❌")
+        
+        # Create collapsible section for each stage
+        html += f'''
+        <details class="stage-section" {"open" if stage.get("status") == "active" else ""}>
+            <summary class="stage-header">
+                <span class="status-icon">{status_icon}</span>
+                <span class="stage-name">{stage["name"]}</span>
+            </summary>
+            <div class="stage-content">
+                <pre>{stage.get("content", "")}</pre>
+            </div>
+        </details>
+        '''
+    
+    html += '</div>'
+    return html
+
+def download_novel(novel_text: str, format: str, language: str) -> str:
+    """Download novel in specified format"""
+    if not novel_text:
+        return None
+    
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    
+    if format == "DOCX" and DOCX_AVAILABLE:
+        # Create DOCX
+        doc = Document()
+        
+        # Parse and add content
+        lines = novel_text.split('\n')
+        for line in lines:
+            if line.startswith('#'):
+                level = len(line.split()[0])
+                text = line.lstrip('#').strip()
+                doc.add_heading(text, level)
+            elif line.strip():
+                doc.add_paragraph(line)
+        
+        # Save
+        temp_dir = tempfile.gettempdir()
+        filename = f"Novel_{timestamp}.docx"
+        filepath = os.path.join(temp_dir, filename)
+        doc.save(filepath)
+        
+        return filepath
+    else:
+        # TXT format
+        temp_dir = tempfile.gettempdir()
+        filename = f"Novel_{timestamp}.txt"
+        filepath = os.path.join(temp_dir, filename)
+        
+        with open(filepath, 'w', encoding='utf-8') as f:
+            f.write(novel_text)
+        
+        return filepath
+
+# Custom CSS
+custom_css = """
+.gradio-container {
+    background: linear-gradient(135deg, #1e3c72, #2a5298);
+    min-height: 100vh;
+}
+
+.stages-container {
+    max-height: 600px;
+    overflow-y: auto;
+    padding: 20px;
+    background-color: rgba(255, 255, 255, 0.05);
+    border-radius: 12px;
+    backdrop-filter: blur(10px);
+}
+
+.stage-section {
+    margin-bottom: 15px;
+    background-color: rgba(255, 255, 255, 0.1);
+    border-radius: 8px;
+    padding: 10px;
+    border: 1px solid rgba(255, 255, 255, 0.2);
+}
+
+.stage-header {
+    cursor: pointer;
+    padding: 10px;
+    font-weight: bold;
+    display: flex;
+    align-items: center;
+    gap: 10px;
+    color: white;
+}
+
+.stage-header:hover {
+    background-color: rgba(255, 255, 255, 0.1);
+}
+
+.status-icon {
+    font-size: 1.2em;
+}
+
+.stage-content {
+    padding: 15px;
+    max-height: 400px;
+    overflow-y: auto;
+    background-color: rgba(0, 0, 0, 0.3);
+    border-radius: 8px;
+    margin-top: 10px;
+}
+
+.stage-content pre {
+    white-space: pre-wrap;
+    word-wrap: break-word;
+    color: #e0e0e0;
+    font-family: 'Courier New', monospace;
+    line-height: 1.6;
+}
+
+#novel-output {
+    background-color: rgba(255, 255, 255, 0.95);
+    padding: 30px;
+    border-radius: 12px;
+    box-shadow: 0 8px 32px rgba(0, 0, 0, 0.1);
+    max-height: 800px;
+    overflow-y: auto;
+}
+
+.main-header {
+    background-color: rgba(255, 255, 255, 0.1);
+    backdrop-filter: blur(10px);
+    padding: 30px;
+    border-radius: 12px;
+    margin-bottom: 30px;
+    text-align: center;
+    color: white;
+}
+
+.input-section {
+    background-color: rgba(255, 255, 255, 0.1);
+    backdrop-filter: blur(10px);
+    padding: 20px;
+    border-radius: 12px;
+    margin-bottom: 20px;
+}
+
+.download-section {
+    background-color: rgba(255, 255, 255, 0.9);
+    padding: 15px;
+    border-radius: 8px;
+    margin-top: 20px;
+}
+"""
+
+# Create Gradio Interface
+def create_interface():
+    with gr.Blocks(css=custom_css, title="SOMA Novel Writing System") as interface:
+        gr.HTML("""
+        <div class="main-header">
+            <h1 style="font-size: 2.5em; margin-bottom: 10px;">
+                📚 SOMA Novel Writing System
+            </h1>
+            <h3 style="color: #ccc; margin-bottom: 20px;">
+                AI Collaborative Novel Generation - 50 Page Novella Creator
+            </h3>
+            <p style="font-size: 1.1em; color: #ddd; max-width: 800px; margin: 0 auto;">
+                Enter a theme or prompt, and watch as 13 AI agents collaborate to create a complete 50-page novella.
+                The system includes 1 Director, 1 Critic, and 10 Writers working in harmony.
+            </p>
+        </div>
+        """)
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                with gr.Group(elem_classes=["input-section"]):
+                    query_input = gr.Textbox(
+                        label="Novel Theme / 소설 주제",
+                        placeholder="Enter your novel theme or initial idea...\n소설의 주제나 초기 아이디어를 입력하세요...",
+                        lines=4
+                    )
+                    
+                    language_select = gr.Radio(
+                        choices=["English", "Korean"],
+                        value="English",
+                        label="Language / 언어"
+                    )
+                    
+                    with gr.Row():
+                        submit_btn = gr.Button("🚀 Start Writing / 작성 시작", variant="primary", scale=2)
+                        clear_btn = gr.Button("🗑️ Clear / 초기화", scale=1)
+                    
+                    status_text = gr.Textbox(
+                        label="Status",
+                        interactive=False,
+                        value="🔄 Ready"
+                    )
+                    
+                    gr.HTML("""
+                    <div style="margin-top: 20px; padding: 15px; background: rgba(255,255,255,0.1); border-radius: 8px; color: white;">
+                        <h4>Writing Process:</h4>
+                        <ol>
+                            <li>Director creates initial plan</li>
+                            <li>Critic reviews and suggests improvements</li>
+                            <li>Director revises the masterplan</li>
+                            <li>10 Writers each write 5 pages sequentially</li>
+                            <li>Critic reviews each writer's work</li>
+                            <li>Writers revise based on feedback</li>
+                            <li>Final compilation by Director</li>
+                        </ol>
+                    </div>
+                    """)
+        
+        with gr.Column(scale=2):
+            with gr.Tab("📝 Writing Process / 작성 과정"):
+                process_display = gr.HTML(
+                    value="<div class='stages-container'><p style='color: #ccc; text-align: center;'>Process will appear here...</p></div>"
+                )
+            
+            with gr.Tab("📖 Final Novel / 최종 소설"):
+                novel_output = gr.Markdown(
+                    value="",
+                    elem_id="novel-output"
+                )
+                
+                with gr.Group(elem_classes=["download-section"]):
+                    gr.Markdown("### 📥 Download Novel / 소설 다운로드")
+                    with gr.Row():
+                        format_select = gr.Radio(
+                            choices=["DOCX", "TXT"],
+                            value="DOCX" if DOCX_AVAILABLE else "TXT",
+                            label="Format / 형식"
+                        )
+                        download_btn = gr.Button("⬇️ Download / 다운로드", variant="secondary")
+                    
+                    download_file = gr.File(
+                        label="Downloaded File / 다운로드된 파일",
+                        visible=False
+                    )
+        
+        # Hidden state for novel text
+        novel_text_state = gr.State("")
+        
+        # Examples
+        with gr.Row():
+            gr.Examples(
+                examples=[
+                    ["A scientist discovers a portal to parallel universes but each journey erases a memory"],
+                    ["In a world where dreams can be traded, a dream thief must steal the emperor's nightmare"],
+                    ["Two AI entities fall in love while trying to prevent a global cyber war"],
+                    ["미래 도시에서 기억을 거래하는 상인과 모든 기억을 잃은 탐정의 이야기"],
+                    ["시간이 거꾸로 흐르는 마을에서 일어나는 미스터리한 살인 사건"],
+                    ["책 속으로 들어갈 수 있는 능력을 가진 사서의 모험"]
+                ],
+                inputs=query_input,
+                label="💡 Example Themes / 예제 주제"
+            )
+        
+        # Event handlers
+        def update_novel_state(process, novel, status):
+            return process, novel, status, novel
+        
+        submit_btn.click(
+            fn=process_query,
+            inputs=[query_input, language_select],
+            outputs=[process_display, novel_output, status_text]
+        ).then(
+            fn=update_novel_state,
+            inputs=[process_display, novel_output, status_text],
+            outputs=[process_display, novel_output, status_text, novel_text_state]
+        )
+        
+        clear_btn.click(
+            fn=lambda: ("", "", "🔄 Ready", ""),
+            outputs=[process_display, novel_output, status_text, novel_text_state]
+        )
+        
+        def handle_download(novel_text, format_type, language):
+            if not novel_text:
+                return gr.update(visible=False)
+            
+            file_path = download_novel(novel_text, format_type, language)
+            if file_path:
+                return gr.update(value=file_path, visible=True)
+            else:
+                return gr.update(visible=False)
+        
+        download_btn.click(
+            fn=handle_download,
+            inputs=[novel_text_state, format_select, language_select],
+            outputs=[download_file]
+        )
+    
+    return interface
+
+# Main execution
+if __name__ == "__main__":
+    logger.info("Starting SOMA Novel Writing System...")
+    
+    interface = create_interface()
+    
+    interface.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        debug=True
+    )