Update app.py

app.py

@@ -72,14 +72,6 @@ def refresh_registry_data():
 def on_load():
     return gr.Dropdown(choices=refresh_registry_data())
 
-def update_version_dd(standard):
-    if not standard:
-        return gr.Dropdown(choices=[])
-    versions = sorted(set(
-        d["version"] for d in db_registry if d["standard"] == standard
-    ))
-    return gr.Dropdown(choices=versions)
-
 def update_category_dd(standard, version):
     print("DEBUG category:", standard, version, flush=True)
 
@@ -91,31 +83,24 @@ def update_category_dd(standard, version):
     try:
         target = next(d for d in db_registry if d["standard"] == standard and d["version"] == version)
         conn = sqlite3.connect(target["path"])
-        
         conn.text_factory = safe_text_factory 
 
-        tables = pd.read_sql("SELECT name FROM sqlite_master WHERE type='table';", conn)['name'].tolist()
+        tables = pd.read_sql(
+            "SELECT name FROM sqlite_master WHERE type='table';",
+            conn
+        )['name'].tolist()
 
-        print("TABLES:", tables, flush=True)   # 👈 이것도 추가
+        print("TABLES:", tables, flush=True)
 
         main_t = f"{standard}_{version}"
+        print("MAIN TABLE:", main_t, flush=True)
 
-        print("MAIN TABLE:", main_t, flush=True)  # 👈 이것도 추가
-    if not standard or not version:
-        return gr.Dropdown(choices=[])
-
-    choices = ["ALL"]
-
-    try:
-        target = next(d for d in db_registry if d["standard"] == standard and d["version"] == version)
-        conn = sqlite3.connect(target["path"])
-          
-        conn.text_factory = safe_text_factory 
-
-        tables = pd.read_sql("SELECT name FROM sqlite_master WHERE type='table';", conn)['name'].tolist()
-        main_t = f"{standard}_{version}"
         if main_t not in tables:
-            main_t = tables[0]
+            print("⚠️ main table not found → fallback", flush=True)
+            main_t = tables[0] if tables else None
+
+        if not main_t:
+            return gr.Dropdown(choices=[])
 
         cols = pd.read_sql(f"PRAGMA table_info([{main_t}])", conn)['name'].tolist()
         lower = [c.lower() for c in cols]
@@ -124,7 +109,10 @@ def update_category_dd(standard, version):
             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():
                 choices.append(f"{str(r[ch]).strip()}.{str(r[ca]).strip()}")