Update app.py

app.py

@@ -201,7 +201,6 @@ def display_data(standard, version, selection):
         return pd.DataFrame({"Info": ["선택 필요"]})
 
     try:
-        # 실시간 경로 확인을 위해 직접 경로 생성
         db_path = os.path.join(UPLOAD_DIR, f"{standard}_{version}.db")
         if not os.path.exists(db_path):
             return pd.DataFrame({"Error": [f"파일을 찾을 수 없습니다: {db_path}"]})
@@ -218,7 +217,6 @@ def display_data(standard, version, selection):
         if main_t not in tables:
             main_t = tables[0]
 
-
         full_table_name = None
         pattern = re.compile(f"^{standard}[_\\s-]*{version}[_\\s-]*", re.IGNORECASE)
 
@@ -231,29 +229,27 @@ def display_data(standard, version, selection):
                 full_table_name = t
                 break
 
+        # --- [1] 별도 테이블 조회 (Consolidated 등 모든 열 출력) ---
         if full_table_name and full_table_name != main_t:
             df = pd.read_sql(f"SELECT * FROM [{full_table_name}]", conn)
-            
-            # version 열 제외 및 컬럼명 정리
             df = df.drop(columns=[c for c in df.columns if c.lower() == "version"], errors="ignore")
             df.columns = [c.replace("_", " ").title() for c in df.columns]
 
-            # 모든 셀에 대해 이미지 렌더링(BLOB 변환) 적용
             for col in df.columns:
                 df[col] = df[col].apply(blob_to_base64_html)
 
             conn.close()
             return df
 
-
+        # --- [2] 메인 테이블 조회 (ALL/카테고리 - 2열로 제한) ---
         cursor = conn.cursor()
         cursor.execute(f"PRAGMA table_info([{main_t}])")
         cols_info = cursor.fetchall()
         cols = [c[1] for c in cols_info]
-        lower = [c.lower().strip() for c in cols]
+        lower_cols = [c.lower().strip() for c in cols]
 
         real_ch, real_cat = None, None
-        for c, lc in zip(cols, lower):
+        for c, lc in zip(cols, lower_cols):
             if lc == "chapter": real_ch = c
             elif lc == "category": real_cat = c
 
@@ -273,14 +269,15 @@ def display_data(standard, version, selection):
         rows = cursor.fetchall()
         df = pd.DataFrame(rows, columns=cols)
 
+        # 🚨 [핵심 복구] 다시 section과 description만 남깁니다.
+        df.columns = [c.lower().strip() for c in df.columns]
         if 'section' in df.columns and 'description' in df.columns:
             df = df[['section', 'description']]
 
+        # 이미지 렌더링 적용
         for col in df.columns:
-            if 'description' in col.lower() or 'desc' in col.lower():
-                df[col] = df[col].apply(blob_to_base64_html)
-            else:
-                df[col] = df[col].fillna("").astype(str)
+            df[col] = df[col].apply(blob_to_base64_html)
+
         conn.close()
         return df