ssboost commited on
Commit
d788c0e
ยท
verified ยท
1 Parent(s): b9d6115

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -12,6 +12,13 @@ import glob
12
  from datetime import datetime
13
  from gradio_client import Client
14
 
 
 
 
 
 
 
 
15
  # ๋กœ๊น… ์„ค์ •
16
  logging.basicConfig(
17
  level=logging.INFO,
@@ -27,9 +34,11 @@ logger = logging.getLogger(__name__)
27
  # API ํด๋ผ์ด์–ธํŠธ ์„ค์ •
28
  def get_api_client():
29
  """ํ™˜๊ฒฝ๋ณ€์ˆ˜์—์„œ API ์—”๋“œํฌ์ธํŠธ๋ฅผ ๊ฐ€์ ธ์™€ ํด๋ผ์ด์–ธํŠธ ์ƒ์„ฑ"""
30
- endpoint = os.getenv('API_ENDPOINT')
31
  if not endpoint:
32
  raise ValueError("API_ENDPOINT ํ™˜๊ฒฝ๋ณ€์ˆ˜๊ฐ€ ์„ค์ •๋˜์ง€ ์•Š์•˜์Šต๋‹ˆ๋‹ค.")
 
 
33
  return Client(endpoint)
34
 
35
  # ์„ธ์…˜๋ณ„ ์ž„์‹œ ํŒŒ์ผ ๊ด€๋ฆฌ๋ฅผ ์œ„ํ•œ ๋”•์…”๋„ˆ๋ฆฌ
 
12
  from datetime import datetime
13
  from gradio_client import Client
14
 
15
+ # ํ™˜๊ฒฝ๋ณ€์ˆ˜ ๋กœ๋“œ๋ฅผ ์œ„ํ•œ dotenv ์‚ฌ์šฉ
16
+ try:
17
+ from dotenv import load_dotenv
18
+ load_dotenv()
19
+ except ImportError:
20
+ pass
21
+
22
  # ๋กœ๊น… ์„ค์ •
23
  logging.basicConfig(
24
  level=logging.INFO,
 
34
  # API ํด๋ผ์ด์–ธํŠธ ์„ค์ •
35
  def get_api_client():
36
  """ํ™˜๊ฒฝ๋ณ€์ˆ˜์—์„œ API ์—”๋“œํฌ์ธํŠธ๋ฅผ ๊ฐ€์ ธ์™€ ํด๋ผ์ด์–ธํŠธ ์ƒ์„ฑ"""
37
+ endpoint = os.getenv('API_ENDPOINT', '').strip()
38
  if not endpoint:
39
  raise ValueError("API_ENDPOINT ํ™˜๊ฒฝ๋ณ€์ˆ˜๊ฐ€ ์„ค์ •๋˜์ง€ ์•Š์•˜์Šต๋‹ˆ๋‹ค.")
40
+ if endpoint.startswith('#') or len(endpoint) == 0:
41
+ raise ValueError("API_ENDPOINT ํ™˜๊ฒฝ๋ณ€์ˆ˜๊ฐ€ ์˜ฌ๋ฐ”๋ฅด๊ฒŒ ์„ค์ •๋˜์ง€ ์•Š์•˜์Šต๋‹ˆ๋‹ค.")
42
  return Client(endpoint)
43
 
44
  # ์„ธ์…˜๋ณ„ ์ž„์‹œ ํŒŒ์ผ ๊ด€๋ฆฌ๋ฅผ ์œ„ํ•œ ๋”•์…”๋„ˆ๋ฆฌ