Update app.py

app.py

@@ -15,6 +15,8 @@ import re
 import traceback
 import requests # API 호출을 위해 필요
 
+from typing import Optional, Tuple, Any, Dict, List
+
 # --- Together AI SDK ---
 from together import Together
 
@@ -315,22 +317,72 @@ def handle_cancel_search(data):
 @app.route('/get_reg_list', methods=['POST'])
 def get_reg_list():
     data = request.get_json()
-    selected_regions = data.get('regions', []) or ["국내", "북미", "유럽"]
+    selected_regions = data.get('regions', [])
+
+    # 지역이 선택되지 않았으면 전체 지역으로 설정
+    if not selected_regions:
+        selected_regions = ["국내", "북미", "유럽"]
 
     all_reg_list_part = []
+    all_reg_list_section = []
+    all_reg_list_chapter = []
+    all_reg_list_jo = []
+
     for region in selected_regions:
         rag = region_rag_objects.get(region)
         if not rag:
-            continue
+            continue  # 해당 지역 RAG가 없으면 건너뜀
+
         try:
-            conn = rag["sqlite_conn"]
-            parts = reg_embedding_system.get_unique_metadata_values(conn, "regulation_part")
-            all_reg_list_part.extend(parts)
+            # 이미 로드된 SQLite 연결 재사용
+            sqlite_conn = rag["sqlite_conn"]
+            reg_list_part = get_unique_metadata_values(sqlite_conn, "regulation_part")
+            reg_list_section = get_unique_metadata_values(sqlite_conn, "regulation_section")
+            reg_list_chapter = get_unique_metadata_values(sqlite_conn, "chapter_section")
+            reg_list_jo = get_unique_metadata_values(sqlite_conn, "jo")
+
+            # 문자열이면 리스트로 변환
+            if isinstance(reg_list_part, str):
+                reg_list_part = [reg_list_part]
+            
+            if isinstance(reg_list_section, str):
+                reg_list_section = [reg_list_section]
+            
+            if isinstance(reg_list_chapter, str):
+                reg_list_chapter = [reg_list_chapter]
+            
+            if isinstance(reg_list_jo, str):
+                reg_list_jo = [reg_list_jo]
+
+            all_reg_list_part.extend(reg_list_part)
+            all_reg_list_section.extend(reg_list_section)
+            all_reg_list_chapter.extend(reg_list_chapter)
+            all_reg_list_jo.extend(reg_list_jo)
         except Exception as e:
-            logger.info(f"[{region}] 법규 로드 실패: {e}")
+            print(f"[{region}] DB 연결 오류: {e}")
+
+    # 중복 제거
+    #unique_reg_list_part = list(set(all_reg_list_part)) 
+    unique_reg_list_part = sorted(set(all_reg_list_part), key=reg_embedding_system.natural_sort_key)
+
+    #unique_reg_list_section = list(set(all_reg_list_section))
+    unique_reg_list_section = sorted(set(all_reg_list_section), key=reg_embedding_system.natural_sort_key)
 
-    unique_parts = sorted(set(all_reg_list_part), key=reg_embedding_system.natural_sort_key)
-    return jsonify(reg_list_part="\n".join(unique_parts))
+    #unique_reg_list_chapter = list(set(all_reg_list_chapter))
+    unique_reg_list_chapter = sorted(set(all_reg_list_chapter), key=reg_embedding_system.natural_sort_key)
+
+    #unique_reg_list_jo = list(set(all_reg_list_jo))
+    unique_reg_list_jo = sorted(set(all_reg_list_jo), key=reg_embedding_system.natural_sort_key)
+
+    text_result_part = "\n".join(str(item) for item in unique_reg_list_part)
+    text_result_section = "\n".join(str(item) for item in unique_reg_list_section)
+    text_result_chapter = "\n".join(str(item) for item in unique_reg_list_chapter)
+    text_result_jo = "\n".join(str(item) for item in unique_reg_list_jo)
+
+    return jsonify(reg_list_part=text_result_part, 
+                   reg_list_section=text_result_section, 
+                   reg_list_chapter=text_result_chapter, 
+                   reg_list_jo=text_result_jo)
 
 # --- SocketIO ---
 @socketio.on('connect')
@@ -566,6 +618,66 @@ def create_combined_filters(grouped_regulations):
     
     return filters
 
+def natural_sort_key(text):
+    """숫자가 포함된 문자열을 자연스럽게 정렬 (예: item1, item2, item10)"""
+    return [int(c) if c.isdigit() else c.lower() for c in re.split('([0-9]+)', str(text))]
+
+def get_unique_metadata_values(
+    sqlite_conn: sqlite3.Connection,
+    key_name: str,
+    partial_match: Optional[str] = None
+) -> List[str]:
+    """
+    SQLite 'documents' 테이블에서 특정 컬럼(key_name)의 중복되지 않은
+    모든 고유 값 리스트를 반환합니다.
+
+    Args:
+        sqlite_conn: SQLite 데이터베이스 연결 객체.
+        key_name: 고유한 값을 가져올 컬럼 이름 (예: 'regulation_name', 'part_name').
+        partial_match: (선택 사항) 해당 문자열을 포함하는 값만 검색할 때 사용.
+
+    Returns:
+        중복이 제거된 고유한 값들의 리스트.
+    """
+
+    text_result = ""
+    if not sqlite_conn:
+        print("[경고] SQLite 연결이 없어 고유 값 검색을 수행할 수 없습니다.")
+        return text_result
+
+    cursor = sqlite_conn.cursor()
+
+    # SQL 쿼리 ���성
+    # 1. 컬럼 이름에 백틱(`)을 사용하여 안전성 확보
+    # 2. DISTINCT를 사용하여 중복 제거
+
+    sql_query = f"SELECT DISTINCT `{key_name}` FROM documents"
+    params = []
+
+    # 부분 문자열 검색 (LIKE) 조건 추가
+    if partial_match:
+        sql_query += f" WHERE `{key_name}` LIKE ?"
+        params.append(f"%{partial_match}%")
+
+    try:
+        cursor.execute(sql_query, params)
+
+        # 쿼리 결과에서 첫 번째 항목 (값)만 추출
+        unique_values = [row[0] for row in cursor.fetchall() if row[0] is not None]
+        unique_values.sort(key=natural_sort_key)
+        text_result = "\n".join(str(value) for value in unique_values)
+
+        return text_result
+
+    except sqlite3.OperationalError as e:
+        # 컬럼 이름이 DB에 없을 때 발생하는 에러 처리
+        print(f"[에러] SQLite 쿼리 실행 실패 (컬럼 '{key_name}' 이름 오류 가능): {e}")
+        return text_result
+    except Exception as e:
+        print(f"[에러] 고유 값 검색 중 알 수 없는 오류 발생: {e}")
+        return text_result
+
+
 # --- 실행 ---
 if __name__ == '__main__':
     # 로컬 개발용