Auto-deploy from GitHub Actions - 2025-12-12 09:03:43

app/__init__.py

@@ -234,7 +234,7 @@ def migrate_database(app: Flask) -> None:
                         # Boolean 타입은 DB에 따라 다를 수 있으므로 주의 (PostgreSQL/SQLite 모두 BOOLEAN 지원)
                         conn.execute(text("ALTER TABLE \"user\" ADD COLUMN must_change_password BOOLEAN DEFAULT FALSE NOT NULL"))
             
-            # 2. uploaded_file 테이블 - uploaded_by, parent_file_id 컬럼
+            # 2. uploaded_file 테이블 - uploaded_by, parent_file_id, tags 컬럼
             if inspector.has_table('uploaded_file'):
                 columns = [c['name'] for c in inspector.get_columns('uploaded_file')]
                 with db.engine.begin() as conn:
@@ -245,6 +245,10 @@ def migrate_database(app: Flask) -> None:
                     if 'parent_file_id' not in columns:
                         logger.info("uploaded_file 테이블에 parent_file_id 컬럼 추가 중...")
                         conn.execute(text("ALTER TABLE uploaded_file ADD COLUMN parent_file_id INTEGER"))
+                    
+                    if 'tags' not in columns:
+                        logger.info("uploaded_file 테이블에 tags 컬럼 추가 중...")
+                        conn.execute(text("ALTER TABLE uploaded_file ADD COLUMN tags TEXT"))
             
             # 3. document_chunk 테이블 - chunk_metadata 컬럼
             if inspector.has_table('document_chunk'):

app/database.py

@@ -47,6 +47,7 @@ class UploadedFile(db.Model):
     uploaded_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
     uploaded_by = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=True)
     parent_file_id = db.Column(db.Integer, db.ForeignKey('uploaded_file.id'), nullable=True)  # 이어서 업로드한 경우 원본 파일 ID
+    tags = db.Column(db.Text, nullable=True)  # AI가 생성한 태그 (JSON 문자열 또는 쉼표 구분)
     
     # 관계
     parent_file = db.relationship('UploadedFile', remote_side=[id], backref='child_files')
@@ -65,6 +66,7 @@ class UploadedFile(db.Model):
             'uploaded_at': self.uploaded_at.isoformat() if self.uploaded_at else None,
             'uploaded_by': self.uploaded_by,
             'parent_file_id': self.parent_file_id,
+            'tags': self.tags,
             'chunk_count': chunk_count,
             'child_count': len(self.child_files) if self.child_files else 0
         }

app/routes.py

@@ -4943,6 +4943,169 @@ def process_graph(file_id):
     except Exception as e:
         return jsonify({'error': f'Graph Extraction 중 오류가 발생했습니다: {str(e)}', 'step': 'graph'}), 500
 
+@main_bp.route('/api/files/<int:file_id>/process/tags', methods=['POST'])
+@login_required
+def generate_tags(file_id):
+    """태그 생성 (Parent Chunk, 회차 분석, GraphRAG 활용)"""
+    try:
+        file = UploadedFile.query.get_or_404(file_id)
+        
+        # 권한 확인
+        if not current_user.is_admin and file.uploaded_by != current_user.id:
+            return jsonify({'error': '권한이 없습니다.'}), 403
+            
+        # 모델 확인
+        if not file.model_name:
+            return jsonify({'error': '파일에 연결된 AI 모델이 없습니다.'}), 400
+            
+        # 데이터 수집
+        # 1. Parent Chunk
+        parent_chunk = ParentChunk.query.filter_by(file_id=file_id).first()
+        parent_chunk_text = ""
+        if parent_chunk:
+            parent_chunk_text = f"""
+[작품 전체 정보]
+세계관: {parent_chunk.world_view or '없음'}
+주요 캐릭터: {parent_chunk.characters or '없음'}
+주요 스토리: {parent_chunk.story or '없음'}
+"""
+        
+        # 2. 회차 분석 (요약)
+        analyses = EpisodeAnalysis.query.filter_by(file_id=file_id).order_by(EpisodeAnalysis.id).limit(10).all()
+        analysis_text = ""
+        if analyses:
+            analysis_text = "[주요 회차 분석 내용]\n"
+            for analysis in analyses:
+                # 내용이 너무 길면 앞부분만 사용
+                content_preview = analysis.analysis_content[:500] + "..." if len(analysis.analysis_content) > 500 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)}"
+            
+        # 데이터가 너무 없으면 태그 생성 불가
+        if not parent_chunk_text and not analysis_text and not entity_text:
+            return jsonify({'error': '태그를 생성할 충분한 데이터(Parent Chunk, 회차 분석 등)가 없습니다.'}), 400
+            
+        # 프롬프트 구성
+        prompt = f"""
+다음은 웹소설의 분석 데이터입니다. 이 데이터를 바탕으로 이 작품을 잘 설명하는 태그 10개~20개를 생성해주세요.
+
+{parent_chunk_text}
+
+{analysis_text}
+
+{entity_text}
+
+요구사항:
+1. 장르, 분위기, 주요 소재, 캐릭터 특성 등을 포함하는 태그를 생성하세요.
+2. 반드시 JSON 배열 형식으로만 응답하세요. (예: ["판타지", "로맨스", "성장물", "마법"])
+3. 다른 설명 없이 JSON 배열만 출력하세요.
+"""
+
+        # AI 호출
+        print(f"[태그 생성] '{file.original_filename}' 태그 생성 요청 (모델: {file.model_name})")
+        
+        response_text = None
+        
+        # Gemini 모델인 경우
+        if 'gemini' in file.model_name.lower():
+            try:
+                gemini_model_name = file.model_name
+                if gemini_model_name.lower().startswith('gemini:'):
+                    gemini_model_name = gemini_model_name.split(':', 1)[1].strip()
+                
+                gemini_client = get_gemini_client()
+                if gemini_client.is_configured():
+                    result = gemini_client.generate_response(
+                        prompt=prompt,
+                        model_name=gemini_model_name,
+                        temperature=0.7,
+                        max_output_tokens=1024
+                    )
+                    if not result['error'] and result.get('response'):
+                        response_text = result['response'].strip()
+            except Exception as e:
+                print(f"[태그 생성] Gemini 오류: {str(e)}")
+        
+        # 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로 호출 구현
+                import requests
+                from app.core.config import get_config
+                config = get_config()
+                
+                ollama_response = requests.post(
+                    f"{config.OLLAMA_API_URL}/api/generate",
+                    json={
+                        'model': file.model_name,
+                        'prompt': prompt,
+                        'stream': False,
+                        'options': {'temperature': 0.7}
+                    },
+                    timeout=120
+                )
+                
+                if ollama_response.status_code == 200:
+                    response_data = ollama_response.json()
+                    response_text = response_data.get('response', '').strip()
+            except Exception as e:
+                print(f"[태그 생성] Ollama 오류: {str(e)}")
+
+        if not response_text:
+            return jsonify({'error': 'AI 응답을 받지 못했습니다.'}), 500
+            
+        # JSON 파싱
+        import json
+        import re
+        
+        tags = []
+        try:
+            # JSON 추출 시도
+            json_match = re.search(r'\[.*\]', response_text, re.DOTALL)
+            if json_match:
+                tags = json.loads(json_match.group(0))
+            else:
+                # JSON 형식이 아니면 쉼표로 구분하여 처리 시도
+                tags = [t.strip() for t in response_text.replace('[', '').replace(']', '').replace('"', '').split(',')]
+        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
+            
+        # DB 저장 (JSON 문자열로 저장)
+        file.tags = json.dumps(tags, ensure_ascii=False)
+        db.session.commit()
+        
+        return jsonify({
+            'message': '태그가 생성되었습니다.',
+            'tags': tags,
+            'file_id': file.id
+        }), 200
+        
+    except Exception as e:
+        db.session.rollback()
+        return jsonify({'error': f'태그 생성 중 오류가 발생했습니다: {str(e)}'}), 500
+
 @main_bp.route('/api/files/<int:file_id>/metadata', methods=['POST'])
 @login_required
 def create_file_metadata(file_id):

templates/admin_webnovels.html

@@ -992,8 +992,28 @@
                         const toggleButtonText = isPublic ? '비공개로' : '공개로';
                         const toggleButtonClass = isPublic ? 'btn-warning' : 'btn-success';
                         
+                        // 태그 표시
+                        let tagsHtml = '';
+                        if (file.tags) {
+                            try {
+                                const tags = JSON.parse(file.tags);
+                                if (tags && tags.length > 0) {
+                                    tagsHtml = '<div style="margin-top: 6px; display: flex; flex-wrap: wrap; gap: 4px;">';
+                                    tags.forEach(tag => {
+                                        tagsHtml += `<span style="background: #e8f0fe; color: #1a73e8; padding: 2px 6px; border-radius: 4px; font-size: 11px; border: 1px solid #d2e3fc;">#${escapeHtml(tag)}</span>`;
+                                    });
+                                    tagsHtml += '</div>';
+                                }
+                            } catch (e) {
+                                console.error('태그 파싱 오류:', e);
+                            }
+                        }
+
                         row.innerHTML = `
-                            <td>${fileNameDisplay}</td>
+                            <td>
+                                <div>${fileNameDisplay}</div>
+                                ${tagsHtml}
+                            </td>
                             <td>${modelName}</td>
                             <td class="file-size">${fileSize}</td>
                             <td>${uploadDate}</td>
@@ -1004,12 +1024,13 @@
                                 </button>
                             </td>
                             <td>
-                                <div class="file-actions">
+                                <div class="file-actions" style="flex-wrap: wrap; row-gap: 4px;">
                                     ${(file.has_parent_chunk !== undefined && file.has_parent_chunk) ? 
                                         `<button class="btn btn-primary" onclick="viewParentChunk(${file.id}, '${escapeHtml(file.original_filename)}')" style="padding: 4px 8px; font-size: 12px; margin-right: 4px;">Parent Chunk</button>` :
                                         `<button class="btn btn-secondary" onclick="createParentChunk(${file.id}, '${escapeHtml(file.original_filename)}')" style="padding: 4px 8px; font-size: 12px; margin-right: 4px;">Parent Chunk 생성</button>`
                                     }
                                     <button class="btn btn-info" onclick="createMetadata(${file.id}, '${escapeHtml(file.original_filename)}')" style="padding: 4px 8px; font-size: 12px; margin-right: 4px;">메타데이터 생성</button>
+                                    <button class="btn" style="background: #673ab7; color: white; padding: 4px 8px; font-size: 12px; margin-right: 4px;" onclick="createTags(${file.id}, '${escapeHtml(file.original_filename)}')">태그 생성</button>
                                     <button class="btn btn-primary" onclick="continueUpload(${file.id})" style="padding: 4px 8px; font-size: 12px; margin-right: 4px;">이어서 업로드</button>
                                     <button class="btn btn-secondary" onclick="deleteFile(${file.id})" style="padding: 4px 8px; font-size: 12px;">삭제</button>
                                 </div>
@@ -1896,6 +1917,43 @@
             }
         }
 
+        // 태그 생성
+        async function createTags(fileId, fileName) {
+            if (!confirm(`"${fileName}" 파일의 태그를 생성하시겠습니까?\n\n이 작업은 AI 모델을 사용하여 Parent Chunk와 회차별 분석 데이터를 종합하여 분석합니다.`)) {
+                return;
+            }
+
+            const button = event.target;
+            const originalText = button.textContent;
+            button.disabled = true;
+            button.textContent = '생성 중...';
+
+            try {
+                const response = await fetch(`/api/files/${fileId}/process/tags`, {
+                    method: 'POST',
+                    credentials: 'include'
+                });
+                const data = await response.json();
+
+                if (response.ok) {
+                    let message = '태그 생성이 완료되었습니다.';
+                    if (data.tags && data.tags.length > 0) {
+                        message += `\n생성된 태그: ${data.tags.join(', ')}`;
+                    }
+                    showAlert(message, 'success');
+                    loadFiles(); // 파일 목록 새로고침
+                } else {
+                    showAlert(`태그 생성 실패: ${data.error || '알 수 없는 오류'}`, 'error');
+                }
+            } catch (error) {
+                showAlert(`태그 생성 중 오류가 발생했습니다: ${error.message}`, 'error');
+                console.error('태그 생성 오류:', error);
+            } finally {
+                button.disabled = false;
+                button.textContent = originalText;
+            }
+        }
+
         // 모달 외부 클릭 시 닫기
         window.addEventListener('click', (event) => {
             const modal = document.getElementById('parentChunkModal');