fix: GraphRAG 생성 로직 개선 및 회차 분석 후 자동 Graph Extraction 추가

- create_chunks_for_file에서 Graph Extraction이 회차 분석 성공 여부와 관계없이 실행되도록 수정
- 회차 분석 API 성공 후 Graph Extraction 자동 실행 추가
- parent_chunk 변수 스코프 문제 해결
- 로깅 미들웨어 추가 및 개선

EXAONE_설치_가이드.md

@@ -164,6 +164,8 @@ tokenizer = AutoTokenizer.from_pretrained("LGAI-EXAONE/EXAONE-3.0-7.8B-Instruct"
 
 
 
+
+
 
 
 

EXAONE_추가_안내.md

@@ -81,6 +81,8 @@ Ollama를 거치지 않고 Python에서 직접 Hugging Face 모델을 사용할
 
 
 
+
+
 
 
 

add_exaone_model.py

@@ -159,6 +159,8 @@ if __name__ == "__main__":
 
 
 
+
+
 
 
 

app/__init__.py

@@ -2,7 +2,7 @@
 Flask 애플리케이션 초기화
 """
 
-from flask import Flask
+from flask import Flask, request
 from flask_login import LoginManager
 import sqlite3
 from pathlib import Path
@@ -69,6 +69,20 @@ def create_app() -> Flask:
     from app.routes import main_bp
     app.register_blueprint(main_bp)
     
+    # 요청 로깅 미들웨어 추가
+    @app.before_request
+    def log_request_info():
+        """각 HTTP 요청 정보를 로깅"""
+        logger.info(f"[요청] {request.method} {request.path} - IP: {request.remote_addr}")
+        if request.args:
+            logger.debug(f"[요청 파라미터] {dict(request.args)}")
+    
+    @app.after_request
+    def log_response_info(response):
+        """각 HTTP 응답 정보를 로깅"""
+        logger.info(f"[응답] {request.method} {request.path} - 상태: {response.status_code}")
+        return response
+    
     # 데이터베이스 초기화 및 마이그레이션
     with app.app_context():
         db.create_all()

app/routes.py

@@ -1003,36 +1003,44 @@ def create_chunks_for_file(file_id, content, skip_episode_analysis=False, skip_g
                 db.session.add(episode_analysis)
                 db.session.commit()
                 print(f"[회차 분석] 완료: {len(episode_sections)}개 회차 분석 결과를 하나의 텍스트로 저장")
-                
-                # 회차별 Graph Extraction 실행
-                if not skip_graph_extraction:
-                    print(f"[Graph Extraction] 회차별 Graph Extraction 시작...")
-                    graph_extraction_success_count = 0
-                    for section_type, section_title, section_content, section_metadata in episode_sections:
-                        try:
-                            print(f"[Graph Extraction] '{section_title}' Graph Extraction 중...")
-                            success = extract_graph_from_episode(
-                                episode_content=section_content,
-                                episode_title=section_title,
-                                file_id=file_id,
-                                full_content=content,
-                                parent_chunk=parent_chunk,
-                                model_name=model_name
-                            )
-                            if success:
-                                graph_extraction_success_count += 1
-                                print(f"[Graph Extraction] '{section_title}' Graph Extraction 완료")
-                            else:
-                                print(f"[Graph Extraction] '{section_title}' Graph Extraction 실패")
-                        except Exception as e:
-                            print(f"[Graph Extraction] '{section_title}' Graph Extraction 중 오류: {str(e)}")
-                            import traceback
-                            traceback.print_exc()
-                            continue
-                    
-                    print(f"[Graph Extraction] 완료: {graph_extraction_success_count}/{len(episode_sections)}개 회차 Graph Extraction 성공")
             else:
                 print(f"[회차 분석] 경고: 분석 결과가 없습니다.")
+        
+        # 회차별 Graph Extraction 실행 (회차 분석 성공 여부와 관계없이 실행)
+        if episode_sections and model_name and not skip_graph_extraction:
+            print(f"[Graph Extraction] 회차별 Graph Extraction 시작...")
+            
+            # Parent Chunk 가져오기 (회차 분석 블록 밖에서도 사용 가능하도록)
+            parent_chunk = None
+            try:
+                parent_chunk = ParentChunk.query.filter_by(file_id=file_id).first()
+            except:
+                pass
+            
+            graph_extraction_success_count = 0
+            for section_type, section_title, section_content, section_metadata in episode_sections:
+                try:
+                    print(f"[Graph Extraction] '{section_title}' Graph Extraction 중...")
+                    success = extract_graph_from_episode(
+                        episode_content=section_content,
+                        episode_title=section_title,
+                        file_id=file_id,
+                        full_content=content,
+                        parent_chunk=parent_chunk,
+                        model_name=model_name
+                    )
+                    if success:
+                        graph_extraction_success_count += 1
+                        print(f"[Graph Extraction] '{section_title}' Graph Extraction 완료")
+                    else:
+                        print(f"[Graph Extraction] '{section_title}' Graph Extraction 실패")
+                except Exception as e:
+                    print(f"[Graph Extraction] '{section_title}' Graph Extraction 중 오류: {str(e)}")
+                    import traceback
+                    traceback.print_exc()
+                    continue
+            
+            print(f"[Graph Extraction] 완료: {graph_extraction_success_count}/{len(episode_sections)}개 회차 Graph Extraction 성공")
         else:
             if not model_name:
                 print(f"[회차 분석] 모델명이 없어 회차 분석을 건너뜁니다.")
@@ -3734,11 +3742,35 @@ def process_episode_analysis(file_id):
             db.session.add(episode_analysis)
             db.session.commit()
             
+            # 회차 분석 성공 후 Graph Extraction 자동 실행
+            print(f"[단계 3: 회차 분석] Graph Extraction 자동 실행 시작...")
+            graph_success_count = 0
+            for section_type, section_title, section_content, section_metadata in episode_sections:
+                try:
+                    print(f"[단계 3: 회차 분석] '{section_title}' Graph Extraction 중...")
+                    success = extract_graph_from_episode(
+                        episode_content=section_content,
+                        episode_title=section_title,
+                        file_id=file_id,
+                        full_content=content,
+                        parent_chunk=parent_chunk,
+                        model_name=file.model_name
+                    )
+                    if success:
+                        graph_success_count += 1
+                        print(f"[단계 3: 회차 분석] '{section_title}' Graph Extraction 완료")
+                except Exception as e:
+                    print(f"[단계 3: 회차 분석] '{section_title}' Graph Extraction 중 오류: {str(e)}")
+                    continue
+            
+            print(f"[단계 3: 회차 분석] Graph Extraction 완료: {graph_success_count}/{len(episode_sections)}개 회차 성공")
+            
             return jsonify({
                 'file_id': file_id,
                 'filename': file.original_filename,
                 'episode_count': len(episode_sections),
-                'message': f'{len(episode_sections)}개 회차 분석이 완료되었습니다.',
+                'graph_success_count': graph_success_count,
+                'message': f'{len(episode_sections)}개 회차 분석이 완료되었습니다. (Graph Extraction: {graph_success_count}/{len(episode_sections)}개 성공)',
                 'step': 'episode-analysis',
                 'completed': True
             }), 200

download_exaone_model.py

@@ -87,6 +87,8 @@ if __name__ == "__main__":
 
 
 
+
+
 
 
 

install_exaone_direct.py

@@ -92,6 +92,8 @@ if __name__ == "__main__":
 
 
 
+
+
 
 
 

install_exaone_simple.py

@@ -69,6 +69,8 @@ if __name__ == "__main__":
 
 
 
+
+
 
 
 

run.py

@@ -12,28 +12,46 @@ from app import create_app
 
 app = create_app()
 
-# 로깅 설정
-if not app.debug:
-    # 프로덕션 환경에서는 파일로 로깅
-    if not os.path.exists('logs'):
-        os.mkdir('logs')
-    file_handler = RotatingFileHandler('logs/server.log', maxBytes=10240000, backupCount=10)
-    file_handler.setFormatter(logging.Formatter(
-        '%(asctime)s %(levelname)s: %(message)s [in %(pathname)s:%(lineno)d]'
-    ))
-    file_handler.setLevel(logging.INFO)
-    app.logger.addHandler(file_handler)
-    app.logger.setLevel(logging.INFO)
-    app.logger.info('서버 시작')
-
-# 디버그 모드에서도 콘솔에 모든 로그 출력
-logging.basicConfig(
-    level=logging.DEBUG,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
-    handlers=[
-        logging.StreamHandler(sys.stdout)
-    ]
-)
+# 로깅 설정 - 콘솔과 파일 모두에 출력
+if not os.path.exists('logs'):
+    os.mkdir('logs')
+
+# 파일 핸들러 설정
+file_handler = RotatingFileHandler('logs/server.log', maxBytes=10240000, backupCount=10)
+file_handler.setFormatter(logging.Formatter(
+    '%(asctime)s %(levelname)s: %(message)s [in %(pathname)s:%(lineno)d]'
+))
+file_handler.setLevel(logging.INFO)
+
+# 콘솔 핸들러 설정
+console_handler = logging.StreamHandler(sys.stdout)
+console_handler.setFormatter(logging.Formatter(
+    '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+))
+console_handler.setLevel(logging.INFO)
+
+# Flask 앱 로거 설정
+app.logger.setLevel(logging.INFO)
+app.logger.addHandler(file_handler)
+app.logger.addHandler(console_handler)
+
+# 루트 로거 설정 (모든 로거에 적용)
+root_logger = logging.getLogger()
+root_logger.setLevel(logging.INFO)
+root_logger.addHandler(console_handler)
+
+# Werkzeug 로거 설정 (HTTP 요청 로깅)
+werkzeug_logger = logging.getLogger('werkzeug')
+werkzeug_logger.setLevel(logging.INFO)
+# Werkzeug의 기본 핸들러 제거하고 새로 추가
+werkzeug_logger.handlers.clear()
+werkzeug_handler = logging.StreamHandler(sys.stdout)
+werkzeug_handler.setFormatter(logging.Formatter(
+    '%(asctime)s - %(levelname)s - %(message)s'
+))
+werkzeug_logger.addHandler(werkzeug_handler)
+
+app.logger.info('서버 시작')
 
 if __name__ == '__main__':
     try: