Update app.py

app.py

@@ -184,20 +184,24 @@ def display_data(standard, version, selection):
         if main_t not in tables:
             main_t = tables[0]
 
-        # 🚀 잘린 short_name으로 원래 풀네임 테이블 찾기 (Reverse Lookup)
+        # --------------------------
+        # 테이블 reverse lookup
+        # --------------------------
         full_table_name = None
         pattern = re.compile(f"^{standard}[_\\s-]*{version}[_\\s-]*", re.IGNORECASE)
-        
+
         for t in tables:
-            if t == selection: # 풀네임이 그대로 매칭되는 경우
+            if t == selection:
                 full_table_name = t
                 break
             short = pattern.sub("", t).strip(" _")
-            if short == selection: # 드롭다운의 짧은 이름과 매칭되는 실제 DB 테이블 찾기
+            if short == selection:
                 full_table_name = t
                 break
 
-        # 테이블(A4/1 등)을 선택한 경우
+        # --------------------------
+        # 🔹 TableA/B 선택된 경우
+        # --------------------------
         if full_table_name:
             df = pd.read_sql(f"SELECT * FROM [{full_table_name}]", conn)
             df.columns = [c.strip() for c in df.columns]
@@ -210,7 +214,9 @@ def display_data(standard, version, selection):
             conn.close()
             return df
 
-        # ALL 또는 챕터.카테고리 선택한 경우
+        # --------------------------
+        # 메인 테이블 구조 파악
+        # --------------------------
         cols = pd.read_sql(f"PRAGMA table_info([{main_t}])", conn)['name'].tolist()
         lower = [c.lower().strip() for c in cols]
 
@@ -221,18 +227,44 @@ def display_data(standard, version, selection):
             elif lc == "category":
                 real_cat = c
 
+        # --------------------------
+        # 🔹 ALL 선택
+        # --------------------------
         if selection == "ALL":
-        # 🚀 무조건 메인 테이블만 사용 (TableA/B 완전 차단)
-        query = f"SELECT * FROM [{main_t}]"
-    
-        # category 컬럼 있으면 'Table' 제외
-        if real_cat:
-            query += f"""
-            WHERE IFNULL(UPPER(TRIM(CAST([{real_cat}] AS TEXT))), '') != 'TABLE'
-            """
-
-        df = pd.read_sql(query, conn)
-
+            query = f"SELECT * FROM [{main_t}]"
+
+            if real_cat:
+                query += f"""
+                WHERE IFNULL(UPPER(TRIM(CAST([{real_cat}] AS TEXT))), '') != 'TABLE'
+                """
+
+            df = pd.read_sql(query, conn)
+
+        # --------------------------
+        # 🔹 chapter.category 선택
+        # --------------------------
+        elif "." in selection and real_ch and real_cat:
+            ch, ca = selection.split(".", 1)
+
+            df = pd.read_sql(
+                f"""
+                SELECT * FROM [{main_t}]
+                WHERE REPLACE(TRIM(CAST([{real_ch}] AS TEXT)), ' ', '') = ?
+                AND REPLACE(TRIM(CAST([{real_cat}] AS TEXT)), ' ', '') = ?
+                """,
+                conn,
+                params=[ch.replace(" ", ""), ca.replace(" ", "")]
+            )
+
+        # --------------------------
+        # fallback
+        # --------------------------
+        else:
+            df = pd.read_sql(f"SELECT * FROM [{main_t}]", conn)
+
+        # --------------------------
+        # 후처리
+        # --------------------------
         df.columns = [c.lower().strip() for c in df.columns]
 
         if 'section' in df.columns and 'description' in df.columns: