Auto-deploy from GitHub Actions - 2025-12-12 09:16:33

app/routes.py

@@ -4946,7 +4946,7 @@ def process_graph(file_id):
 @main_bp.route('/api/files/<int:file_id>/process/tags', methods=['POST'])
 @login_required
 def generate_tags(file_id):
-    """태그 생성 (Parent Chunk, 회차 분석, GraphRAG 활용)"""
+    """태그 생성 (Parent Chunk, 회차 분석, GraphRAG 활용) - 계층적 구조"""
     try:
         file = UploadedFile.query.get_or_404(file_id)
         
@@ -4964,47 +4964,79 @@ def generate_tags(file_id):
         parent_chunk_text = ""
         if parent_chunk:
             parent_chunk_text = f"""
-[작품 전체 정보]
+[Parent Chunk]
 세계관: {parent_chunk.world_view or '없음'}
 주요 캐릭터: {parent_chunk.characters or '없음'}
 주요 스토리: {parent_chunk.story or '없음'}
+에피소드 요약: {parent_chunk.episodes or '없음'}
+기타: {parent_chunk.others or '없음'}
 """
         
         # 2. 회차 분석 (요약)
-        analyses = EpisodeAnalysis.query.filter_by(file_id=file_id).order_by(EpisodeAnalysis.id).limit(10).all()
+        analyses = EpisodeAnalysis.query.filter_by(file_id=file_id).order_by(EpisodeAnalysis.id).limit(20).all()
         analysis_text = ""
         if analyses:
-            analysis_text = "[주요 회차 분석 내용]\n"
+            analysis_text = "[회차별 분석 (요약)]\n"
             for analysis in analyses:
                 # 내용이 너무 길면 앞부분만 사용
-                content_preview = analysis.analysis_content[:500] + "..." if len(analysis.analysis_content) > 500 else analysis.analysis_content
+                content_preview = analysis.analysis_content[:300] + "..." if len(analysis.analysis_content) > 300 else analysis.analysis_content
                 analysis_text += f"- {analysis.episode_title}: {content_preview}\n"
         
-        # 3. GraphRAG (주요 인물, 키워드)
-        entities = GraphEntity.query.filter_by(file_id=file_id).limit(20).all()
-        entity_text = ""
-        if entities:
-            entity_names = [e.entity_name for e in entities]
-            entity_text = f"[주요 등장 요소]\n{', '.join(entity_names)}"
+        # 3. GraphRAG (주요 인물, 관계, 사건)
+        entities = GraphEntity.query.filter_by(file_id=file_id).limit(30).all()
+        relationships = GraphRelationship.query.filter_by(file_id=file_id).limit(30).all()
+        events = GraphEvent.query.filter_by(file_id=file_id).limit(30).all()
+        
+        graph_text = ""
+        if entities or relationships or events:
+            graph_text = "[GraphRAG 데이터]\n"
+            if entities:
+                entity_names = [f"{e.entity_name}({e.entity_type})" for e in entities]
+                graph_text += f"엔티티: {', '.join(entity_names)}\n"
+            if relationships:
+                rel_strs = [f"{r.source}->{r.target}({r.relationship_type})" for r in relationships]
+                graph_text += f"관계: {', '.join(rel_strs)}\n"
+            if events:
+                event_names = [e.event_name for e in events]
+                graph_text += f"사건: {', '.join(event_names)}\n"
             
         # 데이터가 너무 없으면 태그 생성 불가
-        if not parent_chunk_text and not analysis_text and not entity_text:
+        if not parent_chunk_text and not analysis_text and not graph_text:
             return jsonify({'error': '태그를 생성할 충분한 데이터(Parent Chunk, 회차 분석 등)가 없습니다.'}), 400
             
         # 프롬프트 구성
         prompt = f"""
-다음은 웹소설의 분석 데이터입니다. 이 데이터를 바탕으로 이 작품을 잘 설명하는 태그 10개~20개를 생성해주세요.
+다음은 웹소설의 분석 데이터입니다. 이 데이터를 바탕으로 각 섹션별로 핵심 태그를 추출해주세요.
 
 {parent_chunk_text}
 
 {analysis_text}
 
-{entity_text}
+{graph_text}
 
-요구사항:
-1. 장르, 분위기, 주요 소재, 캐릭터 특성 등을 포함하는 태그를 생성하세요.
-2. 반드시 JSON 배열 형식으로만 응답하세요. (예: ["판타지", "로맨스", "성장물", "마법"])
-3. 다른 설명 없이 JSON 배열만 출력하세요.
+[요구사항]
+다음 JSON 형식으로 태그를 생성하세요. 각 항목당 5~10개의 핵심 키워드를 추출하세요. 없는 항목은 빈 배열로 두세요.
+
+{{
+  "parent_chunk": {{
+    "world_view": ["태그1", "태그2"],
+    "characters": ["태그1", "태그2"],
+    "story": ["태그1", "태그2"],
+    "others": ["태그1", "태그2"]
+  }},
+  "episodes": {{
+    "world_view": ["태그1", "태그2"],
+    "characters": ["태그1", "태그2"],
+    "items": ["태그1", "태그2"]
+  }},
+  "graph_rag": {{
+    "characters": ["태그1", "태그2"],
+    "relationships": ["태그1", "태그2"],
+    "events": ["태그1", "태그2"]
+  }}
+}}
+
+반드시 JSON 형식으로만 응답하세요. 다른 설명은 포함하지 마세요. JSON 구문을 정확히 지켜주세요.
 """
 
         # AI 호출
@@ -5024,8 +5056,8 @@ def generate_tags(file_id):
                     result = gemini_client.generate_response(
                         prompt=prompt,
                         model_name=gemini_model_name,
-                        temperature=0.7,
-                        max_output_tokens=1024
+                        temperature=0.5,
+                        max_output_tokens=2048
                     )
                     if not result['error'] and result.get('response'):
                         response_text = result['response'].strip()
@@ -5035,15 +5067,7 @@ def generate_tags(file_id):
         # Ollama 모델인 경우 (또는 Gemini 실패 시)
         if not response_text:
             try:
-                ollama_model_name = file.model_name
-                if ollama_model_name.lower().startswith('gemini:'): # Gemini 모델명이지만 Ollama로 시도하는 경우 (설정 오류 등)
-                     # 기본 Ollama 모델 사용하거나 에러 처리
-                     pass
-                
-                # Ollama 호출 로직 (generate_response_with_ollama 사용)
-                # 여기서는 직접 호출 대신 기존 함수 활용
-                # app/routes.py에 ollama 관련 함수가 있는지 확인 필요하지만, 
-                # 간단하게 requests로 호출 구현
+                # Ollama 호출 로직
                 import requests
                 from app.core.config import get_config
                 config = get_config()
@@ -5054,9 +5078,10 @@ def generate_tags(file_id):
                         'model': file.model_name,
                         'prompt': prompt,
                         'stream': False,
-                        'options': {'temperature': 0.7}
+                        'options': {'temperature': 0.5},
+                        'format': 'json'  # Ollama JSON 모드 활성화
                     },
-                    timeout=120
+                    timeout=180
                 )
                 
                 if ollama_response.status_code == 200:
@@ -5072,33 +5097,28 @@ def generate_tags(file_id):
         import json
         import re
         
-        tags = []
+        tags_data = {}
         try:
             # JSON 추출 시도
-            json_match = re.search(r'\[.*\]', response_text, re.DOTALL)
+            json_match = re.search(r'\{.*\}', response_text, re.DOTALL)
             if json_match:
-                tags = json.loads(json_match.group(0))
+                json_str = json_match.group(0)
+                tags_data = json.loads(json_str)
             else:
-                # JSON 형식이 아니면 쉼표로 구분하여 처리 시도
-                tags = [t.strip() for t in response_text.replace('[', '').replace(']', '').replace('"', '').split(',')]
+                # JSON 파싱 실패 시
+                print(f"[태그 생성] JSON 형식이 아님: {response_text[:100]}...")
+                return jsonify({'error': 'AI 응답이 올바른 JSON 형식이 아닙니다.'}), 500
         except Exception as e:
             print(f"[태그 생성] 파싱 오류: {str(e)}")
-            # 쉼표로 단순 분리 시도
-            tags = [t.strip() for t in response_text.split(',')]
-            
-        # 빈 태그 제거 및 문자열로 변환하여 저장
-        tags = [t for t in tags if t]
-        
-        if not tags:
-            return jsonify({'error': '태그를 추출하지 못했습니다.'}), 500
+            return jsonify({'error': f'태그 파싱 중 오류가 발생했습니다: {str(e)}'}), 500
             
         # DB 저장 (JSON 문자열로 저장)
-        file.tags = json.dumps(tags, ensure_ascii=False)
+        file.tags = json.dumps(tags_data, ensure_ascii=False)
         db.session.commit()
         
         return jsonify({
             'message': '태그가 생성되었습니다.',
-            'tags': tags,
+            'tags': tags_data,
             'file_id': file.id
         }), 200
         

templates/admin_files.html

@@ -674,6 +674,19 @@
         </div>
     </div>
 
+    <!-- 태그 보기 모달 -->
+    <div id="tagsModal" class="modal">
+        <div class="modal-content">
+            <div class="modal-header">
+                <div class="modal-title" id="tagsModalTitle">태그 목록</div>
+                <button class="modal-close" onclick="closeTagsModal()">&times;</button>
+            </div>
+            <div id="tagsContent" class="modal-body">
+                태그를 불러오는 중...
+            </div>
+        </div>
+    </div>
+
     <!-- vis-network 라이브러리 -->
     <script type="text/javascript" src="https://unpkg.com/vis-network@latest/standalone/umd/vis-network.min.js"></script>
 
@@ -798,6 +811,7 @@
                             <div class="file-actions">
                                 <button class="btn btn-secondary" onclick="viewSummary(${file.id}, '${escapeHtml(file.original_filename)}')" style="padding: 4px 8px; font-size: 12px; margin-right: 4px;">요약 내용 보기</button>
                                 <button class="btn btn-info" onclick="viewChunks(${file.id}, '${escapeHtml(file.original_filename)}')" style="padding: 4px 8px; font-size: 12px; margin-right: 4px;">청크 보기</button>
+                                <button class="btn" onclick="viewTags(${file.id}, '${escapeHtml(file.original_filename)}')" style="padding: 4px 8px; font-size: 12px; margin-right: 4px; background: #673ab7; color: white;">태그 보기</button>
                                 <button class="btn btn-primary" onclick="viewGraphRAG(${file.id}, '${escapeHtml(file.original_filename)}')" style="padding: 4px 8px; font-size: 12px; margin-right: 4px;">GraphRAG 보기</button>
                                 <button class="btn btn-success" onclick="viewGraphRAGVisualization(${file.id}, '${escapeHtml(file.original_filename)}')" style="padding: 4px 8px; font-size: 12px;">그래프 시각화</button>
                             </div>
@@ -1835,6 +1849,113 @@
             }
         });
 
+        async function viewTags(fileId, fileName) {
+            const modal = document.getElementById('tagsModal');
+            const title = document.getElementById('tagsModalTitle');
+            const content = document.getElementById('tagsContent');
+            
+            title.textContent = `태그 목록 - ${fileName}`;
+            content.innerHTML = '<div style="text-align: center; padding: 24px; color: #5f6368;">태그를 불러오는 중...</div>';
+            modal.classList.add('active');
+            
+            try {
+                // 파일 정보 다시 조회 (태그 정보 포함)
+                const response = await fetch('/api/files', { credentials: 'include' });
+                const data = await response.json();
+                const file = data.files.find(f => f.id === fileId);
+                
+                if (file && file.tags) {
+                    try {
+                        const tags = JSON.parse(file.tags);
+                        let html = '';
+                        
+                        // 계층적 구조 렌더링
+                        
+                        // Parent Chunk
+                        if (tags.parent_chunk) {
+                            html += '<div style="margin-bottom: 24px;">';
+                            html += '<h3 style="font-size: 15px; font-weight: 700; color: #1a73e8; margin-bottom: 12px; border-bottom: 2px solid #e8f0fe; padding-bottom: 8px;">Parent Chunk (전체 요약)</h3>';
+                            html += renderTagsGroup(tags.parent_chunk);
+                            html += '</div>';
+                        }
+                        
+                        // Episodes
+                        if (tags.episodes) {
+                            html += '<div style="margin-bottom: 24px;">';
+                            html += '<h3 style="font-size: 15px; font-weight: 700; color: #1a73e8; margin-bottom: 12px; border-bottom: 2px solid #e8f0fe; padding-bottom: 8px;">회차별 분석</h3>';
+                            html += renderTagsGroup(tags.episodes);
+                            html += '</div>';
+                        }
+                        
+                        // GraphRAG
+                        if (tags.graph_rag) {
+                            html += '<div style="margin-bottom: 24px;">';
+                            html += '<h3 style="font-size: 15px; font-weight: 700; color: #1a73e8; margin-bottom: 12px; border-bottom: 2px solid #e8f0fe; padding-bottom: 8px;">GraphRAG (지식 그래프)</h3>';
+                            html += renderTagsGroup(tags.graph_rag);
+                            html += '</div>';
+                        }
+                        
+                        // 구버전 태그 (배열) 호환성 - 계층적 구조가 아닐 경우
+                        if (!tags.parent_chunk && !tags.episodes && !tags.graph_rag && Array.isArray(tags)) {
+                            html += '<div style="display: flex; flex-wrap: wrap; gap: 8px;">';
+                            tags.forEach(tag => {
+                                html += `<span style="background: #e8f0fe; color: #1a73e8; padding: 6px 12px; border-radius: 16px; font-size: 13px; border: 1px solid #d2e3fc;">#${escapeHtml(tag)}</span>`;
+                            });
+                            html += '</div>';
+                        }
+                        
+                        if (!html) {
+                            html = '<div style="text-align: center; padding: 20px; color: #5f6368;">태그 형식이 올바르지 않거나 비어있습니다.</div>';
+                        }
+                        
+                        content.innerHTML = html;
+                    } catch (e) {
+                        console.error('태그 파싱 오류:', e);
+                        content.innerHTML = '<div style="text-align: center; padding: 20px; color: #c5221f;">태그 데이터를 파싱할 수 없습니다.</div>';
+                    }
+                } else {
+                    content.innerHTML = '<div style="text-align: center; padding: 20px; color: #5f6368;">생성된 태그가 없습니다.</div>';
+                }
+            } catch (error) {
+                console.error('태그 로드 오류:', error);
+                content.innerHTML = '<div style="text-align: center; padding: 20px; color: #c5221f;">태그 정보를 불러오는 중 오류가 발생했습니다.</div>';
+            }
+        }
+        
+        function renderTagsGroup(group) {
+            let html = '<div style="display: grid; grid-template-columns: repeat(auto-fill, minmax(200px, 1fr)); gap: 16px;">';
+            
+            // 키 이름을 한글로 매핑
+            const keyMap = {
+                'world_view': '세계관',
+                'characters': '인물',
+                'story': '스토리',
+                'others': '기타',
+                'items': '아이템/소재',
+                'relationships': '관계',
+                'events': '사건'
+            };
+
+            for (const [key, values] of Object.entries(group)) {
+                if (values && values.length > 0) {
+                    const label = keyMap[key] || key;
+                    html += `<div style="background: #f8f9fa; padding: 12px; border-radius: 8px; border: 1px solid #e8eaed;">`;
+                    html += `<div style="font-size: 13px; font-weight: 700; color: #5f6368; margin-bottom: 8px;">${escapeHtml(label)}</div>`;
+                    html += `<div style="display: flex; flex-wrap: wrap; gap: 6px;">`;
+                    values.forEach(tag => {
+                        html += `<span style="background: white; border: 1px solid #dadce0; padding: 4px 8px; border-radius: 4px; font-size: 12px; color: #3c4043;">${escapeHtml(tag)}</span>`;
+                    });
+                    html += `</div></div>`;
+                }
+            }
+            html += '</div>';
+            return html;
+        }
+
+        function closeTagsModal() {
+            document.getElementById('tagsModal').classList.remove('active');
+        }
+
         // 페이지 로드 시 초기화
         window.addEventListener('load', () => {
             loadModelFilter();