Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ from fastapi import FastAPI, HTTPException
|
|
3 |
from pydantic import BaseModel
|
4 |
import uvicorn
|
5 |
import os
|
|
|
6 |
|
7 |
# 从环境变量中读取 cookies
|
8 |
cookies_str = os.getenv("COOKIES_STR")
|
@@ -12,9 +13,22 @@ if not cookies_str:
|
|
12 |
# 打印调试信息
|
13 |
print(f"COOKIES_STR: {cookies_str}")
|
14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
try:
|
16 |
print("Attempting to initialize ChatBot with provided cookies...")
|
17 |
-
chatbot = hugchat.ChatBot(cookies=
|
18 |
print("ChatBot initialized successfully!")
|
19 |
except Exception as e:
|
20 |
print(f"Failed to initialize ChatBot: {e}")
|
|
|
3 |
from pydantic import BaseModel
|
4 |
import uvicorn
|
5 |
import os
|
6 |
+
import requests
|
7 |
|
8 |
# 从环境变量中读取 cookies
|
9 |
cookies_str = os.getenv("COOKIES_STR")
|
|
|
13 |
# 打印调试信息
|
14 |
print(f"COOKIES_STR: {cookies_str}")
|
15 |
|
16 |
+
def parse_cookies(cookies_str):
|
17 |
+
cookies = {}
|
18 |
+
for cookie in cookies_str.split('; '):
|
19 |
+
name, value = cookie.split('=', 1)
|
20 |
+
cookies[name] = value
|
21 |
+
return cookies
|
22 |
+
|
23 |
+
try:
|
24 |
+
cookies = parse_cookies(cookies_str)
|
25 |
+
print("Successfully parsed COOKIES_STR")
|
26 |
+
except Exception as e:
|
27 |
+
raise Exception(f"Failed to parse COOKIES_STR: {e}")
|
28 |
+
|
29 |
try:
|
30 |
print("Attempting to initialize ChatBot with provided cookies...")
|
31 |
+
chatbot = hugchat.ChatBot(cookies=cookies)
|
32 |
print("ChatBot initialized successfully!")
|
33 |
except Exception as e:
|
34 |
print(f"Failed to initialize ChatBot: {e}")
|