Update app.py

app.py

@@ -83,34 +83,31 @@ def update_version_dd(standard):
     return gr.Dropdown(choices=versions)
 
 def update_category_dd(standard, version):
-    print("DEBUG category:", standard, version, flush=True)
-
     if not standard or not version:
         return gr.update(choices=[], value=None, interactive=False)
 
     choices = ["ALL"]
+    
+    # db_registry 대신 실시간 파일 경로 생성
+    db_path = os.path.join(UPLOAD_DIR, f"{standard}_{version}.db")
+
+    # 파일이 존재하지 않으면 에러 내지 않고 ALL만 띄움
+    if not os.path.exists(db_path):
+        print(f"[경고] {db_path} 파일이 폴더에 없습니다!")
+        return gr.update(choices=choices, value=None, interactive=True)
 
     try:
-        target = next(d for d in db_registry if d["standard"] == standard and d["version"] == version)
-        conn = sqlite3.connect(target["path"])
+        conn = sqlite3.connect(db_path)
         conn.text_factory = safe_text_factory 
 
-        tables = pd.read_sql(
-            "SELECT name FROM sqlite_master WHERE type='table';",
-            conn
-        )['name'].tolist()
-
-        print("TABLES:", tables, flush=True)
-
+        tables = pd.read_sql("SELECT name FROM sqlite_master WHERE type='table';", conn)['name'].tolist()
         main_t = f"{standard}_{version}"
-        print("MAIN TABLE:", main_t, flush=True)
-
+        
         if main_t not in tables:
-            print("⚠️ main table not found → fallback", flush=True)
             main_t = tables[0] if tables else None
 
         if not main_t:
-            return gr.update(choices=[], value=None, interactive=False)
+            return gr.update(choices=choices, value=None, interactive=True)
 
         cols = pd.read_sql(f"PRAGMA table_info([{main_t}])", conn)['name'].tolist()
         lower = [c.lower() for c in cols]
@@ -118,23 +115,20 @@ def update_category_dd(standard, version):
         if 'chapter' in lower and 'category' in lower:
             ch = cols[lower.index('chapter')]
             ca = cols[lower.index('category')]
-        
-            df = pd.read_sql(
-                f"SELECT DISTINCT [{ch}], [{ca}] FROM [{main_t}]",
-                conn
-            )
-        
+    
+            df = pd.read_sql(f"SELECT DISTINCT [{ch}], [{ca}] FROM [{main_t}]", conn)
+    
             for _, r in df.iterrows():
                 chapter = str(r[ch]).strip()
                 category = str(r[ca]).strip()
-        
-                if chapter and category:
+                # None이나 빈 값 필터링
+                if chapter and category and chapter.lower() != 'none' and category.lower() != 'none':
                     choices.append(f"{chapter}.{category}")
 
         conn.close()
 
     except Exception as e:
-        print(f"[Error in update_category_dd] standard: {standard}, version: {version}")
+        print(f"[Error in update_category_dd] {e}")
         traceback.print_exc()
 
     return gr.update(
@@ -453,32 +447,34 @@ def update_comp_standard(base_std, base_ver):
         return gr.Dropdown(choices=[], interactive=False)
 
 def update_comp_version(base_std, base_ver, comp_std):
-    if not base_std or not base_ver or not comp_std:
+    if not comp_std:
         return gr.update(choices=[], value=None, interactive=False)
 
     try:
-        conn = sqlite3.connect("mapping.db")
-        query = """
-        SELECT DISTINCT Comp_ver
-        FROM Mapping_registry
-        WHERE Base_std=? AND Base_ver=? AND Comp_std=?
-        """
+        # 1. 매핑 DB 우선 조회
+        conn = sqlite3.connect(os.path.join(UPLOAD_DIR, "mapping.db"))
+        query = "SELECT DISTINCT Comp_ver FROM Mapping_registry WHERE Base_std=? AND Base_ver=? AND Comp_std=?"
         df = pd.read_sql(query, conn, params=[base_std, base_ver, comp_std])
         conn.close()
 
-        if df.empty:
+        choices = sorted(df['Comp_ver'].dropna().unique().tolist())
+        
+        # 2. 만약 매핑 DB에서 버전을 못 찾았다면, 실제 폴더를 뒤져서 찾아옴 (실시간)
+        if not choices:
+            for f in os.listdir(UPLOAD_DIR):
+                if f.startswith(comp_std + "_") and f.endswith(".db"):
+                    ver = f.replace(comp_std + "_", "").replace(".db", "")
+                    choices.append(ver)
+            choices = sorted(list(set(choices)))
+
+        if not choices:
             return gr.update(choices=[], value=None, interactive=False)
 
-        return gr.update(
-            choices=sorted(df['Comp_ver'].tolist()),
-            value=None,
-            interactive=True
-        )
+        return gr.update(choices=choices, value=None, interactive=True)
 
     except Exception as e:
         print("[update_comp_version error]", e)
         return gr.update(choices=[], value=None, interactive=False)
-
 # --------------------------
 # UI
 # --------------------------