Update app.py

app.py

@@ -5,7 +5,6 @@ import os
 import re
 import base64
 import difflib
-import traceback
 
 # ==========================================
 # 1. Environment Setup & Data Helpers
@@ -57,15 +56,12 @@ def fetch_available_standards():
 
     return sorted(list(set([d["standard"] for d in db_registry])))
 
-# ==========================================
-# 💡 [복구된 핵심 함수] 이 부분을 추가해 주세요!
-# ==========================================
 def fetch_database_records(std, ver, cat, table_type):
     try:
         db_path = os.path.join(UPLOAD_DIR, f"{std}_{ver}.db")
         conn = sqlite3.connect(db_path)
+        conn.text_factory = decode_sqlite_text
         
-        # Table_Config 읽어오기
         conn_map = sqlite3.connect(os.path.join(UPLOAD_DIR, "mapping.db"))
         config_df = pd.read_sql("SELECT * FROM Table_Config WHERE TRIM(Table_Type)=?", conn_map, params=[table_type.strip()])
         conn_map.close()
@@ -76,7 +72,6 @@ def fetch_database_records(std, ver, cat, table_type):
         anchors = [x.strip() for x in config_df.iloc[0]['Anchor_Column'].split(',')]
         displays = [x.strip() for x in config_df.iloc[0]['Display_Columns'].split(',')] if pd.notna(config_df.iloc[0]['Display_Columns']) else anchors
         
-        # 메인 테이블 찾기
         tables = pd.read_sql("SELECT name FROM sqlite_master WHERE type='table';", conn)['name'].tolist()
         valid_tables = [t for t in tables if t.lower() not in ['mapping_registry', 'mapping_table', 'table_config', 'sqlite_sequence']]
         main_table = f"{std}_{ver}"
@@ -88,7 +83,6 @@ def fetch_database_records(std, ver, cat, table_type):
             
         query = f"SELECT * FROM [{main_table}]"
         
-        # 카테고리 필터링
         cols = pd.read_sql(f"PRAGMA table_info([{main_table}])", conn)['name'].tolist()
         lower_cols = [c.lower() for c in cols]
         
@@ -109,7 +103,6 @@ def fetch_database_records(std, ver, cat, table_type):
         df = pd.read_sql(query, conn)
         conn.close()
         
-        # 화면에 보여줄 열 정리
         final_cols = []
         for d in displays:
             for c in df.columns:
@@ -122,6 +115,9 @@ def fetch_database_records(std, ver, cat, table_type):
         if not final_cols:
             return df, real_anchors
             
+        for c in final_cols:
+            df[c] = df[c].apply(convert_blob_to_html_img)
+            
         return df[final_cols], real_anchors
         
     except Exception as e:
@@ -133,9 +129,6 @@ def generate_html_diff(text1, text2):
     try:
         s1, s2 = str(text1), str(text2)
         
-        # ========================================================
-        # 🛡️ [안전 장치 1] 글자가 너무 길면 (예: 1000자 이상) 컴퓨터 터짐 방지
-        # ========================================================
         if len(s1) > 1000 or len(s2) > 1000:
             return s1, s2
             
@@ -143,17 +136,13 @@ def generate_html_diff(text1, text2):
         if not words1 or not words2: 
             return s1, s2
             
-        # ========================================================
-        # 🛡️ [안전 장치 2] 한글 vs 영어처럼 겹치는 단어가 5%도 안 되면 비교 생략
-        # ========================================================
         common_words = set(words1) & set(words2)
         if len(common_words) / min(len(words1), len(words2)) < 0.05:
             return s1, s2
 
-        # 안전 검사를 통과한 진짜 비슷한 텍스트들만 초고속 비교!
         matcher = difflib.SequenceMatcher(None, words1, words2)
-        
         res1, res2 = [], []
+        
         for tag, i1, i2, j1, j2 in matcher.get_opcodes():
             if tag == 'replace':
                 res1.append(f"<span style='color:#ff4d4f; font-weight:bold;'>{' '.join(words1[i1:i2])}</span>")
@@ -169,10 +158,10 @@ def generate_html_diff(text1, text2):
         return " ".join(res1), " ".join(res2)
     except Exception:
         return text1, text2
+
 # ==========================================
 # 2. UI Component Handlers
 # ==========================================
-
 def load_initial_standards():
     return gr.Dropdown(choices=fetch_available_standards())
 
@@ -182,13 +171,12 @@ def update_version_dropdown(standard):
     versions = sorted(set(d["version"] for d in db_registry if d["standard"] == standard))
     return gr.Dropdown(choices=versions)
 
-# ⬇️ [수정] Version이 선택되면 Category와 함께 Status 값도 같이 리턴합니다.
 def update_base_category_dropdown(standard, version):
     if not standard or not version:
         return gr.update(choices=[], value=None, interactive=False), gr.update(value="")
 
     choices = ["ALL"]
-    status_value = "" # Status 기본값
+    status_value = ""
     
     db_path = os.path.join(UPLOAD_DIR, f"{standard}_{version}.db")
     if not os.path.exists(db_path): 
@@ -197,15 +185,12 @@ def update_base_category_dropdown(standard, version):
     try:
         conn = sqlite3.connect(db_path)
         tables = pd.read_sql("SELECT name FROM sqlite_master WHERE type='table';", conn)['name'].tolist()
-        
         valid_tables = [t for t in tables if t.lower() not in ['mapping_registry', 'mapping_table', 'table_config', 'sqlite_sequence']]
-        
         main_table = f"{standard}_{version}"
         if main_table not in valid_tables:
             main_table = valid_tables[0] if valid_tables else None
 
         if main_table:
-            # 💡 [핵심 추가] DB에서 Status 값을 읽어옵니다.
             try:
                 cols_check = pd.read_sql(f"PRAGMA table_info([{main_table}])", conn)['name'].tolist()
                 if "Status" in cols_check or "status" in cols_check:
@@ -239,8 +224,7 @@ def update_base_category_dropdown(standard, version):
 
         conn.close()
     except Exception: 
-        import traceback
-        traceback.print_exc()
+        pass
         
     return gr.update(choices=choices, value=None, interactive=True), gr.update(value=status_value)
 
@@ -274,7 +258,6 @@ def update_comp_version_dropdown(base_std, base_ver, comp_std):
     except Exception:
         return gr.update(choices=[], value=None, interactive=False)
 
-# ⬇️ [수정] 비교 법규에서도 Category 갱신 시 Status를 읽어옵니다.
 def update_comp_category_dropdown(base_std, base_ver, base_cat, comp_std, comp_ver):
     if not all([base_std, base_ver, base_cat, comp_std, comp_ver]):
         return gr.update(choices=[], value=None, interactive=False), gr.update(value="")
@@ -283,7 +266,6 @@ def update_comp_category_dropdown(base_std, base_ver, base_cat, comp_std, comp_v
     status_value = ""
 
     try:
-        # Status 읽기
         comp_db_path = os.path.join(UPLOAD_DIR, f"{comp_std}_{comp_ver}.db")
         if os.path.exists(comp_db_path):
             c_conn = sqlite3.connect(comp_db_path)
@@ -324,7 +306,6 @@ def update_comp_category_dropdown(base_std, base_ver, base_cat, comp_std, comp_v
         if "Main" in allowed_types:
             final_choices.append("ALL")
             if os.path.exists(comp_db_path):
-                # c_conn is already closed above, reopen if needed
                 c_conn = sqlite3.connect(comp_db_path)
                 if c_main_table:
                     cols = pd.read_sql(f"PRAGMA table_info([{c_main_table}])", c_conn)['name'].tolist()
@@ -346,20 +327,18 @@ def update_comp_category_dropdown(base_std, base_ver, base_cat, comp_std, comp_v
 
         return gr.update(choices=final_choices, value=final_choices[0] if final_choices else None, interactive=True), gr.update(value=status_value)
         
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
+    except Exception:
         return gr.update(choices=[], value=None), gr.update(value="")
 
 def reset_base_selections():
-    # Status까지 4개를 리셋
     return gr.update(value=None), gr.update(choices=[], value=None), gr.update(choices=[], value=None), gr.update(value="")
 
 def reset_comp_selections():
-    # Status까지 4개를 리셋
     return gr.update(value=None), gr.update(choices=[], value=None), gr.update(choices=[], value=None), gr.update(value="")
 
-
+# ==========================================
+# 3. Core Search Logic
+# ==========================================
 def execute_unified_search(base_std, base_ver, base_cat, comp_std, comp_ver, comp_cat, mapped_only, diff_only):
     try:
         def get_type_by_cat(cat):
@@ -433,9 +412,6 @@ def execute_unified_search(base_std, base_ver, base_cat, comp_std, comp_ver, com
             df_base['merge_key'] = df_base[real_anchors_b].apply(lambda row: '-'.join([clean_key_val(x) for x in row]), axis=1)
             df_comp['merge_key'] = df_comp[real_anchors_c].apply(lambda row: '-'.join([clean_key_val(x) for x in row]), axis=1)
 
-            # =========================================================
-            # 💡 [방어막 1] Target_Table 이름 깔끔하게 가져오기
-            # =========================================================
             conn_map = sqlite3.connect(os.path.join(UPLOAD_DIR, "mapping.db"))
             registry_query = """
                 SELECT Target_Table FROM Mapping_registry 
@@ -450,9 +426,6 @@ def execute_unified_search(base_std, base_ver, base_cat, comp_std, comp_ver, com
                 if val and val.lower() not in ["none", "nan"]:
                     target_table_name = val
 
-            # =========================================================
-            # 💡 [방어막 2] 매핑 장부 안전하게 읽어오기 (오타나면 에러 띄움)
-            # =========================================================
             try:
                 q_fw = f"SELECT * FROM [{target_table_name}] WHERE TRIM(Base_std)=? AND TRIM(Base_ver)=? AND TRIM(Comp_std)=? AND TRIM(Comp_ver)=?"
                 df_fw = pd.read_sql(q_fw, conn_map, params=[base_std.strip(), base_ver.strip(), comp_std.strip(), comp_ver.strip()])
@@ -501,9 +474,6 @@ def execute_unified_search(base_std, base_ver, base_cat, comp_std, comp_ver, com
                 df_mapping['Base_section'] = df_mapping['Base_section'].apply(clean_key_val)
                 df_mapping['Comp_section'] = df_mapping['Comp_section'].apply(clean_key_val)
                 
-                # =========================================================
-                # 💣 [방어막 3] 카테시안 폭발 방지: 빈칸끼리 연결된 가짜 매핑 삭제
-                # =========================================================
                 df_mapping = df_mapping[(df_mapping['Base_section'] != "") & (df_mapping['Comp_section'] != "")]
                 df_mapping = df_mapping.dropna().drop_duplicates()
             else:
@@ -575,9 +545,6 @@ def execute_unified_search(base_std, base_ver, base_cat, comp_std, comp_ver, com
 
         return pd.DataFrame({"Info": ["조건을 선택하세요."]})
     except Exception as e:
-        import traceback
-        traceback.print_exc()
-        # 시스템 에러 대신 화면에 깔끔하게 원인을 띄워줍니다.
         error_msg = str(e)
         if "database is locked" in error_msg.lower():
             return pd.DataFrame({"Error": ["🚨 DB가 잠겨있습니다! 켜놓으신 'DB Browser' 프로그램을 완전히 종료한 뒤 다시 조회해 주세요."]})
@@ -598,7 +565,6 @@ with gr.Blocks() as demo:
                 
                 with gr.Row(elem_classes="reset-row"):
                     base_category = gr.Dropdown(label="Category", scale=4)
-                    # ⬇️ elem_classes="reset-btn" 추가!
                     base_reset_btn = gr.Button("↺ 초기화", scale=1, elem_classes="reset-btn")
     
         with gr.Accordion("🔄 비교 법규", open=False):
@@ -609,12 +575,10 @@ with gr.Blocks() as demo:
                 
                 with gr.Row(elem_classes="reset-row"):
                     comp_category = gr.Dropdown(label="Category", choices=[], scale=4)
-                    # ⬇️ elem_classes="reset-btn" 추가!
                     comp_reset_btn = gr.Button("↺ 초기화", scale=1, elem_classes="reset-btn")
 
     with gr.Row(elem_id="search_row"):
         search_btn = gr.Button("🔍 조회", variant="primary", scale=10)
-        # ⬇️ [추가] 매핑 체크박스 추가!
         mapped_only_cb = gr.Checkbox(label="🔗 매핑된 항목만 보기", value=False, elem_id="mapped_cb_item", container=False, scale=1)
         diff_filter_cb = gr.Checkbox(label="💡 변경된 내용만 보기", value=False, elem_id="diff_cb_item", container=False, scale=1)