ewha-cvl commited on
Commit
95b668b
ยท
verified ยท
1 Parent(s): 5d810d4

Update admin.py

Browse files
Files changed (1) hide show
  1. admin.py +0 -17
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.")