delete client ip

main.py

@@ -205,37 +205,37 @@ async def analyze_feedback(request: AnalysisRequest):
 @app.post("/api/optimize-design")
 async def optimize_design(request: OptimizationRequest, client_ip: str = None):
     try:
-        # 获取客户端IP（如果未提供，使用默认值）
-        if client_ip is None or client_ip not in white_list:
+        # 查看客户是否提供了openai_key，获取客户端IP（如果未提供，使用默认值）
+        if not request.openai_key or client_ip is None or client_ip not in white_list:
             print(f'user_ip {client_ip} not in whitelist')
             raise HTTPException(
                 status_code=503,
                 detail="当前用户无生图权限，请点击'想使用自己的OpenAI API Key？'输入您的OpenAI API Key或联系@wangyue161添加白名单后重试。"
             )
 
-        user_rate_limit = white_list[client_ip]
-        
-        current_time = time.time()
-        
-        # 检查是否需要重置计数器（超过24小时）
-        if current_time - optimize_design_timestamps[client_ip] > 3600*24:
-            optimize_design_requests[client_ip] = 0
-            optimize_design_timestamps[client_ip] = current_time
-            
-        # 如果是首次请求，记录时间戳
-        if optimize_design_requests[client_ip] == 0:
-            optimize_design_timestamps[client_ip] = current_time
-            
-        # 检查是否超过限制
-        print(f'user_ip {client_ip}, total requests in routine: {optimize_design_requests[client_ip]}, rate limit: {user_rate_limit}')
-        if optimize_design_requests[client_ip] >= user_rate_limit:
-            raise HTTPException(
-                status_code=503,
-                detail="用户当日改图接口访问已达上限，请24小时后重试"
-            )
-            
-        # 增加请求计数
-        optimize_design_requests[client_ip] += 1
+        # user_rate_limit = white_list[client_ip]
+        #
+        # current_time = time.time()
+        #
+        # # 检查是否需要重置计数器（超过24小时）
+        # if current_time - optimize_design_timestamps[client_ip] > 3600*24:
+        #     optimize_design_requests[client_ip] = 0
+        #     optimize_design_timestamps[client_ip] = current_time
+        #
+        # # 如果是首次请求，记录时间戳
+        # if optimize_design_requests[client_ip] == 0:
+        #     optimize_design_timestamps[client_ip] = current_time
+        #
+        # # 检查是否超过限制
+        # print(f'user_ip {client_ip}, total requests in routine: {optimize_design_requests[client_ip]}, rate limit: {user_rate_limit}')
+        # if optimize_design_requests[client_ip] >= user_rate_limit:
+        #     raise HTTPException(
+        #         status_code=503,
+        #         detail="用户当日改图接口访问已达上限，请24小时后重试"
+        #     )
+        #
+        # # 增加请求计数
+        # optimize_design_requests[client_ip] += 1
 
         # 提取设计类型
         design_type = f"设计类型：{request.text.split()[0]}\n" if len(request.text.split()) > 1 else ""

static/script.js

@@ -301,7 +301,6 @@ document.addEventListener('DOMContentLoaded', function() {
         });
     }
 
-
     // 分析用户输入和图片的函数
     async function analyzeUserInput(text, imageData) {
         try {
@@ -707,8 +706,7 @@ document.addEventListener('DOMContentLoaded', function() {
             const response = await fetch(`${BASE_URL}api/optimize-design`, {
                 method: 'POST',
                 headers: {
-                    'Content-Type': 'application/json',
-                    'X-Forwarded-For': window.location.hostname
+                    'Content-Type': 'application/json'
                 },
                 body: JSON.stringify({
                     text: uploadedText,
@@ -716,7 +714,6 @@ document.addEventListener('DOMContentLoaded', function() {
                     suggestions: suggestions,
                     request_model_id: 'gpt-image-1',
                     openai_key: sessionStorage.getItem('userOpenaiKey') || '',
-                    client_ip: window.location.hostname
                 })
             });
 
@@ -781,16 +778,14 @@ document.addEventListener('DOMContentLoaded', function() {
             const response = await fetch(`${BASE_URL}api/optimize-design`, {
                 method: 'POST',
                 headers: {
-                    'Content-Type': 'application/json',
-                    'X-Forwarded-For': window.location.hostname
+                    'Content-Type': 'application/json'
                 },
                 body: JSON.stringify({
                     text: uploadedText,
                     image_data: uploadedImage,
                     suggestions: suggestions,
                     request_model_id: 'dall-e-2',
-                    openai_key: sessionStorage.getItem('userOpenaiKey') || '',
-                    client_ip: window.location.hostname
+                    openai_key: sessionStorage.getItem('userOpenaiKey') || ''
                 })
             });
             if (response.status === 503) {