Update app.py

app.py

@@ -607,21 +607,34 @@ class NovelWritingSystem:
 
     def get_system_prompts(self, language: str) -> Dict[str, str]:
         """역할별 시스템 프롬프트 생성"""
-        # (기존 코드와 동일)
-        if language == "Korean":
-            return {
+        base_prompts = {
+            "Korean": {
                 "director": "당신은 창의적이고 체계적인 소설 기획 전문가입니다. 흥미롭고 일관성 있는 스토리를 설계하세요.",
                 "critic": "당신은 일관성 검토 전문 비평가입니다. 캐릭터, 플롯, 설정의 일관성을 철저히 점검하고 개선방안을 제시하세요.",
-                "writer1": "당신은 소설의 매력적인 시작을 담당하는 작가입니다. 독자를 사로잡는 도입부를 만드세요.",
-                "writer10": "당신은 완벽한 결말을 만드는 작가입니다. 독자에게 깊은 여운을 남기는 마무리를 하세요."
-            }
-        else:
-            return {
+                "writer_base": "당신은 전문 소설 작가입니다. 주어진 지침에 따라 몰입감 있고 일관성 있는 내용을 작성하세요."
+            },
+            "English": {
                 "director": "You are a creative and systematic novel planning expert. Design engaging and consistent stories.",
                 "critic": "You are a consistency review specialist critic. Thoroughly check character, plot, and setting consistency and suggest improvements.",
-                "writer1": "You are a writer responsible for the captivating beginning. Create an opening that hooks readers.",
-                "writer10": "You are a writer who creates the perfect ending. Create a conclusion that leaves readers with deep resonance."
+                "writer_base": "You are a professional novel writer. Write immersive and consistent content according to the given guidelines."
             }
+        }
+        
+        prompts = base_prompts[language].copy()
+        
+        # 작가별 특수 프롬프트
+        if language == "Korean":
+            prompts["writer1"] = "당신은 소설의 매력적인 시작을 담당하는 작가입니다. 독자를 사로잡는 도입부를 만드세요."
+            prompts["writer10"] = "당신은 완벽한 결말을 만드는 작가입니다. 독자에게 깊은 여운을 남기는 마무리를 하세요."
+        else:
+            prompts["writer1"] = "You are a writer responsible for the captivating beginning. Create an opening that hooks readers."
+            prompts["writer10"] = "You are a writer who creates the perfect ending. Create a conclusion that leaves readers with deep resonance."
+        
+        # writer2-9는 기본 프롬프트 사용
+        for i in range(2, 10):
+            prompts[f"writer{i}"] = prompts["writer_base"]
+        
+        return prompts
 
     # --- 메인 프로세스 ---
     def process_novel_stream(self, query: str, language: str, session_id: Optional[str] = None) -> Generator[Tuple[str, List[Dict[str, Any]], str], None, None]:
@@ -755,6 +768,134 @@ class NovelWritingSystem:
             return "보고서 생성 중 오류 발생"
 
 
+# --- 유틸리티 함수들 ---
+def process_query(query: str, language: str, session_id: Optional[str] = None) -> Generator[Tuple[str, str, str, str], None, None]:
+    """메인 쿼리 처리 함수"""
+    if not query.strip():
+        yield "", "", "❌ 주제를 입력해주세요.", session_id
+        return
+    
+    system = NovelWritingSystem()
+    stages_markdown = ""
+    novel_content = ""
+    
+    for status, stages, current_session_id in system.process_novel_stream(query, language, session_id):
+        stages_markdown = format_stages_display(stages)
+        
+        # 최종 소설 내용 가져오기
+        if stages and all(s.get("status") == "complete" for s in stages[-10:]):
+            novel_content = NovelDatabase.get_writer_content(current_session_id)
+            novel_content = format_novel_display(novel_content)
+        
+        yield stages_markdown, novel_content, status or "🔄 처리 중...", current_session_id
+
+def get_active_sessions(language: str) -> List[str]:
+    """활성 세션 목록 가져오기"""
+    sessions = NovelDatabase.get_active_sessions()
+    return [f"{s['session_id'][:8]}... - {s['user_query'][:50]}... ({s['created_at']})" 
+            for s in sessions]
+
+def auto_recover_session(language: str) -> Tuple[Optional[str], str]:
+    """가장 최근 활성 세션 자동 복구"""
+    latest_session = NovelDatabase.get_latest_active_session()
+    if latest_session:
+        return latest_session['session_id'], f"세션 {latest_session['session_id'][:8]}... 복구됨"
+    return None, "복구할 세션이 없습니다."
+
+def resume_session(session_id: str, language: str) -> Generator[Tuple[str, str, str, str], None, None]:
+    """세션 재개 함수"""
+    if not session_id:
+        yield "", "", "❌ 세션 ID가 없습니다.", session_id
+        return
+    
+    # 드롭다운에서 세션 ID 추출
+    if "..." in session_id:
+        session_id = session_id.split("...")[0]
+    
+    session = NovelDatabase.get_session(session_id)
+    if not session:
+        yield "", "", "❌ 세션을 찾을 수 없습니다.", None
+        return
+    
+    # process_query를 통해 재개
+    yield from process_query(session['user_query'], session['language'], session_id)
+
+def download_novel(novel_text: str, format_type: str, language: str, session_id: str) -> Optional[str]:
+    """소설 다운로드 파일 생성"""
+    if not novel_text or not session_id:
+        return None
+    
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f"novel_{session_id[:8]}_{timestamp}"
+    
+    try:
+        if format_type == "DOCX" and DOCX_AVAILABLE:
+            return export_to_docx(novel_text, filename, language)
+        else:
+            return export_to_txt(novel_text, filename)
+    except Exception as e:
+        logger.error(f"파일 생성 실패: {e}")
+        return None
+
+def format_stages_display(stages: List[Dict]) -> str:
+    """단계별 진행 상황 마크다운 포맷팅"""
+    markdown = "## 🎬 진행 상황\n\n"
+    for i, stage in enumerate(stages):
+        status_icon = "✅" if stage['status'] == 'complete' else "🔄" if stage['status'] == 'active' else "⏳"
+        markdown += f"{status_icon} **{stage['name']}**"
+        if stage.get('consistency_score', 0) > 0:
+            markdown += f" (일관성: {stage['consistency_score']:.1f}/10)"
+        markdown += "\n"
+        if stage['content']:
+            preview = stage['content'][:200] + "..." if len(stage['content']) > 200 else stage['content']
+            markdown += f"> {preview}\n\n"
+    return markdown
+
+def format_novel_display(novel_text: str) -> str:
+    """소설 내용 마크다운 포맷팅"""
+    if not novel_text:
+        return "아직 완성된 내용이 없습니다."
+    
+    # 페이지 구분 추가
+    pages = novel_text.split('\n\n')
+    formatted = "# 📖 완성된 소설\n\n"
+    
+    for i, page in enumerate(pages):
+        if page.strip():
+            formatted += f"### 페이지 {i+1}\n\n{page}\n\n---\n\n"
+    
+    return formatted
+
+def export_to_docx(content: str, filename: str, language: str) -> str:
+    """DOCX 파일로 내보내기"""
+    doc = Document()
+    
+    # 제목 추가
+    title = doc.add_heading('AI 협업 소설', 0)
+    title.alignment = WD_ALIGN_PARAGRAPH.CENTER
+    
+    # 메타데이터
+    doc.add_paragraph(f"생성일: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+    doc.add_paragraph(f"언어: {language}")
+    doc.add_page_break()
+    
+    # 본문 추가
+    paragraphs = content.split('\n\n')
+    for para in paragraphs:
+        if para.strip():
+            doc.add_paragraph(para.strip())
+    
+    # 파일 저장
+    filepath = f"{filename}.docx"
+    doc.save(filepath)
+    return filepath
+
+def export_to_txt(content: str, filename: str) -> str:
+    """TXT 파일로 내보내기"""
+    filepath = f"{filename}.txt"
+    with open(filepath, 'w', encoding='utf-8') as f:
+        f.write(content)
+    return filepath
 
 
 # CSS 스타일