Spaces:
Runtime error
Runtime error
Update admin.py
Browse files
admin.py
CHANGED
|
@@ -6,9 +6,6 @@ import zipfile
|
|
| 6 |
import shutil
|
| 7 |
from typing import List, Tuple
|
| 8 |
|
| 9 |
-
# ์ด ๋ชจ๋์ "๋ณด๊ธฐ์ฉ"๋ง ๋ด๋นํ๋ค. (์ ์ฅ์ app.py์ save_imgs๊ฐ ํจ)
|
| 10 |
-
# app.py์ ์ ์ฅ ๊ฒฝ๋ก(/data/ํ์์คํฌํ-์ด๋ฉ์ผ)์ ์ผ์นํ๋๋ก /data ๋ฐ๋ก ์๋๋ฅผ root๋ก ์ผ๋๋ค.
|
| 11 |
-
|
| 12 |
ALLOWED_IMAGE_EXTS = {".jpg", ".jpeg", ".png", ".tif", ".tiff"}
|
| 13 |
_ADMIN_PASSWORD = os.environ.get("ADMIN_PASSWORD", "") # Hugging Face Space Secret ๊ถ์ฅ
|
| 14 |
|
|
@@ -16,18 +13,12 @@ def _persistent_base_dir() -> str:
|
|
| 16 |
return os.environ.get("HF_PERSISTENT_DIR", "/data")
|
| 17 |
|
| 18 |
def _data_root() -> str:
|
| 19 |
-
# app.py์ save_imgs๊ฐ /data ๋ฐ๋ก ์๋์ ์ธ์
ํด๋๋ฅผ ๋ง๋ค๋ฏ๋ก ๊ทธ๋๋ก ๋ง์ถ๋ค.
|
| 20 |
return _persistent_base_dir()
|
| 21 |
|
| 22 |
def verify_password(input_pw: str) -> bool:
|
| 23 |
-
"""๊ด๋ฆฌ์ ๋น๋ฐ๋ฒํธ ๊ฒ์ฆ (ํ์ด๋ฐ ๊ณต๊ฒฉ ๋ฐฉ์ง ๋น๊ต)"""
|
| 24 |
return bool(_ADMIN_PASSWORD) and hmac.compare_digest(input_pw or "", _ADMIN_PASSWORD)
|
| 25 |
|
| 26 |
def list_sessions() -> Tuple[List[str], str]:
|
| 27 |
-
"""
|
| 28 |
-
/data ์๋์ ์ธ์
ํด๋ ๋ชฉ๋ก์ ๋ฐํํ๋ค.
|
| 29 |
-
๋ฐํ: (ํด๋๋ช
๋ฆฌ์คํธ(์ต์ ์), ์ํ ๋ฉ์์ง)
|
| 30 |
-
"""
|
| 31 |
root = _data_root()
|
| 32 |
os.makedirs(root, exist_ok=True)
|
| 33 |
try:
|
|
@@ -42,10 +33,6 @@ def list_sessions() -> Tuple[List[str], str]:
|
|
| 42 |
return [], f"Error: Failed to list sessions. {e}"
|
| 43 |
|
| 44 |
def session_info(session: str) -> Tuple[str, List[str]]:
|
| 45 |
-
"""
|
| 46 |
-
์ ํํ ์ธ์
์ ๋ฉํ์ ๋ณด(markdown)์ ์ด๋ฏธ์ง ๊ฒฝ๋ก ๋ฆฌ์คํธ๋ฅผ ๋ฐํํ๋ค.
|
| 47 |
-
๋ฐํ: (info_markdown, image_paths)
|
| 48 |
-
"""
|
| 49 |
root = _data_root()
|
| 50 |
if not session:
|
| 51 |
return "Error: No session selected.", []
|
|
@@ -64,7 +51,6 @@ def session_info(session: str) -> Tuple[str, List[str]]:
|
|
| 64 |
except Exception as e:
|
| 65 |
meta_text = f"(Failed to read meta.json: {e})"
|
| 66 |
|
| 67 |
-
# ์๋ณธ ์ด๋ฏธ์ง ์ ์ฒด ์์ง (ํ์ฅ์ ๊ธฐ์ค)
|
| 68 |
originals = []
|
| 69 |
for name in os.listdir(path):
|
| 70 |
p = os.path.join(path, name)
|
|
@@ -84,9 +70,6 @@ def session_info(session: str) -> Tuple[str, List[str]]:
|
|
| 84 |
return info_md
|
| 85 |
|
| 86 |
def zip_session(session: str) -> str:
|
| 87 |
-
"""
|
| 88 |
-
์ธ์
ํด๋๋ฅผ ZIP์ผ๋ก ๋ง๋ค์ด /tmp/{session}.zip ๊ฒฝ๋ก๋ฅผ ๋ฐํํ๋ค.
|
| 89 |
-
"""
|
| 90 |
root = _data_root()
|
| 91 |
if not session:
|
| 92 |
raise ValueError("Error: No session selected.")
|
|
|
|
| 6 |
import shutil
|
| 7 |
from typing import List, Tuple
|
| 8 |
|
|
|
|
|
|
|
|
|
|
| 9 |
ALLOWED_IMAGE_EXTS = {".jpg", ".jpeg", ".png", ".tif", ".tiff"}
|
| 10 |
_ADMIN_PASSWORD = os.environ.get("ADMIN_PASSWORD", "") # Hugging Face Space Secret ๊ถ์ฅ
|
| 11 |
|
|
|
|
| 13 |
return os.environ.get("HF_PERSISTENT_DIR", "/data")
|
| 14 |
|
| 15 |
def _data_root() -> str:
|
|
|
|
| 16 |
return _persistent_base_dir()
|
| 17 |
|
| 18 |
def verify_password(input_pw: str) -> bool:
|
|
|
|
| 19 |
return bool(_ADMIN_PASSWORD) and hmac.compare_digest(input_pw or "", _ADMIN_PASSWORD)
|
| 20 |
|
| 21 |
def list_sessions() -> Tuple[List[str], str]:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
root = _data_root()
|
| 23 |
os.makedirs(root, exist_ok=True)
|
| 24 |
try:
|
|
|
|
| 33 |
return [], f"Error: Failed to list sessions. {e}"
|
| 34 |
|
| 35 |
def session_info(session: str) -> Tuple[str, List[str]]:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
root = _data_root()
|
| 37 |
if not session:
|
| 38 |
return "Error: No session selected.", []
|
|
|
|
| 51 |
except Exception as e:
|
| 52 |
meta_text = f"(Failed to read meta.json: {e})"
|
| 53 |
|
|
|
|
| 54 |
originals = []
|
| 55 |
for name in os.listdir(path):
|
| 56 |
p = os.path.join(path, name)
|
|
|
|
| 70 |
return info_md
|
| 71 |
|
| 72 |
def zip_session(session: str) -> str:
|
|
|
|
|
|
|
|
|
|
| 73 |
root = _data_root()
|
| 74 |
if not session:
|
| 75 |
raise ValueError("Error: No session selected.")
|