tianlong12 commited on
Commit
54ca631
·
verified ·
1 Parent(s): e9f3137

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -1
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=cookies_str)
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}")