Create R155

R155

@@ -0,0 +1,155 @@
+import gradio as gr
+import sqlite3
+import pandas as pd
+import os
+import re
+import shutil
+
+# 1️⃣ 저장소 설정
+UPLOAD_DIR = "uploaded_dbs"
+if not os.path.exists(UPLOAD_DIR):
+    os.makedirs(UPLOAD_DIR)
+
+db_registry = []
+
+def natural_sort_key(s):
+    if s is None: return []
+    return [int(text) if text.isdigit() else text.lower()
+            for text in re.split(r'(\d+)', str(s))]
+
+def refresh_registry_data():
+    global db_registry
+    db_registry = []
+    if not os.path.exists(UPLOAD_DIR): return []
+
+    db_files = [f for f in os.listdir(UPLOAD_DIR) if f.endswith(".db")]
+    for filename in db_files:
+        name_only = filename.replace(".db", "")
+        parts = name_only.split("_")
+        if len(parts) >= 2:
+            db_registry.append({
+                "path": os.path.join(UPLOAD_DIR, filename),
+                "standard": parts[0],
+                "version": parts[1]
+            })
+
+    return sorted(list(set([db["standard"] for db in db_registry])))
+
+# 2️⃣ UI 업데이트 함수들
+def on_load():
+    standards = refresh_registry_data()
+    return gr.Dropdown(choices=standards, value=None)
+
+def handle_upload(files):
+    if files is None: return "파일이 없습니다.", gr.Dropdown()
+    for file in files:
+        shutil.copy(file.name, os.path.join(UPLOAD_DIR, os.path.basename(file.name)))
+    refresh_registry_data()
+    return "파일 저장 완료!", gr.Dropdown(choices=refresh_registry_data(), value=None)
+
+def update_version_dd(standard):
+    if not standard: return gr.Dropdown(choices=[], value=None)
+    versions = sorted(list(set([db["version"] for db in db_registry if db["standard"] == standard])))
+    return gr.Dropdown(choices=versions, value=None)
+
+def update_category_dd(standard, version):
+    if not standard or not version: return gr.Dropdown(choices=[], value=None)
+
+    choices = []
+    try:
+        target_db = next(db for db in db_registry if db["standard"] == standard and db["version"] == version)
+        conn = sqlite3.connect(target_db["path"])
+        all_tables = pd.read_sql("SELECT name FROM sqlite_master WHERE type='table';", conn)['name'].tolist()
+
+        main_t = f"{standard}_{version}"
+        if main_t not in all_tables:
+            main_t = all_tables[0]
+
+        choices.append("ALL")
+
+        cols = pd.read_sql(f"PRAGMA table_info([{main_t}])", conn)['name'].tolist()
+        if 'chapter' in cols and 'category' in cols:
+            df_cat = pd.read_sql(f"SELECT DISTINCT chapter, category FROM [{main_t}]", conn)
+            cats = [f"{row['chapter']}.{row['category']}" for _, row in df_cat.iterrows()]
+            cats.sort(key=natural_sort_key)
+            choices.extend(cats)
+
+        prefix = f"{standard}_{version}_"
+        sub_tables = [t.replace(prefix, "") for t in all_tables if t != main_t]
+        choices.extend(sorted(sub_tables))
+
+        conn.close()
+        return gr.Dropdown(choices=choices, value=None)
+    except Exception as e:
+        return gr.Dropdown(choices=["ALL"], value=None)
+
+# 3️⃣ 데이터 조회 로직 (컬럼 필터링 추가)
+def display_data(standard, version, selection):
+    if not all([standard, version, selection]): return None
+    try:
+        target_db = next(db for db in db_registry if db["standard"] == standard and db["version"] == version)
+        conn = sqlite3.connect(target_db["path"])
+        all_tables = pd.read_sql("SELECT name FROM sqlite_master WHERE type='table';", conn)['name'].tolist()
+
+        main_t = f"{standard}_{version}"
+        if main_t not in all_tables: main_t = all_tables[0]
+
+        is_sub_table = False # 추가 테이블인지 확인용 플래그
+
+        if selection == "ALL":
+            df = pd.read_sql(f"SELECT * FROM [{main_t}]", conn)
+        elif "." in selection:
+            ch, ca = selection.split('.', 1)
+            df = pd.read_sql(f"SELECT * FROM [{main_t}] WHERE chapter=? AND category=?", conn, params=[ch, ca])
+        else:
+            is_sub_table = True
+            actual_table = f"{standard}_{version}_{selection}"
+            if actual_table not in all_tables:
+                actual_table = next((t for t in all_tables if t.endswith(selection)), selection)
+            df = pd.read_sql(f"SELECT * FROM [{actual_table}]", conn)
+
+        conn.close()
+
+        if not df.empty:
+            # 1. 정렬 적용
+            sort_col = 'section' if 'section' in df.columns else df.columns[0]
+            df['sort_key'] = df[sort_col].apply(natural_sort_key)
+            df = df.sort_values(by='sort_key').drop(columns=['sort_key'])
+
+            # 2. [추가 요구사항] 본문(ALL 또는 카테고리)일 때만 section, description만 남기기
+            if not is_sub_table:
+                available_cols = [c for c in ['section', 'description'] if c in df.columns]
+                df = df[available_cols]
+            else:
+                # 추가 테이블일 경우 내부 정렬용 컬럼만 제거하고 전체 유지
+                df = df[[c for c in df.columns if not c.startswith('sort_')]]
+
+        return df
+    except Exception as e:
+        return pd.DataFrame({"Error": [f"조회 실패: {str(e)}"]})
+
+# 4️⃣ UI 구성
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 📜 Regulation ")
+
+    with gr.Accordion("➕ DB 관리", open=False):
+        file_input = gr.File(label="DB 업로드", file_count="multiple", file_types=[".db"])
+        upload_btn = gr.Button("서버 저장", variant="primary")
+        upload_status = gr.Markdown("")
+
+    with gr.Row():
+        standard_dd = gr.Dropdown(label="1. 법규 선택")
+        version_dd = gr.Dropdown(label="2. Version 선택")
+        category_dd = gr.Dropdown(label="3. Category / Table 선택")
+
+    output_df = gr.Dataframe(wrap=True, interactive=False)
+
+    demo.load(on_load, None, standard_dd)
+    upload_btn.click(handle_upload, file_input, [upload_status, standard_dd])
+    standard_dd.change(update_version_dd, standard_dd, version_dd)
+    version_dd.change(update_category_dd, [standard_dd, version_dd], category_dd)
+    category_dd.change(display_data, [standard_dd, version_dd, category_dd], output_df)
+
+if __name__ == "__main__":
+
+    demo.launch(share=True)