Junhui Ji commited on
Commit
df5d8bf
·
1 Parent(s): d22b2fa

update logging, update key popup position

Browse files
Files changed (1) hide show
  1. main.py +5 -8
main.py CHANGED
@@ -32,7 +32,7 @@ current_connections = 0
32
  optimize_design_requests = defaultdict(int) # 记录每个IP的请求次数
33
  optimize_design_timestamps = defaultdict(float) # 记录每个IP的首次请求时间
34
  white_list = eval(os.getenv("WHITELIST"))
35
- logging.log(logging.INFO, white_list)
36
 
37
  class ConnectionLimitMiddleware(BaseHTTPMiddleware):
38
  async def dispatch(self, request: Request, call_next):
@@ -65,9 +65,6 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
65
  OPENAI_API_IMAGE_EDIT_KEY = os.getenv("OPENAI_API_IMAGE_EDIT_KEY")
66
  GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
67
  SEARCH_ENGINE_ID = os.getenv("SEARCH_ENGINE_ID", "27acb0d55ad504716")
68
- print(OPENAI_API_KEY)
69
- print(OPENAI_API_IMAGE_EDIT_KEY)
70
- print(GOOGLE_API_KEY)
71
 
72
  class ScreenshotRequest(BaseModel):
73
  url: str
@@ -113,7 +110,7 @@ async def capture_screenshot(request: ScreenshotRequest):
113
  filename = f"{domain}_{timestamp}.{request.format}"
114
  filepath = os.path.join(CACHE_DIR, filename)
115
 
116
- logging.log(logging.INFO, f"开始为 {request.url} 生成截图...")
117
 
118
  # 默认请求头
119
  default_headers = {
@@ -153,7 +150,7 @@ async def capture_screenshot(request: ScreenshotRequest):
153
 
154
  await browser.close()
155
 
156
- logging.log(logging.DEBUG, f"截图完成: {filepath}")
157
 
158
  # 返回截图URL
159
  screenshot_url = f"/screenshots/{filename}"
@@ -210,7 +207,7 @@ async def optimize_design(request: OptimizationRequest, client_ip: str = None):
210
  try:
211
  # 获取客户端IP(如果未提供,使用默认值)
212
  if client_ip is None or client_ip not in white_list:
213
- logging.log(logging.INFO, f'user_ip {client_ip} not in whitelist')
214
  raise HTTPException(
215
  status_code=503,
216
  detail="当前用户无生图权限,请点击'想使用自己的OpenAI API Key?'输入您的OpenAI API Key或联系@wangyue161添加白名单后重试。"
@@ -230,7 +227,7 @@ async def optimize_design(request: OptimizationRequest, client_ip: str = None):
230
  optimize_design_timestamps[client_ip] = current_time
231
 
232
  # 检查是否超过限制
233
- logging.log(logging.INFO, f'user_ip {client_ip}, total requests in routine: {optimize_design_requests[client_ip]}, rate limit: {user_rate_limit}')
234
  if optimize_design_requests[client_ip] >= user_rate_limit:
235
  raise HTTPException(
236
  status_code=503,
 
32
  optimize_design_requests = defaultdict(int) # 记录每个IP的请求次数
33
  optimize_design_timestamps = defaultdict(float) # 记录每个IP的首次请求时间
34
  white_list = eval(os.getenv("WHITELIST"))
35
+ print(logging.INFO, white_list)
36
 
37
  class ConnectionLimitMiddleware(BaseHTTPMiddleware):
38
  async def dispatch(self, request: Request, call_next):
 
65
  OPENAI_API_IMAGE_EDIT_KEY = os.getenv("OPENAI_API_IMAGE_EDIT_KEY")
66
  GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
67
  SEARCH_ENGINE_ID = os.getenv("SEARCH_ENGINE_ID", "27acb0d55ad504716")
 
 
 
68
 
69
  class ScreenshotRequest(BaseModel):
70
  url: str
 
110
  filename = f"{domain}_{timestamp}.{request.format}"
111
  filepath = os.path.join(CACHE_DIR, filename)
112
 
113
+ print(f"开始为 {request.url} 生成截图...")
114
 
115
  # 默认请求头
116
  default_headers = {
 
150
 
151
  await browser.close()
152
 
153
+ print(f"截图完成: {filepath}")
154
 
155
  # 返回截图URL
156
  screenshot_url = f"/screenshots/{filename}"
 
207
  try:
208
  # 获取客户端IP(如果未提供,使用默认值)
209
  if client_ip is None or client_ip not in white_list:
210
+ print(f'user_ip {client_ip} not in whitelist')
211
  raise HTTPException(
212
  status_code=503,
213
  detail="当前用户无生图权限,请点击'想使用自己的OpenAI API Key?'输入您的OpenAI API Key或联系@wangyue161添加白名单后重试。"
 
227
  optimize_design_timestamps[client_ip] = current_time
228
 
229
  # 检查是否超过限制
230
+ print(f'user_ip {client_ip}, total requests in routine: {optimize_design_requests[client_ip]}, rate limit: {user_rate_limit}')
231
  if optimize_design_requests[client_ip] >= user_rate_limit:
232
  raise HTTPException(
233
  status_code=503,