tianlong12 commited on
Commit
c8bf377
·
verified ·
1 Parent(s): 3c30c75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -8,9 +8,19 @@ import os
8
  # 从环境变量中读取登录信息
9
  EMAIL = os.getenv("EMAIL")
10
  PASSWD = os.getenv("PASSWD")
 
 
 
 
 
11
  cookie_path_dir = "./cookies/"
12
  sign = Login(EMAIL, PASSWD)
13
- cookies = sign.login(cookie_dir_path=cookie_path_dir, save_cookies=True)
 
 
 
 
 
14
 
15
  # 创建 ChatBot
16
  chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
 
8
  # 从环境变量中读取登录信息
9
  EMAIL = os.getenv("EMAIL")
10
  PASSWD = os.getenv("PASSWD")
11
+
12
+ # 检查环境变量是否正确读取
13
+ if not EMAIL or not PASSWD:
14
+ raise Exception("EMAIL or PASSWD environment variables are not set")
15
+
16
  cookie_path_dir = "./cookies/"
17
  sign = Login(EMAIL, PASSWD)
18
+
19
+ try:
20
+ cookies = sign.login(cookie_dir_path=cookie_path_dir, save_cookies=True)
21
+ except Exception as e:
22
+ print(f"Login failed: {e}")
23
+ raise
24
 
25
  # 创建 ChatBot
26
  chatbot = hugchat.ChatBot(cookies=cookies.get_dict())