wiizm commited on
Commit
81f6928
ยท
verified ยท
1 Parent(s): 372ce50

Upload app\huggingface_client.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app//huggingface_client.py +52 -0
app//huggingface_client.py ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Hugging Face ํ† ํฐ ๊ด€๋ฆฌ ๋ชจ๋“ˆ
3
+ Google Gemini API ํ‚ค ๊ด€๋ฆฌ ๋ฐฉ์‹๊ณผ ๋™์ผํ•œ ํŒจํ„ด์œผ๋กœ ๊ตฌํ˜„
4
+ """
5
+
6
+ import os
7
+ from typing import Optional
8
+
9
+ def get_huggingface_token() -> str:
10
+ """
11
+ Hugging Face ํ† ํฐ ๊ฐ€์ ธ์˜ค๊ธฐ (ํ™˜๊ฒฝ ๋ณ€์ˆ˜ ์šฐ์„ , ์—†์œผ๋ฉด DB์—์„œ)
12
+
13
+ Returns:
14
+ Hugging Face ํ† ํฐ ๋ฌธ์ž์—ด (์—†์œผ๋ฉด ๋นˆ ๋ฌธ์ž์—ด)
15
+ """
16
+ # ํ™˜๊ฒฝ ๋ณ€์ˆ˜์—์„œ ๋จผ์ € ํ™•์ธ
17
+ token = os.getenv('HUGGINGFACE_HUB_TOKEN', '').strip()
18
+ if not token:
19
+ token = os.getenv('HF_TOKEN', '').strip()
20
+
21
+ if token:
22
+ print(f"[Hugging Face] ํ™˜๊ฒฝ ๋ณ€์ˆ˜์—์„œ ํ† ํฐ ๊ฐ€์ ธ์˜ด (๊ธธ์ด: {len(token)}์ž)")
23
+ return token
24
+
25
+ # DB์—์„œ ๊ฐ€์ ธ์˜ค๊ธฐ (์ˆœํ™˜ ์ฐธ์กฐ ๋ฐฉ์ง€๋ฅผ ์œ„ํ•ด ์—ฌ๊ธฐ์„œ ์ž„ํฌํŠธ)
26
+ try:
27
+ from app.database import SystemConfig
28
+ token = SystemConfig.get_config('huggingface_token', '').strip()
29
+ if token:
30
+ print(f"[Hugging Face] DB์—์„œ ํ† ํฐ ๊ฐ€์ ธ์˜ด (๊ธธ์ด: {len(token)}์ž)")
31
+ else:
32
+ print(f"[Hugging Face] DB์— ํ† ํฐ์ด ์—†๊ฑฐ๋‚˜ ๋น„์–ด์žˆ์Œ")
33
+ return token
34
+ except Exception as e:
35
+ print(f"[Hugging Face] DB์—์„œ ํ† ํฐ ์กฐํšŒ ์‹คํŒจ: {e}")
36
+ return ''
37
+
38
+ def reset_huggingface_token():
39
+ """
40
+ Hugging Face ํ† ํฐ ์บ์‹œ ์ดˆ๊ธฐํ™” (ํ† ํฐ์ด ์—…๋ฐ์ดํŠธ๋œ ํ›„ ํ˜ธ์ถœ)
41
+ """
42
+ # ํ˜„์žฌ๋Š” ๋‹จ์ˆœํžˆ ํ•จ์ˆ˜๋ฅผ ๋‹ค์‹œ ํ˜ธ์ถœํ•˜๋ฉด ๋˜์ง€๋งŒ,
43
+ # ํ–ฅํ›„ ์บ์‹ฑ์ด ์ถ”๊ฐ€๋  ๊ฒฝ์šฐ๋ฅผ ๋Œ€๋น„ํ•œ ํ•จ์ˆ˜
44
+ pass
45
+
46
+
47
+
48
+
49
+
50
+
51
+
52
+