Update app.py

app.py

@@ -36,6 +36,14 @@ def blob_to_base64_html(blob_data):
     except Exception as e:
         return str(blob_data)
 
+# 🚀 [핵심 추가] 텍스트 디코딩 에러 방지용 팩토리 함수
+def safe_text_factory(x):
+    """텍스트 디코딩을 시도하고, 실패하면(PNG 등 이미지일 경우) 원본 바이트를 그대로 반환"""
+    try:
+        return x.decode('utf-8')
+    except UnicodeDecodeError:
+        return x
+
 def natural_sort_key(s):
     if s is None:
         return []
@@ -82,6 +90,9 @@ def update_category_dd(standard, version):
     try:
         target = next(d for d in db_registry if d["standard"] == standard and d["version"] == version)
         conn = sqlite3.connect(target["path"])
+        
+        # 🚀 [핵심 추가] 카테고리 불러올 때도 이미지 에러 차단
+        conn.text_factory = safe_text_factory 
 
         tables = pd.read_sql("SELECT name FROM sqlite_master WHERE type='table';", conn)['name'].tolist()
         main_t = f"{standard}_{version}"
@@ -101,7 +112,7 @@ def update_category_dd(standard, version):
             for _, r in df.iterrows():
                 choices.append(f"{str(r[ch]).strip()}.{str(r[ca]).strip()}")
 
-        # 2. 🚀 테이블 이름 깔끔하게 자르기 (강력한 정규식 적용)
+        # 2. 테이블 이름 깔끔하게 자르기 (강력한 정규식 적용)
         pattern = re.compile(f"^{standard}[_\\s-]*{version}[_\\s-]*", re.IGNORECASE)
 
         for t in tables:
@@ -141,7 +152,7 @@ def highlight_diff(base, comp):
 
     d = list(difflib.ndiff(b_words, c_words))
 
-    b_res, c_res = [], []
+    b_res, c_res = []
     i = 0
 
     while i < len(d):
@@ -178,6 +189,9 @@ def display_data(standard, version, selection):
     try:
         target = next(d for d in db_registry if d["standard"] == standard and d["version"] == version)
         conn = sqlite3.connect(target["path"])
+        
+        # 🚀 [핵심 추가] 데이터 조회 시 이미지 에러 완벽 차단!
+        conn.text_factory = safe_text_factory 
 
         tables = pd.read_sql(
             "SELECT name FROM sqlite_master WHERE type='table';",
@@ -245,14 +259,8 @@ def display_data(standard, version, selection):
         # 🔹 ALL 선택
         # --------------------------
         if selection == "ALL":
-            query = f"SELECT * FROM [{main_t}]"
-
-            if real_cat:
-                query += f"""
-                WHERE IFNULL(UPPER(TRIM(CAST([{real_cat}] AS TEXT))), '') != 'TABLE'
-                """
-
-            cursor.execute(query)
+            # 🚀 [수정] 텍스트 팩토리로 에러를 잡았으므로, 불필요한 필터 조건 삭제
+            cursor.execute(f"SELECT * FROM [{main_t}]")
 
         # --------------------------
         # 🔹 chapter.category
@@ -309,7 +317,7 @@ def apply_diff_row(row, base_col, comp_col):
     if "<img" in base_val or "<img" in comp_val:
         return pd.Series([base_val, comp_val])
 
-    # 🚀 2. [핵심] 두 텍스트가 완전히 똑같으면 무거운 diff 연산 생략 (Fast-path)
+    # 2. [핵심] 두 텍스트가 완전히 똑같으면 무거운 diff 연산 생략 (Fast-path)
     if base_val == comp_val:
         return pd.Series([base_val, comp_val])
 
@@ -359,7 +367,7 @@ def unified_search(bs, bv, bc, cs, cv, cc):
         # NaN 값을 빈 문자열로 처리
         merged = merged.fillna("")
 
-        # 🚀 최적화 포인트: iterrows() 대신 apply() 사용
+        # 최적화 포인트: iterrows() 대신 apply() 사용
         merged[[base_col, comp_col]] = merged.apply(
             lambda row: apply_diff_row(row, base_col, comp_col),
             axis=1
@@ -423,7 +431,7 @@ with gr.Blocks() as demo:
     comp_reset_btn.click(reset_comp, None, [comp_standard, comp_version, comp_category])
 
 # --------------------------
-# CSS (에러 수정됨)
+# CSS
 # --------------------------
 css = """
 table {