Junhui Ji commited on
Commit
4f78b1f
·
1 Parent(s): e8a59d4

update logging, update key popup position

Browse files
Files changed (2) hide show
  1. main.py +4 -1
  2. static/styles.css +2 -2
main.py CHANGED
@@ -32,6 +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
 
36
  class ConnectionLimitMiddleware(BaseHTTPMiddleware):
37
  async def dispatch(self, request: Request, call_next):
@@ -209,9 +210,10 @@ async def optimize_design(request: OptimizationRequest, client_ip: str = None):
209
  try:
210
  # 获取客户端IP(如果未提供,使用默认值)
211
  if client_ip is None or client_ip not in white_list:
 
212
  raise HTTPException(
213
  status_code=503,
214
- detail="当前用户无生图权限,请点击'想使用自己的OpenAI API Key?'输入您的OpenAI API Key或联系@王月(Phoebe)添加白名单后重试。"
215
  )
216
 
217
  user_rate_limit = white_list[client_ip]
@@ -228,6 +230,7 @@ async def optimize_design(request: OptimizationRequest, client_ip: str = None):
228
  optimize_design_timestamps[client_ip] = current_time
229
 
230
  # 检查是否超过限制
 
231
  if optimize_design_requests[client_ip] >= user_rate_limit:
232
  raise HTTPException(
233
  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
+ logging.log(logging.INFO, white_list)
36
 
37
  class ConnectionLimitMiddleware(BaseHTTPMiddleware):
38
  async def dispatch(self, request: Request, call_next):
 
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添加白名单后重试。"
217
  )
218
 
219
  user_rate_limit = white_list[client_ip]
 
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,
static/styles.css CHANGED
@@ -947,8 +947,8 @@ footer {
947
  .api-key-dialog {
948
  display: none;
949
  position: fixed;
950
- top: 0;
951
- left: 0;
952
  width: 100%;
953
  height: 100%;
954
  background: rgba(0, 0, 0, 0.7);
 
947
  .api-key-dialog {
948
  display: none;
949
  position: fixed;
950
+ top: 50%;
951
+ left: 50%;
952
  width: 100%;
953
  height: 100%;
954
  background: rgba(0, 0, 0, 0.7);