Update app.py

app.py

@@ -1,29 +1,19 @@
 from hugchat import hugchat
-from hugchat.login import Login
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import uvicorn
 import os
+import json
 
-# 从环境变量中读取登录信息
-EMAIL = os.getenv("EMAIL")
-PASSWD = os.getenv("PASSWD")
+# 从环境变量中读取 cookies
+cookies_json = os.getenv("COOKIES_JSON")
+if not cookies_json:
+    raise Exception("COOKIES_JSON environment variable is not set")
 
-if not EMAIL or not PASSWD:
-    raise Exception("EMAIL or PASSWD environment variables are not set")
-
-sign = Login(EMAIL, PASSWD)
-
-try:
-    print("Attempting to login...")
-    cookies = sign.login(save_cookies=False)  # 不保存到文件中，直接获取cookie字典
-    print("Login successful!")
-except Exception as e:
-    print(f"Login failed: {e}")
-    raise
+cookies = json.loads(cookies_json)
 
 # 创建 ChatBot
-chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
+chatbot = hugchat.ChatBot(cookies=cookies)
 
 class RequestBody(BaseModel):
     prompt: str