Update app.py

app.py

@@ -276,12 +276,51 @@ class LiteraryQualityEnhancer:
             "couldn't breathe"
         ]
         
-    def enhance_prompt(self, base_prompt: str, role: str, chapter: int) -> str:
+    def enhance_prompt(self, base_prompt: str, role: str, chapter: int, language: str = "English") -> str:
         """문학적 품질 향상을 위한 프롬프트 강화"""
         
-        literary_guidelines = """
+        if language == "Korean":
+            literary_guidelines = """
+\n=== 문학적 품질 요구사항 ===
+
+**반드시 한국어로 작성하세요. 영어 사용은 절대 금지입니다.**
+
+1. **감각적 몰입**
+   - 오감을 모두 활용한 묘사
+   - 예시: "폭발음이 컸다" 대신
+   → "폭발이 공기를 찢으며 고막을 압박했고, 매캐한 연기가 콧속을 찔렀으며 
+      뜨거운 파편이 피부를 스쳤다"
+
+2. **감정적 깊이**
+   - 신체 반응과 내적 독백을 통한 감정 표현
+   - "그는 슬펐다"와 같은 직접적 감정 서술 금지
+   - 몸짓, 생각, 기억 활용
+
+3. **고유한 목소리**
+   - 각 인물마다 독특한 말투 사용
+   - 대화 속 숨겨진 의미(subtext) 활용
+   - 자연스러운 중단, 머뭇거림, 방언
+
+4. **리듬과 속도**
+   - 문장 길이를 다양하게 변화
+   - 긴장감: 짧고 날카로운 문장
+   - 성찰: 길고 흐르는 문장
+   - 강조를 위한 단락 구분
+
+5. **클리셰 금지**
+   절대 사용 금지: {cliches}
+
+6. **상징과 모티프**
+   - 주제와 연결된 반복 이미지/사물
+   - 자연 요소와 감정 상태 연결
+   - 과하지 않게 의미 층위 쌓기
+""".format(cliches=", ".join(self.cliche_patterns[:3]))
+        else:
+            literary_guidelines = """
 \n=== LITERARY QUALITY REQUIREMENTS ===
 
+**You must write in English.**
+
 1. **SENSORY IMMERSION**
    - Use all five senses in descriptions
    - Example: Instead of "The explosion was loud"
@@ -314,7 +353,15 @@ class LiteraryQualityEnhancer:
 """.format(cliches=", ".join(self.cliche_patterns[:3]))
         
         if role.startswith("writer"):
-            literary_guidelines += """
+            if language == "Korean":
+                literary_guidelines += """
+\n7. **챕터별 집중 사항**
+   - 시작: 감각적 디테일이나 흥미로운 행동으로 독자 사로잡기
+   - 중간: 최소 하나의 의미 있는 캐릭터 순간 발전시키기
+   - 끝: 진부한 클리프행어 없이 다음 챕터로의 추진력 생성
+"""
+            else:
+                literary_guidelines += """
 \n7. **CHAPTER-SPECIFIC FOCUS**
    - Opening: Hook with sensory detail or intriguing action
    - Middle: Develop at least one meaningful character moment
@@ -1086,10 +1133,43 @@ Creatively incorporate the information from search results naturally into the no
         """Enhanced Director AI initial prompt with deeper plot structure"""
         base_prompt = self.create_director_initial_prompt(user_query, language)
         
-        plot_structure_guide = """
+        if language == "Korean":
+            plot_structure_guide = """
+
+=== 강화된 플롯 구조 설계 ===
+
+**모든 내용을 한국어로 작성하세요.**
+
+1. **캐릭터 아크를 포함한 3막 구조**
+   - 1막 (챕터 1-3): 설정 확립 + 내적 갈등 도입
+   - 2막 전반부 (챕터 4-6): 외적 갈등 상승 + 내적 변화 시작
+   - 2막 후반부 (챕터 7-8): 위기의 정점 + 가치관 전환
+   - 3막 (챕터 9-10): 클라이맥스 + 새로운 균형
+
+2. **캐릭터 변화 궤적**
+   모든 주요 캐릭터에 대한 상세한 아크 생성:
+   | 캐릭터 | 시작점 | 전환점 | 도착점 |
+   |--------|--------|--------|--------|
+   | [이름] | [신념/결핍] | [시련/깨달음] | [성장/해결] |
+
+3. **주제 레이어링**
+   - 표면 주제: [이야기가 보이는 주제]
+   - 심층 주제: [이야기의 진짜 주제]
+   - 개인 주제: [각 캐릭터의 내적 여정]
+
+4. **긴장 설계**
+   세 가지 유형의 긴장 곡선 계획:
+   - 물리적 긴장: 전투와 위험
+   - 감정적 긴장: 관계와 선택
+   - 도덕적 긴장: 가치관 충돌
+"""
+        else:
+            plot_structure_guide = """
 
 === ENHANCED PLOT STRUCTURE DESIGN ===
 
+**Write all content in English.**
+
 1. **Three-Act Structure with Character Arcs**
    - Act 1 (Chapters 1-3): Setting establishment + Internal conflict introduction
    - Act 2 First Half (Chapters 4-6): External conflict rising + Internal change beginning
@@ -1114,24 +1194,15 @@ Creatively incorporate the information from search results naturally into the no
    - Moral Tension: Value conflicts
 """
         
-        if language == "Korean":
-            return base_prompt + plot_structure_guide.replace(
-                "Three-Act Structure", "3막 구조"
-            ).replace(
-                "Character Transformation", "캐릭터 변화"
-            ).replace(
-                "Theme Layering", "주제 레이어링"
-            ).replace(
-                "Tension Design", "긴장감 설계"
-            )
-        else:
-            return base_prompt + plot_structure_guide
+        return base_prompt + plot_structure_guide
     
     def create_director_initial_prompt(self, user_query: str, language: str = "English") -> str:
         """Director AI initial prompt - Novel planning for 10 writers"""
         if language == "Korean":
             return f"""당신은 30페이지 분량의 중편 소설을 기획하는 문학 감독자입니다.
 
+**언어 설정: 모든 내용을 한국어로 작성하세요.**
+
 사용자 요청: {user_query}
 
 다음 요소들을 체계적으로 구성하여 30페이지 중편 소설의 기초를 만드세요:
@@ -1167,6 +1238,8 @@ Creatively incorporate the information from search results naturally into the no
         else:
             return f"""You are a literary director planning a 30-page novella.
 
+**Language Setting: Write all content in English.**
+
 User Request: {user_query}
 
 Systematically compose the following elements to create the foundation for a 30-page novella:
@@ -1204,10 +1277,38 @@ Provide clear guidelines for each writer to compose 3 pages."""
         """Enhanced critic's review of director's plan"""
         base_prompt = self.create_critic_director_prompt(director_plan, language)
         
-        enhanced_criteria = """
+        if language == "Korean":
+            enhanced_criteria = """
+
+=== 강화된 비평 기준 ===
+
+**모든 비평을 한국어로 작성하세요.**
+
+**문학적 품질 평가:**
+1. 독창성 점수 (1-10): 이 컨셉이 얼마나 신선하고 독특한가?
+2. 감정적 공명 (1-10): 독자가 감정적으로 연결될 수 있는가?
+3. 주제적 깊이 (1-10): 주제가 겹겹이 쌓여 있고 의미가 있는가?
+4. 캐릭터 복잡성 (1-10): 캐릭터가 다차원적인가?
+
+**구조적 분석:**
+- 각 챕터가 전체 아크에서 명확한 목적을 가지고 있는가?
+- 페이싱 문제가 있을 수 있는가?
+- 클라이맥스가 효과적으로 배치되었는가?
+
+**실행 가능성 체크:**
+- 10명의 다른 작가가 일관성을 유지할 수 있는가?
+- 캐릭터 목소리가 충분히 구별되는가?
+- 여러 작가를 위한 세계관이 충분히 명확한가?
+
+각 비평 포인트에 대해 구체적인 예시를 제공하세요.
+"""
+        else:
+            enhanced_criteria = """
 
 === ENHANCED CRITIQUE CRITERIA ===
 
+**Write all critique in English.**
+
 **Literary Quality Assessment:**
 1. Originality Score (1-10): How fresh and unique is this concept?
 2. Emotional Resonance (1-10): Will readers connect emotionally?
@@ -1234,6 +1335,8 @@ Provide specific examples for each critique point.
         if language == "Korean":
             return f"""당신은 문학 비평가입니다. 감독자의 소설 기획을 검토하고 개선점을 제시하세요.
 
+**언어 설정: 모든 내용을 한국어로 작성하세요.**
+
 감독자의 기획:
 {director_plan}
 
@@ -1267,6 +1370,8 @@ Provide specific examples for each critique point.
         else:
             return f"""You are a literary critic. Review the director's novel plan and suggest improvements.
 
+**Language Setting: Write all content in English.**
+
 Director's Plan:
 {director_plan}
 
@@ -1303,6 +1408,8 @@ Provide specific and constructive feedback."""
         if language == "Korean":
             return f"""감독자로서 비평가의 피드백을 반영하여 소설 기획을 수정합니다.
 
+**언어 설정: 모든 내용을 한국어로 작성하세요.**
+
 초기 기획:
 {initial_plan}
 
@@ -1341,6 +1448,8 @@ Provide specific and constructive feedback."""
         else:
             return f"""As director, revise the novel plan reflecting the critic's feedback.
 
+**Language Setting: Write all content in English.**
+
 Initial Plan:
 {initial_plan}
 
@@ -1385,15 +1494,26 @@ Create a final masterplan that all 10 writers can clearly understand."""
         # Add character context
         character_context = self.state_tracker.get_character_context(writer_number)
         
-        # Add literary enhancement
-        literary_guidelines = self.literary_enhancer.enhance_prompt(base_prompt, f"writer{writer_number}", writer_number)
+        # Add literary enhancement with language parameter
+        literary_guidelines = self.literary_enhancer.enhance_prompt(base_prompt, f"writer{writer_number}", writer_number, language)
         
         # Combine all elements
         enhanced_prompt = base_prompt + character_context + literary_guidelines
         
         if writer_number > 1:
             # Add consistency reminder
-            consistency_reminder = """
+            if language == "Korean":
+                consistency_reminder = """
+
+=== 일관성 요구사항 ===
+이전 챕터와의 일관성을 반드시 유지하세요:
+- 캐릭터 상태 확인 (생사, 부상, 위치)
+- 확립된 기술과 설정 유지
+- 이전 챕터의 시간선 따르기
+- 캐릭터 관계와 감정 상태 존중
+"""
+            else:
+                consistency_reminder = """
 
 === CONSISTENCY REQUIREMENTS ===
 You MUST maintain consistency with previous chapters:
@@ -1414,6 +1534,8 @@ You MUST maintain consistency with previous chapters:
         if language == "Korean":
             return f"""당신은 작성자 {writer_number}번입니다. 30페이지 중편 소설의 {pages_start}-{pages_end}페이지(3페이지)를 작성하세요.
 
+**언어 설정: 반드시 한국어로 작성하세요. 영어 사용은 절대 금지입니다.**
+
 감독자의 마스터플랜:
 {director_plan}
 
@@ -1445,10 +1567,12 @@ You MUST maintain consistency with previous chapters:
    - 독자의 관심 유지
 
 **작성 시작:**
-이제 1,400-1,500단어 분량의 소설을 작성하세요. 페이지 구분 표시는 하지 마세요."""
+이제 1,400-1,500단어 분량의 소설을 한국어로 작성하세요. 페이지 구분 표시는 하지 마세요."""
         else:
             return f"""You are Writer #{writer_number}. Write pages {pages_start}-{pages_end} (3 pages) of the 30-page novella.
 
+**Language Setting: You must write in English.**
+
 Director's Masterplan:
 {director_plan}
 
@@ -1480,7 +1604,7 @@ Director's Masterplan:
    - Maintain reader interest
 
 **BEGIN WRITING:**
-Now write your 1,400-1,500 word section. Do not use any page markers."""
+Now write your 1,400-1,500 word section in English. Do not use any page markers."""
     
     def create_enhanced_critic_writer_prompt(self, writer_number: int, writer_content: str, 
                                            director_plan: str, all_previous_content: str, 
@@ -1495,15 +1619,38 @@ Now write your 1,400-1,500 word section. Do not use any page markers."""
         # Get quality scores
         quality_scores = self.quality_scorer.score_content(writer_content, self.state_tracker, writer_number)
         
-        consistency_section = f"""
+        if language == "Korean":
+            consistency_section = f"""
+
+=== 일관성 검증 결과 ===
+{'일관성 오류가 발견되지 않았습니다.' if not consistency_errors else '치명적 오류 발견:'}
+"""
+            for error in consistency_errors:
+                consistency_section += f"- {error}\n"
+            
+            quality_section = f"""
+
+=== 품질 평가 ===
+문학적 품질: {quality_scores['literary_quality']:.1f}/10
+일관성: {quality_scores['consistency']:.1f}/10
+감정적 깊이: {quality_scores['emotional_depth']:.1f}/10
+독창성: {quality_scores['originality']:.1f}/10
+페이싱: {quality_scores['pacing']:.1f}/10
+
+전체 점수: {sum(quality_scores.values())/len(quality_scores):.1f}/10
+
+**모든 피드백을 한국어로 제공하세요.**
+"""
+        else:
+            consistency_section = f"""
 
 === CONSISTENCY VALIDATION RESULTS ===
 {'No consistency errors detected.' if not consistency_errors else 'CRITICAL ERRORS FOUND:'}
 """
-        for error in consistency_errors:
-            consistency_section += f"- {error}\n"
-        
-        quality_section = f"""
+            for error in consistency_errors:
+                consistency_section += f"- {error}\n"
+            
+            quality_section = f"""
 
 === QUALITY ASSESSMENT ===
 Literary Quality: {quality_scores['literary_quality']:.1f}/10
@@ -1513,6 +1660,8 @@ Originality: {quality_scores['originality']:.1f}/10
 Pacing: {quality_scores['pacing']:.1f}/10
 
 Overall Score: {sum(quality_scores.values())/len(quality_scores):.1f}/10
+
+**Provide all feedback in English.**
 """
         
         return base_prompt + consistency_section + quality_section
@@ -1522,6 +1671,8 @@ Overall Score: {sum(quality_scores.values())/len(quality_scores):.1f}/10
         if language == "Korean":
             return f"""작성자 {writer_number}번의 작품을 비평합니다.
 
+**언어 설정: 모든 비평을 한국어로 작성하세요.**
+
 감독자의 마스터플랜:
 {director_plan}
 
@@ -1564,6 +1715,8 @@ Overall Score: {sum(quality_scores.values())/len(quality_scores):.1f}/10
         else:
             return f"""Critiquing Writer #{writer_number}'s work.
 
+**Language Setting: Write all critique in English.**
+
 Director's Masterplan:
 {director_plan}
 
@@ -1609,6 +1762,8 @@ Clearly distinguish between mandatory revisions and optional improvements."""
         if language == "Korean":
             return f"""작성자 {writer_number}번으로서 비평가의 피드백을 반영하여 수정합니다.
 
+**언어 설정: 반드시 한국어로 작성하세요. 영어 사용은 절대 금지입니다.**
+
 초기 작성 내용:
 {initial_content}
 
@@ -1635,11 +1790,13 @@ Clearly distinguish between mandatory revisions and optional improvements."""
    - 이전/이후 내용과의 자연스러운 연결
    - 수정으로 인한 새로운 모순 방지
 
-수정된 최종본을 제시하세요. 페이지 마크는 절대 사용하지 마세요.
+수정된 최종본을 한국어로 제시하세요. 페이지 마크는 절대 사용하지 마세요.
 반드시 1,400-1,500단어 분량을 유지하세요."""
         else:
             return f"""As Writer #{writer_number}, revise based on critic's feedback.
 
+**Language Setting: You must write in English.**
+
 Initial Content:
 {initial_content}
 
@@ -1666,7 +1823,7 @@ Write a revision reflecting:
    - Natural connection with previous/next content
    - Prevent new contradictions from revisions
 
-Present the revised final version. Never use page markers.
+Present the revised final version in English. Never use page markers.
 You MUST maintain 1,400-1,500 words."""
     
     def create_test_writer_remaining_prompt(self, director_plan: str, writer1_content: str, language: str) -> str:
@@ -1674,6 +1831,8 @@ You MUST maintain 1,400-1,500 words."""
         if language == "Korean":
             return f"""[테스트 모드] 당신은 나머지 9개 챕터(Chapter 2-10)를 작성하는 특별 작가입니다.
 
+**언어 설정: 반드시 한국어로 작성하세요. 영어 사용은 절대 금지입니다.**
+
 감독자의 마스터플랜:
 {director_plan}
 
@@ -1686,24 +1845,27 @@ You MUST maintain 1,400-1,500 words."""
 3. 총 12,600-13,500 단어 (9개 챕터)
 4. Chapter 1과 자연스럽게 이어지도록 작성하세요
 5. 마스터플랜의 모든 요소를 포함하여 완결된 이야기를 만드세요
+6. 모든 내용을 한국어로 작성하세요
 
 **필수 형식:**
 반드시 아래와 같이 챕터를 명확히 구분하세요:
 
 [Chapter 2]
-(1,400-1,500 단어의 내용)
+(1,400-1,500 단어의 한국어 내용)
 
 [Chapter 3]
-(1,400-1,500 단어의 내용)
+(1,400-1,500 단어의 한국어 내용)
 
 ...이런 식으로 [Chapter 10]까지...
 
 각 챕터는 반드시 [Chapter 숫자] 형식으로 시작해야 합니다.
 챕터 사이에는 빈 줄을 넣어 구분하세요.
-Chapter 2부터 10까지 작성하세요."""
+Chapter 2부터 10까지 한국어로 작성하세요."""
         else:
             return f"""[TEST MODE] You are a special writer creating the remaining 9 chapters (Chapters 2-10).
 
+**Language Setting: You must write in English.**
+
 Director's Masterplan:
 {director_plan}
 
@@ -1716,21 +1878,22 @@ Writer 1 has already written Chapter 1:
 3. Total 12,600-13,500 words (9 chapters)
 4. Continue naturally from Chapter 1
 5. Include all elements from the masterplan to create a complete story
+6. Write all content in English
 
 **MANDATORY FORMAT:**
 You MUST clearly separate chapters as follows:
 
 [Chapter 2]
-(1,400-1,500 words of content)
+(1,400-1,500 words of content in English)
 
 [Chapter 3]
-(1,400-1,500 words of content)
+(1,400-1,500 words of content in English)
 
 ...continue this way until [Chapter 10]...
 
 Each chapter MUST start with [Chapter number] format.
 Leave blank lines between chapters for separation.
-Write Chapters 2-10 now."""
+Write Chapters 2-10 in English now."""
     
     def simulate_streaming(self, text: str, role: str) -> Generator[str, None, None]:
         """Simulate streaming in test mode"""
@@ -1780,12 +1943,18 @@ Write Chapters 2-10 now."""
             # Enhanced system prompt for writers
             if role.startswith("writer"):
                 if language == "Korean":
-                    system_prompts[role] += "\n\n**절대적 요구사항**: 당신은 반드시 1,400-1,500단어를 작성해야 합니다. 이것은 협상 불가능한 요구사항입니다."
+                    system_prompts[role] += "\n\n**절대적 요구사항**: 당신은 반드시 1,400-1,500단어를 작성해야 합니다. 이것은 협상 불가능한 요구사항입니다. 반드시 한국어로 작성하세요."
                 else:
                     system_prompts[role] += "\n\n**ABSOLUTE REQUIREMENT**: You MUST write 1,400-1,500 words. This is non-negotiable."
             
+            # Add language enforcement to system prompt
+            if language == "Korean":
+                language_enforcement = "\n\n**중요**: 모든 응답은 반드시 한국어로 작성하세요. 영어 사용은 절대 금지입니다."
+            else:
+                language_enforcement = "\n\n**IMPORTANT**: All responses must be written in English."
+            
             full_messages = [
-                {"role": "system", "content": system_prompts.get(role, "")},
+                {"role": "system", "content": system_prompts.get(role, "") + language_enforcement},
                 *messages
             ]
             
@@ -1803,6 +1972,12 @@ Write Chapters 2-10 now."""
                 temperature = 0.6
                 top_p = 0.9
             
+            # Add language hint to first user message if needed
+            if messages and messages[0]["role"] == "user" and language == "Korean":
+                # Prepend language instruction to first user message
+                original_content = messages[0]["content"]
+                messages[0]["content"] = "**반드시 한국어로 응답하세요.**\n\n" + original_content
+            
             payload = {
                 "model": self.model_id,
                 "messages": full_messages,
@@ -1813,7 +1988,18 @@ Write Chapters 2-10 now."""
                 "stream_options": {"include_usage": True}
             }
             
-            logger.info(f"API streaming call started - Role: {role}, Max tokens: {max_tokens}")
+            # Some models support language parameter
+            if language == "Korean":
+                payload["language"] = "ko"
+            else:
+                payload["language"] = "en"
+            
+            logger.info(f"API streaming call started - Role: {role}, Max tokens: {max_tokens}, Temperature: {temperature}, Language: {language}")
+            
+            # Log the first part of system prompt for debugging
+            if full_messages and full_messages[0]["role"] == "system":
+                system_preview = full_messages[0]["content"][:200] + "..." if len(full_messages[0]["content"]) > 200 else full_messages[0]["content"]
+                logger.debug(f"System prompt preview: {system_preview}")
             
             response = requests.post(
                 self.api_url,
@@ -2291,6 +2477,14 @@ Five years ago, her younger sister Seojin had died in a car accident. The memory
                 # Get appropriate prompt based on stage
                 prompt = self.get_stage_prompt(stage_idx, role, query, language, stages, test_quick_mode)
                 
+                # Log prompt language for debugging
+                if "한국어로" in prompt or "Korean" in prompt[:500]:
+                    logger.info(f"Prompt contains Korean language instruction")
+                elif "English" in prompt[:500]:
+                    logger.info(f"Prompt contains English language instruction")
+                else:
+                    logger.warning(f"Prompt may not have clear language instruction")
+                
                 # Create stage info for web search and validation
                 stage_info = {
                     'stage_idx': stage_idx,
@@ -2455,6 +2649,8 @@ Five years ago, her younger sister Seojin had died in a car accident. The memory
     def get_stage_prompt(self, stage_idx: int, role: str, query: str, 
                         language: str, stages: List[Dict], test_mode: bool = False) -> str:
         """Get appropriate prompt for each stage with enhancements"""
+        logger.info(f"Getting prompt for stage {stage_idx}, role: {role}, language: {language}")
+        
         # Stage 0: Director Initial
         if stage_idx == 0:
             return self.create_enhanced_director_initial_prompt(query, language)
@@ -2487,11 +2683,13 @@ Five years ago, her younger sister Seojin had died in a car accident. The memory
             if "초안" in stages[stage_idx]["name"] or "Draft" in stages[stage_idx]["name"]:
                 # Get accumulated content from DB
                 accumulated_content = NovelDatabase.get_all_writer_content(self.current_session_id)
+                logger.info(f"Creating writer {writer_num} draft prompt in {language}")
                 return self.create_enhanced_writer_prompt(writer_num, final_plan, accumulated_content, language)
             else:  # Revision
                 # Find the initial draft and critic feedback
                 initial_draft_idx = stage_idx - 2
                 critic_feedback_idx = stage_idx - 1
+                logger.info(f"Creating writer {writer_num} revision prompt in {language}")
                 return self.create_writer_revision_prompt(
                     writer_num, 
                     stages[initial_draft_idx]["content"],