Update app.py

app.py

@@ -64,21 +64,48 @@ def update_category_dd(standard, version):
         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()
+        
+        # 1. 메인 테이블 결정 로직 강화
         main_t = f"{standard}_{version}"
-        if main_t not in all_tables: main_t = all_tables[0]
+        if main_t not in all_tables:
+            # 파일명과 완벽히 일치하는 테이블이 없으면, standard와 version을 포함하는 가장 짧은 이름을 선택
+            candidates = [t for t in all_tables if standard in t and version in t]
+            main_t = min(candidates, key=len) if candidates else all_tables[0]
         
-        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()]
+        # 2. 컬럼 정보 가져오기 (대소문자 무시 처리)
+        cols_info = pd.read_sql(f"PRAGMA table_info([{main_t}])", conn)
+        actual_cols = cols_info['name'].tolist()
+        lower_cols = [c.lower() for c in actual_cols] # 비교용 소문자 리스트
+
+        # 3. Chapter / Category 목록 생성
+        if 'chapter' in lower_cols and 'category' in lower_cols:
+            # 실제 DB에 저장된 대소문자 그대로의 컬럼명을 찾음
+            col_chapter = actual_cols[lower_cols.index('chapter')]
+            col_category = actual_cols[lower_cols.index('category')]
+            
+            # DISTINCT 쿼리로 목록 추출
+            df_cat = pd.read_sql(f"SELECT DISTINCT [{col_chapter}], [{col_category}] FROM [{main_t}]", conn)
+            
+            cats = []
+            for _, row in df_cat.iterrows():
+                ch = str(row[col_chapter]) if row[col_chapter] is not None else ""
+                ca = str(row[col_category]) if row[col_category] is not None else ""
+                if ch or ca: # 둘 중 하나라도 내용이 있으면 추가
+                    cats.append(f"{ch}.{ca}")
+            
             cats.sort(key=natural_sort_key)
             choices.extend(cats)
         
+        # 4. 부속 테이블(Annex 등) 추가
         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()
-    except: pass
+    except Exception as e:
+        print(f"Error: {e}")
+        pass
+        
     return gr.Dropdown(choices=choices, value=None)
 
 # 3️⃣ 데이터 조회 로직 (정렬 로직 제거 버전)