Update app.py

app.py

@@ -131,7 +131,7 @@ def load_rag_objects():
 # --- 웹 ---
 @app.route('/')
 def index():
-    return render_template('chat_v02.html')
+    return render_template('chat_v03.html')
 
 # 전역 변수에 기본값 추가
 Search_each_all_mode = True  # 기본값을 클라이언트에서 제어 가능
@@ -159,12 +159,9 @@ def handle_search_query(data):
         
         emit('search_status', {'status': 'processing', 'message': '검색 요청을 처리하는 중입니다...'})
         
-        logger.info(f"선택된 지역 : {regions}")
-        logger.info(f"선택된 법규 : {selected_regulations}")
-        if Search_each_all_mode:
-            logger.info(f"검색 모드 : 각각 검색")
-        else:
-            logger.info(f"검색 모드 : 통합 검색")
+        logger.info("선택된 지역:", regions)
+        logger.info("선택된 법규:", selected_regulations)
+        logger.info("검색 모드:", "각각 검색" if Search_each_all_mode else "통합 검색")
 
         # 법규 타입별로 필터 구분
         filters = {
@@ -175,7 +172,7 @@ def handle_search_query(data):
         }
 
         # 번역 진행 상황 알림
-        emit('search_status', {'status': 'translating', 'message': '질문을 번역하는 중입니다...'})
+        emit('search_status', {'status': 'translating', 'message': '질문에 대해 생각 중입니다...'})
         
         if session_id not in active_sessions:
             emit('search_cancelled', {'message': '검색이 취소되었습니다.'})
@@ -197,79 +194,83 @@ def handle_search_query(data):
             if os.path.exists(output_path):
                 os.remove(output_path)
                 print(f"기존 파일 삭제 완료: {output_path}")
+            
+             # 통합 검색 모드 - 타입별로 그룹화
+            grouped_regulations = group_regulations_by_type(selected_regulations)
+            emit('search_status', {'status': 'searching', 'message': f'선택된 {len(selected_regulations)}개 법규를 타입별로 통합하여 검색 중...'})
+            
+            # 타입별로 필터 생성
+            combined_filters = create_combined_filters(grouped_regulations)
+            print(f"통합 필터: {combined_filters}")
+
+            combined_cleaned_filter = {k: v for k, v in combined_filters.items() if v}
 
             if Search_each_all_mode:
                 # 각각 검색 모드
-                emit('search_status', {'status': 'searching', 'message': f'선택된 {len(selected_regulations)}개 법규를 각각 검색 중...'})
-
-                for i, regulation in enumerate(selected_regulations):
-                    if session_id not in active_sessions:
-                        emit('search_cancelled', {'message': '검색이 취소되었습니다.'})
-                        emit('search_status', {'status': 'processing', 'message': 'Ready to search'})
-                        return
-                        
-                    regulation_title = regulation.get('title', '')
-                    regulation_id = regulation.get('id', '')
-                    regulation_type = regulation.get('type', 'part')  # 타입 정보 추출
-
-                    emit('search_status', {
-                        'status': 'searching_regulation', 
-                        'message': f'법규 {i+1}/{len(selected_regulations)}: [{regulation_type.upper()}] {regulation_title} 검색 중...',
-                        'progress': (i / len(selected_regulations)) * 100
-                    })
-
-                    # 법규 타입별 필터 생성
-                    current_filters = create_filter_by_type(regulation_type, regulation_title)
-                    print(f"[{regulation_type}] 필터에 추가: {regulation_title}")
-                    print(f"생성된 필터: {current_filters}")
-
-                    Rag_Results = search_DB_from_multiple_regions(Translated_query, regions, region_rag_objects, current_filters)
-                    
-                    if session_id not in active_sessions:
-                        emit('search_cancelled', {'message': '검색이 취소되었습니다.'})
-                        emit('search_status', {'status': 'processing', 'message': 'Ready to search'})
-                        return
-                        
-                    emit('search_status', {
-                        'status': 'ai_processing', 
-                        'message': f'AI가 [{regulation_type.upper()}] {regulation_title}에 대한 답변을 생성 중...'
-                    })
-                    
-                    AImessage = RegAI(query, Rag_Results, ResultFile_FolderAddress)
-                    logger.info(f"Answer: {AImessage}")
-
-                    if session_id not in active_sessions:
-                        emit('search_cancelled', {'message': '검색이 취소되었습니다.'})
-                        return
-
-                    # 각 법규별 결과를 실시간으로 전송 (타입 정보 포함)
-                    emit('regulation_result', {
-                        'regulation_title': f"[{regulation_type.upper()}] {regulation_title}",
-                        'regulation_index': i + 1,
-                        'total_regulations': len(selected_regulations),
-                        'regulation_type': regulation_type,
-                        'result': AImessage
-                    })
-
-                    # 파일에 저장
-                    if isinstance(AImessage, str) and AImessage.strip():
-                        with open(output_path, "a", encoding="utf-8") as f:
-                            cont_selected_num += 1
-                            from datetime import datetime
-                            stamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-                            f.write(f"\n--- [{stamp}] message #{cont_selected_num} --- Regulation Type: {regulation_type} --- Regulation Name : {regulation_title} ---\n {AImessage}")
+                emit('search_status', {'status': 'searching', 'message': f'선택된 {len(combined_cleaned_filter)}개 법규를 각각 검색 중...'})
+
+                total_search_num = sum(len(v) for v in combined_cleaned_filter.values())
+                i = 0
+                for RegType, RegNames in combined_cleaned_filter.items():
+                    if RegNames:  # 값이 비어있지 않은 경우만 처리
+                        for RegName in RegNames:
+                            i = i + 1
+                            #RegType는 법규 유형, RegName은 법규 명칭
+
+                            if session_id not in active_sessions:
+                                emit('search_cancelled', {'message': '검색이 취소되었습니다.'})
+                                emit('search_status', {'status': 'processing', 'message': 'Ready to search'})
+                                return
+
+                            emit('search_status', {
+                                'status': 'searching_regulation', 
+                                'message': f'법규 {i}/{len(combined_cleaned_filter)}: {RegName} 검색 중...',
+                                'progress': (i / len(combined_cleaned_filter)) * 100
+                            })
+
+                            # 법규 타입별 필터 생성
+                            current_filters = create_filter_by_type(RegType, RegName)
+                            print(f"생성된 필터: {current_filters}")
+
+                            Rag_Results = search_DB_from_multiple_regions(Translated_query, regions, region_rag_objects, current_filters, False) #마지막 False값은 유사한 값에 대한 검색을 하지 않겠다는 의미
+
+                            if Rag_Results:
+                                if session_id not in active_sessions:
+                                    emit('search_cancelled', {'message': '검색이 취소되었습니다.'})
+                                    emit('search_status', {'status': 'processing', 'message': 'Ready to search'})
+                                    return
+                                    
+                                emit('search_status', {
+                                    'status': 'ai_processing', 
+                                    'message': f'AI가 {RegName}에 대한 답변을 생성 중...'
+                                })
+                                
+                                AImessage = RegAI(query, Rag_Results, ResultFile_FolderAddress)
+                                logger.info(f"Answer: {AImessage}")
+
+                                if session_id not in active_sessions:
+                                    emit('search_cancelled', {'message': '검색이 취소되었습니다.'})
+                                    return
+
+                                # 각 법규별 결과를 실시간으로 전송 (타입 정보 포함)
+                                emit('regulation_result', {
+                                    'regulation_title': f"[{RegName}]",
+                                    'regulation_index': i,
+                                    'total_regulations': total_search_num,
+                                    'result': AImessage
+                                })
+
+                                # 파일에 저장
+                                if isinstance(AImessage, str) and AImessage.strip():
+                                    with open(output_path, "a", encoding="utf-8") as f:
+                                        cont_selected_num += 1
+                                        from datetime import datetime
+                                        stamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                                        f.write(f"\n--- [{stamp}] message #{cont_selected_num} --- Regulation Type: {RegType} --- Regulation Name : {RegName} ---\n {AImessage}")
 
                 emit('search_complete', {'status': 'completed', 'message': '모든 법규 검색이 완료되었습니다.'})
             else:
-                # 통합 검색 모드 - 타입별로 그룹화
-                grouped_regulations = group_regulations_by_type(selected_regulations)
-                emit('search_status', {'status': 'searching', 'message': f'선택된 {len(selected_regulations)}개 법규를 타입별로 통합하여 검색 중...'})
-                
-                # 타입별로 필터 생성
-                combined_filters = create_combined_filters(grouped_regulations)
-                logger.info(f"통합 필터: {combined_filters}")
-
-                Rag_Results = search_DB_from_multiple_regions(Translated_query, regions, region_rag_objects, combined_filters)
+                Rag_Results = search_DB_from_multiple_regions(Translated_query, regions, region_rag_objects, combined_filters, True)
                 
                 if session_id in active_sessions:
                     emit('search_status', {'status': 'ai_processing', 'message': 'AI가 통합 답변을 생성 중...'})
@@ -285,7 +286,7 @@ def handle_search_query(data):
             emit('search_status', {'status': 'searching_all', 'message': '전체 법규에서 검색 중...'})
             
             # 필터 없이 검색
-            Rag_Results = search_DB_from_multiple_regions(Translated_query, regions, region_rag_objects, None)
+            Rag_Results = search_DB_from_multiple_regions(Translated_query, regions, region_rag_objects, None, True)
             
             if session_id in active_sessions:
                 emit('search_status', {'status': 'ai_processing', 'message': 'AI가 답변을 생성 중...'})
@@ -490,7 +491,7 @@ def Gemma3_AI_Translate(query_txt):
 
 # --- 검색 ---
 # 검색 함수 수정
-def search_DB_from_multiple_regions(query, selected_regions, region_rag_objects, custom_filters=None):
+def search_DB_from_multiple_regions(query, selected_regions, region_rag_objects, custom_filters=None, failsafe_mode=True):
     global Filtered_search
     global filters
     
@@ -527,7 +528,8 @@ def search_DB_from_multiple_regions(query, selected_regions, region_rag_objects,
                     query=query,
                     k=search_document_number,
                     metadata_filter=search_filters,
-                    sqlite_conn=sqlite_conn
+                    sqlite_conn=sqlite_conn,
+                    failsafe_search=failsafe_mode
                 )
             else:
                 results = reg_embedding_system.smart_search_vectorstore(
@@ -536,7 +538,8 @@ def search_DB_from_multiple_regions(query, selected_regions, region_rag_objects,
                     k=search_document_number,
                     vectorstore=vectorstore,
                     sqlite_conn=sqlite_conn,
-                    enable_detailed_search=True
+                    enable_detailed_search=True,
+                    failsafe_search=failsafe_mode
                 )
 
             print(f"[{region}] 검색 완료: {len(results)}건")