Spaces:
Sleeping
Sleeping
Junhui Ji
commited on
Commit
·
4f78b1f
1
Parent(s):
e8a59d4
update logging, update key popup position
Browse files- main.py +4 -1
- 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
|
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:
|
951 |
-
left:
|
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);
|