feat: ?뱀냼??李쎌옉 ?€?쒕낫??諛??묒뾽怨듦컙 UI 援ы쁽

app/__init__.py

@@ -263,6 +263,9 @@ def create_app() -> Flask:
     from app.routes import main_bp
     app.register_blueprint(main_bp)
 
+    from app.routers.creation import creation_bp
+    app.register_blueprint(creation_bp)
+
     @app.context_processor
     def inject_i18n():
         lang = session.get('lang', 'vi')

app/agent/agent.py

@@ -0,0 +1,67 @@
+import asyncio
+import os
+from typing import Optional, Union
+from pydantic_ai import Agent, RunContext
+from pydantic_ai.models import Model
+from app.agent.deps import NovelWriterDeps
+from app.models.project_config import ProjectMode
+
+def get_model() -> Union[str, Model]:
+    """환경 변수에 따라 적절한 모델을 반환합니다."""
+    if os.environ.get('OPENAI_API_KEY'):
+        return 'openai:gpt-4o'
+    elif os.environ.get('GEMINI_API_KEY'):
+        return 'google-gla:gemini-1.5-flash'
+    else:
+        # API 키가 없을 경우 테스트 모델을 반환하여 서버 기동을 허용합니다.
+        from pydantic_ai.models.test import TestModel
+        return TestModel()
+
+# PydanticAI 에이전트 정의
+novel_agent = Agent(
+    get_model(),
+    deps_type=NovelWriterDeps,
+    system_prompt="당신은 전문 웹소설 작가 어시스턴트입니다."
+)
+
+@novel_agent.system_prompt
+async def dynamic_system_prompt(ctx: RunContext[NovelWriterDeps]) -> str:
+    """Fetch facts and context in parallel to build a dynamic system prompt."""
+    project = ctx.deps.current_project
+    
+    # Zep(Context)과 Mem0(Facts) 데이터를 병렬로 조회
+    context_task = ctx.deps.zep_client.get_session_context(project.project_id, "default_session")
+    facts_task = ctx.deps.mem0_client.get_facts(project.project_id)
+    
+    context, facts = await asyncio.gather(context_task, facts_task)
+    
+    facts_str = "\n- ".join(facts)
+    
+    prompt = f"""
+[현재 프로젝트: {project.title}]
+모드: {project.mode.value}
+
+[기존 설정 및 사실관계 (Mem0)]
+- {facts_str}
+
+[스토리 맥락 및 요약 (Zep)]
+{context}
+
+위 정보를 바탕으로 소설 작성을 도와주세요. 
+일관성 있는 캐릭터와 전개를 유지하는 것이 가장 중요합니다.
+"""
+    return prompt
+
+@novel_agent.tool
+async def consult_knowledge_base(ctx: RunContext[NovelWriterDeps], query: str) -> str:
+    """기존 자료나 참고 설정을 검색합니다. (REFERENCE 모드에서만 사용 가능)"""
+    project = ctx.deps.current_project
+    
+    if project.mode != ProjectMode.REFERENCE:
+        return "현재 '순수 창작(CREATIVE)' 모드이므로 외부 지식 베이스를 사용할 수 없습니다."
+    
+    if not project.reference_source_id:
+        return "참고 자료 ID가 설정되지 않았습니다."
+        
+    return await ctx.deps.rag_service.query_knowledge(query, project.reference_source_id)
+

app/agent/deps.py

@@ -0,0 +1,13 @@
+from dataclasses import dataclass
+from typing import Any
+from app.models.project_config import ProjectConfig
+
+@dataclass
+class NovelWriterDeps:
+    """Dependency injection for the Novel AI Agent."""
+    user_id: str
+    current_project: ProjectConfig
+    mem0_client: Any  # Mem0 client instance
+    zep_client: Any   # Zep client instance
+    rag_service: Any  # Interface for existing VectorDB/GraphRAG
+

app/models/project_config.py

@@ -0,0 +1,18 @@
+from enum import Enum
+from typing import Optional
+from pydantic import BaseModel, Field
+
+class ProjectMode(str, Enum):
+    CREATIVE = "CREATIVE"   # 순수 창작 모드
+    REFERENCE = "REFERENCE" # 기존 작품 참고 모드
+
+class ProjectConfig(BaseModel):
+    project_id: str = Field(..., description="Unique ID for the novel project")
+    title: str = Field(..., description="Title of the novel")
+    mode: ProjectMode = Field(default=ProjectMode.CREATIVE)
+    reference_source_id: Optional[str] = Field(None, description="Vector DB collection ID for REFERENCE mode")
+    user_id: str = Field(..., description="Owner of the project")
+
+    class Config:
+        from_attributes = True
+

app/routers/creation.py

@@ -0,0 +1,82 @@
+from flask import Blueprint, request, jsonify, render_template, current_app
+from typing import List
+from app.models.project_config import ProjectConfig, ProjectMode
+from app.agent.agent import novel_agent
+from app.agent.deps import NovelWriterDeps
+from app.services.memory_service import ZepService, Mem0Service
+from app.services.rag_service import RAGService
+import uuid
+import asyncio
+
+creation_bp = Blueprint('creation', __name__, url_prefix='/creation')
+
+# 임시 인메모리 저장소 (실제 운영 시 DB 연동 필요)
+projects_db = {}
+
+def get_deps(project_id: str, user_id: str = "user_123") -> NovelWriterDeps:
+    if project_id not in projects_db:
+        return None
+    
+    project = projects_db[project_id]
+    
+    # 설정값은 실제 환경변수 등에서 가져와야 함
+    return NovelWriterDeps(
+        user_id=user_id,
+        current_project=project,
+        mem0_client=Mem0Service(api_key="mem0_key"),
+        zep_client=ZepService(api_key="zep_key", api_url="http://zep:8000"),
+        rag_service=RAGService()
+    )
+
+@creation_bp.route('/webnovel', methods=['GET'])
+def webnovel():
+    """웹소설 대시보드 페이지"""
+    return render_template('novel_dashboard.html', title="웹소설 창작")
+
+@creation_bp.route('/workspace/<project_id>', methods=['GET'])
+def workspace(project_id: str):
+    """웹소설 창작 작업공간 페이지"""
+    if project_id not in projects_db:
+        return "Project not found", 404
+    project = projects_db[project_id]
+    return render_template('novel_workspace.html', project=project)
+
+@creation_bp.route('/api/projects', methods=['POST'])
+def create_project():
+    """새로운 웹소설 프로젝트를 생성합니다."""
+    data = request.get_json()
+    title = data.get('title')
+    mode_str = data.get('mode', 'CREATIVE')
+    reference_id = data.get('reference_id')
+    
+    project_id = str(uuid.uuid4())
+    project = ProjectConfig(
+        project_id=project_id,
+        title=title,
+        mode=ProjectMode(mode_str),
+        reference_source_id=reference_id,
+        user_id="user_123"
+    )
+    projects_db[project_id] = project
+    return jsonify(project.model_dump())
+
+@creation_bp.route('/api/projects', methods=['GET'])
+def list_projects():
+    """모든 프로젝트 목록을 가져옵니다."""
+    return jsonify([p.model_dump() for p in projects_db.values()])
+
+@creation_bp.route('/api/chat/<project_id>', methods=['POST'])
+async def chat_with_agent(project_id: str):
+    """에이전트와 채팅(집필 도움)을 진행합니다."""
+    data = request.get_json()
+    message = data.get('message')
+    
+    deps = get_deps(project_id)
+    if not deps:
+        return jsonify({"error": "Project not found"}), 404
+    
+    # PydanticAI 에이전트 실행 (async)
+    # Flask에서 async route를 지원하려면 flask[async]가 필요하거나 직접 루프를 돌려야 함
+    # 최신 Flask 3.0.0은 async def를 지원함
+    result = await novel_agent.run(message, deps=deps)
+    return jsonify({"reply": result.data})

app/routes.py

@@ -44,10 +44,19 @@ def get_default_admin_menu():
     return {
         "version": 1,
         "sections": [
+            {
+                "label": "창작",
+                "roles": ["admin", "webtoon_pm", "webtoon_admin", "producer"],
+                "items": [
+                    {"label": "웹소설", "endpoint": "creation.webnovel"},
+                ],
+            },
             {
                 "label": "사이트 관리",
                 "roles": ["admin"],
                 "items": [
+                    {"label": "설정", "endpoint": "main.admin_settings"},
+                    {"label": "RAG 관리", "endpoint": "main.webnovels"},
                     {"label": "사용자 관리", "endpoint": "main.admin", "roles": ["admin"]},
                     {"label": "토큰 통계", "endpoint": "main.admin_tokens", "roles": ["admin"]},
                     {"label": "메뉴 관리", "endpoint": "main.admin_menu", "roles": ["admin"]},

app/services/memory_service.py

@@ -0,0 +1,32 @@
+import asyncio
+from typing import List, Dict, Any, Optional
+
+class ZepService:
+    """Zep (Long-term Memory) integration."""
+    def __init__(self, api_key: str, api_url: str):
+        self.api_key = api_key
+        self.api_url = api_url
+
+    async def get_session_context(self, project_id: str, session_id: str) -> str:
+        """Fetch session history and summaries from Zep."""
+        zep_session_id = f"{project_id}_{session_id}"
+        # TODO: Implement actual Zep client call
+        await asyncio.sleep(0.1) # Simulate I/O
+        return f"Summary of project {project_id} from Zep..."
+
+class Mem0Service:
+    """Mem0 (Entity/Fact Memory) integration."""
+    def __init__(self, api_key: str):
+        self.api_key = api_key
+
+    async def get_facts(self, project_id: str) -> List[str]:
+        """Fetch character and setting facts filtered by project_id."""
+        # TODO: Implement actual Mem0 client call with metadata filter
+        await asyncio.sleep(0.1) # Simulate I/O
+        return [f"Character info for project {project_id} from Mem0"]
+
+    async def add_fact(self, project_id: str, fact: str):
+        """Store a new fact into Mem0."""
+        # TODO: Implement storage
+        pass
+

app/services/rag_service.py

@@ -0,0 +1,16 @@
+from typing import List, Dict, Any
+from app.vector_db import VectorDBManager
+
+class RAGService:
+    """Wrapper for the existing RAG (VectorDB + GraphRAG) system."""
+    def __init__(self):
+        self.vector_db = VectorDBManager()
+
+    async def query_knowledge(self, query: str, collection_id: str) -> str:
+        """Search across existing RAG knowledge base."""
+        # 기존 vector_db.search_with_rerank 또는 유사한 메서드 활용
+        # 여기서는 인터페이스 예시만 작성
+        results = self.vector_db.search_with_rerank(query, top_k=3)
+        context = "\n".join([r['content'] for r in results])
+        return context
+

requirements.txt

@@ -21,3 +21,7 @@ psycopg2-binary  # PostgreSQL support for external database
 ollama
 holidays
 
+# AI Agent Frameworks
+pydantic-ai
+zep-python
+mem0ai

templates/novel_dashboard.html

@@ -0,0 +1,132 @@
+{% extends "admin.html" %}
+
+{% block content %}
+<div class="container-fluid mt-4">
+    <div class="d-flex justify-content-between align-items-center mb-4">
+        <h2><i class="fas fa-book-open me-2"></i>웹소설 프로젝트 대시보드</h2>
+        <button class="btn btn-primary" data-bs-toggle="modal" data-bs-target="#createProjectModal">
+            <i class="fas fa-plus me-1"></i> 새 프로젝트 생성
+        </button>
+    </div>
+
+    <div class="row" id="projectList">
+        <!-- 프로젝트 카드가 여기에 동적으로 추가됩니다 -->
+        <div class="col-12 text-center py-5">
+            <div class="spinner-border text-primary" role="status">
+                <span class="visually-hidden">Loading...</span>
+            </div>
+            <p class="mt-2 text-muted">프로젝트를 불러오는 중...</p>
+        </div>
+    </div>
+</div>
+
+<!-- 프로젝트 생성 모달 -->
+<div class="modal fade" id="createProjectModal" tabindex="-1" aria-hidden="true">
+    <div class="modal-dialog">
+        <div class="modal-content">
+            <div class="modal-header">
+                <h5 class="modal-title">새 웹소설 프로젝트</h5>
+                <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
+            </div>
+            <div class="modal-body">
+                <form id="createProjectForm">
+                    <div class="mb-3">
+                        <label class="form-label">제목</label>
+                        <input type="text" class="form-control" id="projectTitle" required placeholder="소설 제목을 입력하세요">
+                    </div>
+                    <div class="mb-3">
+                        <label class="form-label">창작 모드</label>
+                        <select class="form-select" id="projectMode">
+                            <option value="CREATIVE">순수 창작 (Pure Imagination)</option>
+                            <option value="REFERENCE">기존 작품 참고 (RAG Enabled)</option>
+                        </select>
+                    </div>
+                    <div class="mb-3 d-none" id="referenceSection">
+                        <label class="form-label">참고 자료 ID (RAG Collection)</label>
+                        <input type="text" class="form-control" id="referenceId" placeholder="참고할 벡터 컬렉션 ID">
+                    </div>
+                </form>
+            </div>
+            <div class="modal-footer">
+                <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">취소</button>
+                <button type="button" class="btn btn-primary" onclick="createNewProject()">생성하기</button>
+            </div>
+        </div>
+    </div>
+</div>
+
+<script>
+document.getElementById('projectMode').addEventListener('change', function() {
+    const refSection = document.getElementById('referenceSection');
+    if (this.value === 'REFERENCE') {
+        refSection.classList.remove('d-none');
+    } else {
+        refSection.classList.add('d-none');
+    }
+});
+
+async function loadProjects() {
+    try {
+        const response = await fetch('/creation/api/projects');
+        const projects = await response.json();
+        const container = document.getElementById('projectList');
+        
+        if (projects.length === 0) {
+            container.innerHTML = `
+                <div class="col-12 text-center py-5 text-muted">
+                    <i class="fas fa-folder-open fa-3x mb-3"></i>
+                    <p>생성된 프로젝트가 없습니다. 첫 프로젝트를 만들어보세요!</p>
+                </div>`;
+            return;
+        }
+
+        container.innerHTML = projects.map(p => `
+            <div class="col-md-4 mb-4">
+                <div class="card h-100 shadow-sm">
+                    <div class="card-body">
+                        <div class="d-flex justify-content-between align-items-start mb-2">
+                            <h5 class="card-title text-truncate" style="max-width: 80%;">${p.title}</h5>
+                            <span class="badge ${p.mode === 'CREATIVE' ? 'bg-success' : 'bg-info'}">${p.mode}</span>
+                        </div>
+                        <p class="card-text text-muted small">ID: ${p.project_id.substring(0, 8)}...</p>
+                    </div>
+                    <div class="card-footer bg-transparent border-top-0 d-grid">
+                        <a href="/creation/workspace/${p.project_id}" class="btn btn-outline-primary">
+                            <i class="fas fa-pen-nib me-1"></i> 집필 시작
+                        </a>
+                    </div>
+                </div>
+            </div>
+        `).join('');
+    } catch (error) {
+        console.error('Failed to load projects:', error);
+    }
+}
+
+async function createNewProject() {
+    const title = document.getElementById('projectTitle').value;
+    const mode = document.getElementById('projectMode').value;
+    const reference_id = document.getElementById('referenceId').value;
+
+    if (!title) return alert('제목을 입력해주세요.');
+
+    try {
+        const response = await fetch('/creation/api/projects', {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ title, mode, reference_id })
+        });
+        
+        if (response.ok) {
+            bootstrap.Modal.getInstance(document.getElementById('createProjectModal')).hide();
+            loadProjects();
+        }
+    } catch (error) {
+        alert('프로젝트 생성 실패');
+    }
+}
+
+document.addEventListener('DOMContentLoaded', loadProjects);
+</script>
+{% endblock %}
+

templates/novel_workspace.html

@@ -0,0 +1,190 @@
+{% extends "admin.html" %}
+
+{% block content %}
+<style>
+    .workspace-container {
+        display: flex;
+        height: calc(100vh - 100px);
+        background: #f8f9fa;
+        overflow: hidden;
+    }
+    .editor-area {
+        flex: 1;
+        display: flex;
+        flex-direction: column;
+        padding: 20px;
+        overflow-y: auto;
+    }
+    .memory-sidebar {
+        width: 320px;
+        background: white;
+        border-left: 1px solid #dee2e6;
+        display: flex;
+        flex-direction: column;
+        padding: 15px;
+        overflow-y: auto;
+    }
+    .chat-bubble {
+        max-width: 80%;
+        margin-bottom: 15px;
+        padding: 12px 16px;
+        border-radius: 15px;
+        position: relative;
+    }
+    .bubble-user {
+        align-self: flex-end;
+        background: #007bff;
+        color: white;
+        border-bottom-right-radius: 2px;
+    }
+    .bubble-ai {
+        align-self: flex-start;
+        background: #e9ecef;
+        color: #212529;
+        border-bottom-left-radius: 2px;
+    }
+    .input-box {
+        background: white;
+        padding: 15px;
+        border-top: 1px solid #dee2e6;
+        display: flex;
+        gap: 10px;
+    }
+    .memory-section {
+        margin-bottom: 25px;
+    }
+    .memory-section h6 {
+        font-weight: 700;
+        color: #495057;
+        text-transform: uppercase;
+        font-size: 0.8rem;
+        margin-bottom: 10px;
+        display: flex;
+        align-items: center;
+        gap: 5px;
+    }
+    .fact-item {
+        background: #fff3cd;
+        padding: 8px 12px;
+        border-radius: 8px;
+        font-size: 0.9rem;
+        margin-bottom: 8px;
+        border-left: 4px solid #ffc107;
+    }
+    .summary-text {
+        font-size: 0.9rem;
+        color: #6c757d;
+        line-height: 1.5;
+    }
+</style>
+
+<div class="workspace-container">
+    <!-- 집필 구역 (에디터/채팅) -->
+    <div class="editor-area">
+        <div class="d-flex justify-content-between align-items-center mb-4">
+            <h4><i class="fas fa-edit me-2"></i>{{ project.title }} <small class="text-muted">({{ project.mode }})</small></h4>
+            <div id="loadingStatus" class="spinner-border spinner-border-sm text-primary d-none" role="status"></div>
+        </div>
+        
+        <div id="chatWindow" class="d-flex flex-column flex-grow-1" style="min-height: 300px;">
+            <div class="chat-bubble bubble-ai">
+                안녕하세요! 소설 <b>"{{ project.title }}"</b>의 집필을 도와드릴 AI 어시스턴트입니다. 
+                캐릭터 설정이나 다음 전개에 대해 무엇이든 물어보세요.
+            </div>
+        </div>
+
+        <div class="input-box mt-auto">
+            <textarea id="userInput" class="form-control" rows="2" placeholder="에이전트에게 지시하거나 질문하세요... (Enter로 전송)"></textarea>
+            <button id="sendBtn" class="btn btn-primary" onclick="sendMessage()">
+                <i class="fas fa-paper-plane"></i>
+            </button>
+        </div>
+    </div>
+
+    <!-- 메모리 사이드바 -->
+    <div class="memory-sidebar">
+        <div class="memory-section">
+            <h6><i class="fas fa-users text-warning"></i> 캐릭터 및 설정 (Mem0)</h6>
+            <div id="factContainer">
+                <div class="text-muted small">설정을 불러오는 중...</div>
+            </div>
+        </div>
+
+        <div class="memory-section">
+            <h6><i class="fas fa-history text-info"></i> 스토리 문맥 요약 (Zep)</h6>
+            <div id="summaryContainer" class="summary-text italic">
+                요약을 불러오는 중...
+            </div>
+        </div>
+
+        <div class="mt-auto">
+            <button class="btn btn-sm btn-outline-secondary w-100" onclick="refreshMemory()">
+                <i class="fas fa-sync-alt me-1"></i> 동기화
+            </button>
+        </div>
+    </div>
+</div>
+
+<script>
+const projectId = "{{ project.project_id }}";
+
+async function sendMessage() {
+    const input = document.getElementById('userInput');
+    const message = input.value.trim();
+    if (!message) return;
+
+    appendBubble(message, 'user');
+    input.value = '';
+    
+    document.getElementById('loadingStatus').classList.remove('d-none');
+    
+    try {
+        const response = await fetch(`/creation/api/chat/${projectId}`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ message })
+        });
+        const data = await response.json();
+        appendBubble(data.reply, 'ai');
+    } catch (error) {
+        appendBubble("오류가 발생했습니다.", 'ai');
+    } finally {
+        document.getElementById('loadingStatus').classList.add('d-none');
+        refreshMemory();
+    }
+}
+
+function appendBubble(content, role) {
+    const win = document.getElementById('chatWindow');
+    const div = document.createElement('div');
+    div.className = `chat-bubble bubble-${role}`;
+    div.innerHTML = content.replace(/\n/g, '<br>');
+    win.appendChild(div);
+    win.scrollTop = win.scrollHeight;
+}
+
+async function refreshMemory() {
+    // 실제로는 API를 통해 Zep/Mem0 데이터를 실시간 조회해야 함
+    // 여기서는 목업 데이터를 표시하거나 에이전트 응답 시 함께 갱신
+    try {
+        // 임시 목업 데이터
+        document.getElementById('factContainer').innerHTML = `
+            <div class="fact-item">주인공: 김철수 (25세, 검사)</div>
+            <div class="fact-item">배경: 2024년 가상 서울</div>
+        `;
+        document.getElementById('summaryContainer').innerText = "최근 철수는 강남역 근처에서 의문의 포탈을 발견했습니다.";
+    } catch (e) {}
+}
+
+document.getElementById('userInput').addEventListener('keypress', function(e) {
+    if (e.key === 'Enter' && !e.shiftKey) {
+        e.preventDefault();
+        sendMessage();
+    }
+});
+
+// 초기 로드
+document.addEventListener('DOMContentLoaded', refreshMemory);
+</script>
+{% endblock %}
+