Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -3,10 +3,22 @@ from fastapi import FastAPI, HTTPException
|
|
3 |
from pydantic import BaseModel
|
4 |
import uvicorn
|
5 |
import os
|
6 |
-
import json
|
7 |
|
8 |
-
|
9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
|
11 |
class RequestBody(BaseModel):
|
12 |
prompt: str
|
|
|
3 |
from pydantic import BaseModel
|
4 |
import uvicorn
|
5 |
import os
|
|
|
6 |
|
7 |
+
# 从环境变量中读取 cookies
|
8 |
+
cookies_str = os.getenv("COOKIES_STR")
|
9 |
+
if not cookies_str:
|
10 |
+
raise Exception("COOKIES_STR environment variable is not set")
|
11 |
+
|
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=cookies_str)
|
18 |
+
print("ChatBot initialized successfully!")
|
19 |
+
except Exception as e:
|
20 |
+
print(f"Failed to initialize ChatBot: {e}")
|
21 |
+
raise
|
22 |
|
23 |
class RequestBody(BaseModel):
|
24 |
prompt: str
|