Update app.py

app.py

@@ -56,22 +56,16 @@ 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
         
-        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()
-        
-        if config_df.empty:
-            return pd.DataFrame({"Error": [f"Table_Config에서 '{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
-        
+        # 1. 실제 DB 테이블과 열(Column) 이름을 먼저 훑어봅니다.
         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}"
@@ -79,13 +73,37 @@ def fetch_database_records(std, ver, cat, table_type):
             main_table = valid_tables[0] if valid_tables else None
             
         if not main_table:
+            conn.close()
             return pd.DataFrame({"Error": ["데이터 테이블을 찾을 수 없습니다."]}), []
             
-        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]
         
+        # 2. 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()
+        
+        if config_df.empty:
+            return pd.DataFrame({"Error": [f"Table_Config에서 '{table_type}' 설정을 찾을 수 없습니다."]}), []
+            
+        # 💡 [핵심 해결] Table_Config에 'Main'이 여러 개일 경우, 실제 DB 열 이름과 일치하는 것을 똑똑하게 찾아냅니다!
+        matched_config = None
+        for _, row in config_df.iterrows():
+            anchors_test = [x.strip().lower() for x in str(row['Anchor_Column']).split(',')]
+            if any(a in lower_cols for a in anchors_test):
+                matched_config = row
+                break
+                
+        if matched_config is None:
+            matched_config = config_df.iloc[0] # 못 찾으면 어쩔 수 없이 첫 번째 강제 사용
+            
+        anchors = [x.strip() for x in matched_config['Anchor_Column'].split(',')]
+        displays = [x.strip() for x in matched_config['Display_Columns'].split(',')] if pd.notna(matched_config['Display_Columns']) else anchors
+        
+        # 3. 쿼리 실행 및 카테고리 필터링
+        query = f"SELECT * FROM [{main_table}]"
+        
         conditions = []
         if cat and cat != "ALL":
             if "." in cat and 'chapter' in lower_cols and 'category' in lower_cols:
@@ -103,6 +121,7 @@ def fetch_database_records(std, ver, cat, table_type):
         df = pd.read_sql(query, conn)
         conn.close()
         
+        # 4. 화면에 보여줄 열 정리
         final_cols = []
         for d in displays:
             for c in df.columns: