Update app.py

app.py

@@ -3,85 +3,51 @@ import sqlite3
 import pandas as pd
 import os
 import re
-import shutil
+from huggingface_hub import HfApi # 추가됨
 
-# 1️⃣ 저장소 설정 (폴더 안에 DB를 모으기로 했으므로 유지)
+# --- 설정부 ---
 UPLOAD_DIR = "uploaded_dbs"
-if not os.path.exists(UPLOAD_DIR):
-    os.makedirs(UPLOAD_DIR)
-
-db_registry = []
+# 내 스페이스 정보를 자동으로 가져옵니다.
+REPO_ID = os.environ.get("SPACE_ID") # 예: "아이디/스페이스이름"
+HF_TOKEN = os.environ.get("HF_TOKEN") # 쓰기 권한이 있는 토큰 필요
 
-def natural_sort_key(s):
-    if s is None: return []
-    return [int(text) if text.isdigit() else text.lower()
-            for text in re.split(r'(\d+)', str(s))]
+api = HfApi()
 
-def refresh_registry_data():
-    global db_registry
-    db_registry = []
-    if not os.path.exists(UPLOAD_DIR): return []
-
-    db_files = [f for f in os.listdir(UPLOAD_DIR) if f.endswith(".db")]
-    for filename in db_files:
-        name_only = filename.replace(".db", "")
-        parts = name_only.split("_")
-        if len(parts) >= 2:
-            db_registry.append({
-                "path": os.path.join(UPLOAD_DIR, filename),
-                "standard": parts[0],
-                "version": parts[1]
-            })
-
-    return sorted(list(set([db["standard"] for db in db_registry])))
+if not os.path.exists(UPLOAD_DIR):
+    os.makedirs(UPLOAD_DIR)
 
-# 2️⃣ UI 업데이트 함수들
-def on_load():
-    standards = refresh_registry_data()
-    return gr.Dropdown(choices=standards, value=None)
+# ... (기존 natural_sort_key, refresh_registry_data 등은 동일) ...
 
+# 2️⃣ UI 업데이트 함수 수정
 def handle_upload(files):
     if files is None: return "파일이 없습니다.", gr.Dropdown()
+    
+    saved_count = 0
     for file in files:
-        shutil.copy(file.name, os.path.join(UPLOAD_DIR, os.path.basename(file.name)))
-    refresh_registry_data()
-    return "파일 저장 완료!", gr.Dropdown(choices=refresh_registry_data(), value=None)
-
-def update_version_dd(standard):
-    if not standard: return gr.Dropdown(choices=[], value=None)
-    versions = sorted(list(set([db["version"] for db in db_registry if db["standard"] == standard])))
-    return gr.Dropdown(choices=versions, value=None)
+        filename = os.path.basename(file.name)
+        # 1. 일단 현재 실행 환경(임시)에 저장
+        local_path = os.path.join(UPLOAD_DIR, filename)
+        with open(file.name, "rb") as f_in:
+            with open(local_path, "wb") as f_out:
+                f_out.write(f_in.read())
+        
+        # 2. Hugging Face 저장소로 '영구 업로드' 실행 (Git Commit)
+        if REPO_ID and HF_TOKEN:
+            try:
+                api.upload_file(
+                    path_or_fileobj=local_path,
+                    path_in_repo=f"uploaded_dbs/{filename}", # 저장소 내 경로
+                    repo_id=REPO_ID,
+                    repo_type="space",
+                    token=HF_TOKEN
+                )
+                saved_count += 1
+            except Exception as e:
+                return f"저장소 업로드 에러: {e}", gr.Dropdown()
 
-def update_category_dd(standard, version):
-    if not standard or not version: return gr.Dropdown(choices=[], value=None)
-
-    choices = []
-    try:
-        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()
-
-        main_t = f"{standard}_{version}"
-        if main_t not in all_tables:
-            main_t = all_tables[0]
-
-        choices.append("ALL")
-
-        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()]
-            cats.sort(key=natural_sort_key)
-            choices.extend(cats)
+    refresh_registry_data()
+    return f"{saved_count}개 파일이 저장소에 영구 저장되었습니다! (잠시 후 앱이 재시작될 수 있습니다.)", gr.Dropdown(choices=refresh_registry_data(), value=None)
 
-        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()
-        return gr.Dropdown(choices=choices, value=None)
-    except Exception as e:
-        return gr.Dropdown(choices=["ALL"], value=None)
 
 # 3️⃣ 데이터 조회 로직
 def display_data(standard, version, selection):
@@ -148,5 +114,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     category_dd.change(display_data, [standard_dd, version_dd, category_dd], output_df)
 
 if __name__ == "__main__":
-    # 허깅페이스 배포 시에는 share=True가 필요 없으므로 제거합니다.
+
     demo.launch()