init

app.py

@@ -4,144 +4,185 @@ import os
 import shutil
 import tempfile
 import time
-from util import process_image_edit, process_local_image_edit
+from util import process_image_edit, process_local_image_edit, download_and_check_result_nsfw
 from nfsw import NSFWDetector
 
-# 配置参数
-NSFW_TIME_WINDOW = 5  # 时间窗口：5分钟
-NSFW_LIMIT = 10        # 限制次数：6次
+# Configuration parameters
+FREE_TRY_N = 20        # Free phase: first 20 tries without restrictions
+SLOW_TRY_N = 30        # Slow phase start: 30 tries
+SLOW2_TRY_N = 40        # Slow phase start: 30 tries
+RATE_LIMIT_60 = 75     # Full restriction: blocked after 60 tries
+
+# Time window configuration (minutes)
+PHASE_1_WINDOW = 5     # 20-30 tries: 3 minutes
+PHASE_2_WINDOW = 8     # 30-40 tries: 6 minutes
+PHASE_3_WINDOW = 15    # 40-60 tries: 10 minutes
+MAX_IMAGES_PER_WINDOW = 2  # Max images per time window
 
 IP_Dict = {}
-NSFW_Dict = {}  # 记录每个IP的NSFW违规次数
-NSFW_Time_Dict = {}  # 记录每个IP在特定时间窗口的NSFW检测次数，键格式: "ip_timestamp"
+# IP generation statistics and time window tracking
+IP_Generation_Count = {}  # Record total generation count for each IP
+IP_Rate_Limit_Track = {}  # Record generation count and timestamp in current time window for each IP
+
+def get_ip_generation_count(client_ip):
+    """
+    Get IP generation count
+    """
+    if client_ip not in IP_Generation_Count:
+        IP_Generation_Count[client_ip] = 0
+    return IP_Generation_Count[client_ip]
 
-def get_current_time_window():
+def increment_ip_generation_count(client_ip):
     """
-    获取当前的整点时间窗口
+    Increment IP generation count
+    """
+    if client_ip not in IP_Generation_Count:
+        IP_Generation_Count[client_ip] = 0
+    IP_Generation_Count[client_ip] += 1
+    return IP_Generation_Count[client_ip]
+
+def get_ip_phase(client_ip):
+    """
+    Get current phase for IP
     
     Returns:
-        tuple: (窗口开始时间戳, 窗口结束时间戳)
+        str: 'free', 'rate_limit_1', 'rate_limit_2', 'rate_limit_3', 'blocked'
     """
-    current_time = time.time()
-    # 获取当前时间的分钟数
-    current_struct = time.localtime(current_time)
-    current_minute = current_struct.tm_min
+    count = get_ip_generation_count(client_ip)
     
-    # 计算当前5分钟时间窗口的开始分钟
-    window_start_minute = (current_minute // NSFW_TIME_WINDOW) * NSFW_TIME_WINDOW
-    
-    # 构建窗口开始时间
-    window_start_struct = time.struct_time((
-        current_struct.tm_year, current_struct.tm_mon, current_struct.tm_mday,
-        current_struct.tm_hour, window_start_minute, 0,
-        current_struct.tm_wday, current_struct.tm_yday, current_struct.tm_isdst
-    ))
-    
-    window_start_time = time.mktime(window_start_struct)
-    window_end_time = window_start_time + (NSFW_TIME_WINDOW * 60)
-    
-    return window_start_time, window_end_time
+    if count < FREE_TRY_N:           # 0-19 tries
+        return 'free'
+    elif count < SLOW_TRY_N:         # 20-29 tries
+        return 'rate_limit_1'        # NSFW blur + 3 minutes 2 images
+    elif count < SLOW2_TRY_N:                 # 30-39 tries
+        return 'rate_limit_2'        # NSFW blur + 6 minutes 2 images
+    elif count < RATE_LIMIT_60:      # 40-59 tries
+        return 'rate_limit_3'        # NSFW blur + 10 minutes 2 images
+    else:                            # 60+ tries
+        return 'blocked'             # Generation blocked
 
-def check_nsfw_rate_limit(client_ip):
+def check_rate_limit_for_phase(client_ip, phase):
     """
-    检查IP的NSFW检测频率限制（基于整点时间窗口）
-    
-    Args:
-        client_ip (str): 客户端IP地址
+    Check rate limit for specific phase
     
     Returns:
-        tuple: (是否超过限制, 剩余等待时间)
+        tuple: (is_limited, wait_time_minutes, current_count)
     """
-    current_time = time.time()
-    window_start_time, window_end_time = get_current_time_window()
+    if phase not in ['rate_limit_1', 'rate_limit_2', 'rate_limit_3']:
+        return False, 0, 0
     
-    # 清理不在当前时间窗口的记录
-    current_window_key = f"{client_ip}_{int(window_start_time)}"
+    # Determine time window
+    if phase == 'rate_limit_1':
+        window_minutes = PHASE_1_WINDOW    # 3 minutes
+    elif phase == 'rate_limit_2':
+        window_minutes = PHASE_2_WINDOW    # 6 minutes
+    else:  # rate_limit_3
+        window_minutes = PHASE_3_WINDOW    # 10 minutes
     
-    # 如果没有当前窗口的记录，创建新的
-    if current_window_key not in NSFW_Time_Dict:
-        NSFW_Time_Dict[current_window_key] = 0
+    current_time = time.time()
+    window_key = f"{client_ip}_{phase}"
     
-    # 清理旧的窗口记录（保持内存清洁）
-    keys_to_remove = []
-    for key in NSFW_Time_Dict:
-        if key.startswith(client_ip + "_"):
-            window_time = int(key.split("_")[1])
-            if window_time < window_start_time:
-                keys_to_remove.append(key)
+    # Clean expired records
+    if window_key in IP_Rate_Limit_Track:
+        track_data = IP_Rate_Limit_Track[window_key]
+        # Check if within current time window
+        if current_time - track_data['start_time'] > window_minutes * 60:
+            # Time window expired, reset
+            IP_Rate_Limit_Track[window_key] = {
+                'count': 0,
+                'start_time': current_time,
+                'last_generation': current_time
+            }
+    else:
+        # Initialize
+        IP_Rate_Limit_Track[window_key] = {
+            'count': 0,
+            'start_time': current_time,
+            'last_generation': current_time
+        }
     
-    for key in keys_to_remove:
-        del NSFW_Time_Dict[key]
+    track_data = IP_Rate_Limit_Track[window_key]
     
-    # 检查当前窗口是否超过限制
-    if NSFW_Time_Dict[current_window_key] >= NSFW_LIMIT:
-        # 计算到下一个时间窗口的等待时间
-        wait_time = window_end_time - current_time
-        return True, max(0, wait_time)
+    # Check if exceeded limit
+    if track_data['count'] >= MAX_IMAGES_PER_WINDOW:
+        # Calculate remaining wait time
+        elapsed = current_time - track_data['start_time']
+        wait_time = (window_minutes * 60) - elapsed
+        wait_minutes = max(0, wait_time / 60)
+        return True, wait_minutes, track_data['count']
     
-    return False, 0
+    return False, 0, track_data['count']
 
-def record_nsfw_detection(client_ip):
+def record_generation_attempt(client_ip, phase):
     """
-    记录IP的NSFW检测时间（基于整点时间窗口）
-    
-    Args:
-        client_ip (str): 客户端IP地址
+    Record generation attempt
     """
-    window_start_time, _ = get_current_time_window()
-    current_window_key = f"{client_ip}_{int(window_start_time)}"
-    
-    # 增加当前窗口的计数
-    if current_window_key not in NSFW_Time_Dict:
-        NSFW_Time_Dict[current_window_key] = 0
-    NSFW_Time_Dict[current_window_key] += 1
+    # Increment total count
+    increment_ip_generation_count(client_ip)
     
-    # 记录到NSFW_Dict中（兼容现有逻辑）
-    if client_ip not in NSFW_Dict:
-        NSFW_Dict[client_ip] = 0
-    NSFW_Dict[client_ip] += 1
+    # Record time window count
+    if phase in ['rate_limit_1', 'rate_limit_2', 'rate_limit_3']:
+        window_key = f"{client_ip}_{phase}"
+        current_time = time.time()
+        
+        if window_key in IP_Rate_Limit_Track:
+            IP_Rate_Limit_Track[window_key]['count'] += 1
+            IP_Rate_Limit_Track[window_key]['last_generation'] = current_time
+        else:
+            IP_Rate_Limit_Track[window_key] = {
+                'count': 1,
+                'start_time': current_time,
+                'last_generation': current_time
+            }
 
-def get_current_window_info(client_ip):
+def apply_gaussian_blur_to_image_url(image_url, blur_strength=50):
     """
-    获取当前窗口的统计信息（用于调试）
+    Apply Gaussian blur to image URL
     
     Args:
-        client_ip (str): 客户端IP地址
+        image_url (str): Original image URL
+        blur_strength (int): Blur strength, default 50 (heavy blur)
         
     Returns:
-        dict: 当前窗口的统计信息
+        PIL.Image: Blurred PIL Image object
     """
-    window_start_time, window_end_time = get_current_time_window()
-    current_window_key = f"{client_ip}_{int(window_start_time)}"
-    
-    current_count = NSFW_Time_Dict.get(current_window_key, 0)
-    
-    # 格式化时间显示
-    start_time_str = time.strftime("%H:%M:%S", time.localtime(window_start_time))
-    end_time_str = time.strftime("%H:%M:%S", time.localtime(window_end_time))
-    
-    return {
-        "window_start": start_time_str,
-        "window_end": end_time_str,
-        "current_count": current_count,
-        "limit": NSFW_LIMIT,
-        "window_key": current_window_key
-    }
-
-# 初始化NSFW检测器（从Hugging Face下载）
+    try:
+        import requests
+        from PIL import Image, ImageFilter
+        import io
+        
+        # Download image
+        response = requests.get(image_url, timeout=30)
+        if response.status_code != 200:
+            return None
+        
+        # Convert to PIL Image
+        image_data = io.BytesIO(response.content)
+        image = Image.open(image_data)
+        
+        # Apply heavy Gaussian blur
+        blurred_image = image.filter(ImageFilter.GaussianBlur(radius=blur_strength))
+        
+        return blurred_image
+        
+    except Exception as e:
+        print(f"⚠️ Failed to apply Gaussian blur: {e}")
+        return None
+
+# Initialize NSFW detector (download from Hugging Face)
 try:
-    nsfw_detector = NSFWDetector()  # 自动从Hugging Face下载falconsai_yolov9_nsfw_model_quantized.pt
-    print("✅ NSFW检测器初始化成功")
+    nsfw_detector = NSFWDetector()  # Auto download falconsai_yolov9_nsfw_model_quantized.pt from Hugging Face
+    print("✅ NSFW detector initialized successfully")
 except Exception as e:
-    print(f"❌ NSFW检测器初始化失败: {e}")
+    print(f"❌ NSFW detector initialization failed: {e}")
     nsfw_detector = None
 
 def edit_image_interface(input_image, prompt, request: gr.Request, progress=gr.Progress()):
     """
-    Interface function for processing image editing
+    Interface function for processing image editing with phase-based limitations
     """
     try:
-        # 提取用户IP
+        # Extract user IP
         client_ip = request.client.host
         x_forwarded_for = dict(request.headers).get('x-forwarded-for')
         if x_forwarded_for:
@@ -150,52 +191,54 @@ def edit_image_interface(input_image, prompt, request: gr.Request, progress=gr.P
             IP_Dict[client_ip] = 0
         IP_Dict[client_ip] += 1
 
-        
         if input_image is None:
             return None, "Please upload an image first"
         
         if not prompt or prompt.strip() == "":
             return None, "Please enter editing prompt"
         
-        # 检查prompt长度是否大于3个字符
+        # Check if prompt length is greater than 3 characters
         if len(prompt.strip()) <= 3:
             return None, "❌ Editing prompt must be more than 3 characters"
     except Exception as e:
         print(f"⚠️ Request preprocessing error: {e}")
         return None, "❌ Request processing error"
     
-    # 检查图片是否包含NSFW内容
-    nsfw_result = None
-    if nsfw_detector is not None and input_image is not None:
+    # Get user current phase
+    current_phase = get_ip_phase(client_ip)
+    current_count = get_ip_generation_count(client_ip)
+    
+    print(f"📊 User phase info - IP: {client_ip}, current phase: {current_phase}, generation count: {current_count}")
+    
+    # Check if completely blocked
+    if current_phase == 'blocked':
+        return None, f"❌ You have reached Hugging Face's free generation limit. Please visit https://omnicreator.net/#generator for unlimited generation"
+    
+    # Check rate limit (applies to rate_limit phases)
+    if current_phase in ['rate_limit_1', 'rate_limit_2', 'rate_limit_3']:
+        is_limited, wait_minutes, window_count = check_rate_limit_for_phase(client_ip, current_phase)
+        if is_limited:
+            wait_minutes_int = int(wait_minutes) + 1
+            return None, f"❌ You have reached Hugging Face's free generation limit. Please visit https://omnicreator.net/#generator for unlimited generation, or wait {wait_minutes_int} minutes before generating again"
+    
+    # Handle NSFW detection based on phase
+    is_nsfw_task = False  # Track if this task involves NSFW content
+    
+    # Skip NSFW detection in free phase
+    if current_phase != 'free' and nsfw_detector is not None and input_image is not None:
         try:
-            # 直接使用PIL Image对象进行检测，避免文件路径问题
             nsfw_result = nsfw_detector.predict_pil_label_only(input_image)
             
             if nsfw_result.lower() == "nsfw":
-                print(f"🔍 NSFW检测结果: ❌❌❌ {nsfw_result} - IP: {client_ip}")
-                # 检查NSFW频率限制
-                is_rate_limited, wait_time = check_nsfw_rate_limit(client_ip)
-                
-                if is_rate_limited:
-                    # 超过频率限制，显示等待提示并阻止继续
-                    wait_minutes = int(wait_time / 60) + 1  # 向上取整到分钟
-                    window_info = get_current_window_info(client_ip)
-                    print(f"⚠️ NSFW频率限制 - IP: {client_ip}")
-                    print(f"   时间窗口: {window_info['window_start']} - {window_info['window_end']}")
-                    print(f"   当前计数: {window_info['current_count']}/{NSFW_LIMIT}, 需要等待 {wait_minutes} 分钟")
-                    return None, f"❌ Please wait {wait_minutes} minutes before generating again"
-                else:
-                    # 未超过频率限制，记录此次检测但允许继续处理
-                    record_nsfw_detection(client_ip)
-                    window_info = get_current_window_info(client_ip)
+                is_nsfw_task = True
+                print(f"🔍 Input NSFW detected in {current_phase} phase: ❌❌❌ {nsfw_result} - IP: {client_ip} (will blur result)")
             else:
-                print(f"🔍 NSFW检测结果: ✅✅✅ {nsfw_result} - IP: {client_ip}")
+                print(f"🔍 Input NSFW check passed: ✅✅✅ {nsfw_result} - IP: {client_ip}")
                 
         except Exception as e:
-            print(f"⚠️ NSFW检测失败: {e}")
-            # 检测失败时允许继续处理
+            print(f"⚠️ Input NSFW detection failed: {e}")
+            # Allow continuation when detection fails
     
-
     result_url = None
     status_message = ""
     
@@ -203,27 +246,75 @@ def edit_image_interface(input_image, prompt, request: gr.Request, progress=gr.P
         try:
             nonlocal status_message
             status_message = message
-            # 增加错误处理，防止 progress 更新失败
+            # Add error handling to prevent progress update failure
             if progress is not None:
                 progress(0.5, desc=message)
         except Exception as e:
             print(f"⚠️ Progress update failed: {e}")
     
     try:
-        # 打印成功访问的信息
-        print(f"✅ Processing started - IP: {client_ip}, count: {IP_Dict[client_ip]}, prompt: {prompt.strip()}", flush=True)
+        # Record generation attempt (before actual generation to ensure correct count)
+        record_generation_attempt(client_ip, current_phase)
+        updated_count = get_ip_generation_count(client_ip)
+        
+        print(f"✅ Processing started - IP: {client_ip}, phase: {current_phase}, total count: {updated_count}, prompt: {prompt.strip()}", flush=True)
         
         # Call image editing processing function
         result_url, message = process_image_edit(input_image, prompt.strip(), progress_callback)
         
         if result_url:
             print(f"✅ Processing completed successfully - IP: {client_ip}, result_url: {result_url}", flush=True)
+            
+            # Detect result image NSFW content (only in rate limit phases)
+            if nsfw_detector is not None and current_phase != 'free':
+                try:
+                    if progress is not None:
+                        progress(0.9, desc="Checking result image...")
+                    
+                    is_nsfw, nsfw_error = download_and_check_result_nsfw(result_url, nsfw_detector)
+                    
+                    if nsfw_error:
+                        print(f"⚠️ Result image NSFW detection error - IP: {client_ip}, error: {nsfw_error}")
+                    elif is_nsfw:
+                        is_nsfw_task = True  # Mark task as NSFW
+                        print(f"🔍 Result image NSFW detected in {current_phase} phase: ❌❌❌ - IP: {client_ip} (will blur result)")
+                    else:
+                        print(f"🔍 Result image NSFW check passed: ✅✅✅ - IP: {client_ip}")
+                        
+                except Exception as e:
+                    print(f"⚠️ Result image NSFW detection exception - IP: {client_ip}, error: {str(e)}")
+            
+            # Apply blur if this is an NSFW task in rate limit phases
+            should_blur = False
+            
+            if current_phase in ['rate_limit_1', 'rate_limit_2', 'rate_limit_3'] and is_nsfw_task:
+                should_blur = True
+            
+            # Apply blur processing
+            if should_blur:
+                if progress is not None:
+                    progress(0.95, desc="Applying content filter...")
+                
+                blurred_image = apply_gaussian_blur_to_image_url(result_url)
+                if blurred_image is not None:
+                    final_result = blurred_image  # Return PIL Image object
+                    final_message = f"⚠️ NSFW content detected, content filter applied. NSFW content is prohibited by Hugging Face, but you can generate unlimited content at our official website https://omnicreator.net/#generator"
+                    print(f"🔒 Applied Gaussian blur for NSFW content - IP: {client_ip}")
+                else:
+                    # Blur failed, return original URL with warning
+                    final_result = result_url
+                    final_message = f"⚠️ NSFW content detected, but content filter failed. Please visit https://omnicreator.net/#generator for better experience"
+            else:
+                final_result = result_url
+                final_message = "✅ " + message
+            
             try:
                 if progress is not None:
                     progress(1.0, desc="Processing completed")
             except Exception as e:
                 print(f"⚠️ Final progress update failed: {e}")
-            return result_url, "✅ " + message
+            
+            return final_result, final_message
         else:
             print(f"❌ Processing failed - IP: {client_ip}, error: {message}", flush=True)
             return None, "❌ " + message
@@ -232,14 +323,12 @@ def edit_image_interface(input_image, prompt, request: gr.Request, progress=gr.P
         print(f"❌ Processing exception - IP: {client_ip}, error: {str(e)}")
         return None, f"❌ Error occurred during processing: {str(e)}"
 
-# 不再需要复杂的状态管理函数，已简化为内联函数
-
 def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Progress()):
     """
-    处理局部编辑请求
+    Handle local editing requests (with phase-based limitations)
     """
     try:
-        # 提取用户IP
+        # Extract user IP
         client_ip = request.client.host
         x_forwarded_for = dict(request.headers).get('x-forwarded-for')
         if x_forwarded_for:
@@ -248,7 +337,6 @@ def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Pr
             IP_Dict[client_ip] = 0
         IP_Dict[client_ip] += 1
 
-        
         if image_dict is None:
             return None, "Please upload an image and draw the area to edit"
         
@@ -272,34 +360,40 @@ def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Pr
         print(f"⚠️ Local edit request preprocessing error: {e}")
         return None, "❌ Request processing error"
     
-    # 检查图片是否包含NSFW内容
-    nsfw_result = None
-    if nsfw_detector is not None and base_image is not None:
+    # Get user current phase
+    current_phase = get_ip_phase(client_ip)
+    current_count = get_ip_generation_count(client_ip)
+    
+    print(f"📊 Local edit user phase info - IP: {client_ip}, current phase: {current_phase}, generation count: {current_count}")
+    
+    # Check if completely blocked
+    if current_phase == 'blocked':
+        return None, f"❌ You have reached Hugging Face's free generation limit. Please visit https://omnicreator.net/#generator for unlimited generation"
+    
+    # Check rate limit (applies to rate_limit phases)
+    if current_phase in ['rate_limit_1', 'rate_limit_2', 'rate_limit_3']:
+        is_limited, wait_minutes, window_count = check_rate_limit_for_phase(client_ip, current_phase)
+        if is_limited:
+            wait_minutes_int = int(wait_minutes) + 1
+            return None, f"❌ You have reached Hugging Face's free generation limit. Please visit https://omnicreator.net/#generator for unlimited generation, or wait {wait_minutes_int} minutes before generating again"
+    
+    # Handle NSFW detection based on phase
+    is_nsfw_task = False  # Track if this task involves NSFW content
+    
+    # Skip NSFW detection in free phase
+    if current_phase != 'free' and nsfw_detector is not None and base_image is not None:
         try:
             nsfw_result = nsfw_detector.predict_pil_label_only(base_image)
             
             if nsfw_result.lower() == "nsfw":
-                print(f"🔍 NSFW检测结果: ❌❌❌ {nsfw_result} - IP: {client_ip}")
-                # 检查NSFW频率限制
-                is_rate_limited, wait_time = check_nsfw_rate_limit(client_ip)
-                
-                if is_rate_limited:
-                    wait_minutes = int(wait_time / 60) + 1
-                    window_info = get_current_window_info(client_ip)
-                    print(f"⚠️ NSFW频率限制 - IP: {client_ip}")
-                    print(f"   时间窗口: {window_info['window_start']} - {window_info['window_end']}")
-                    print(f"   当前计数: {window_info['current_count']}/{NSFW_LIMIT}, 需要等待 {wait_minutes} 分钟")
-                    return None, f"❌ Please wait {wait_minutes} minutes before generating again"
-                else:
-                    record_nsfw_detection(client_ip)
-                    window_info = get_current_window_info(client_ip)
+                is_nsfw_task = True
+                print(f"🔍 Local edit input NSFW detected in {current_phase} phase: ❌❌❌ {nsfw_result} - IP: {client_ip} (will blur result)")
             else:
-                print(f"🔍 NSFW检测结果: ✅✅✅ {nsfw_result} - IP: {client_ip}")
+                print(f"🔍 Local edit input NSFW check passed: ✅✅✅ {nsfw_result} - IP: {client_ip}")
                 
         except Exception as e:
-            print(f"⚠️ NSFW检测失败: {e}")
-    
-    # IP访问限制检查
+            print(f"⚠️ Local edit input NSFW detection failed: {e}")
+            # Allow continuation when detection fails
 
     result_url = None
     status_message = ""
@@ -308,26 +402,75 @@ def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Pr
         try:
             nonlocal status_message
             status_message = message
-            # 增加错误处理，防止 progress 更新失败
+            # Add error handling to prevent progress update failure
             if progress is not None:
                 progress(0.5, desc=message)
         except Exception as e:
             print(f"⚠️ Local edit progress update failed: {e}")
     
     try:
-        print(f"✅ Local editing started - IP: {client_ip}, count: {IP_Dict[client_ip]}, prompt: {prompt.strip()}", flush=True)
+        # Record generation attempt (before actual generation to ensure correct count)
+        record_generation_attempt(client_ip, current_phase)
+        updated_count = get_ip_generation_count(client_ip)
+        
+        print(f"✅ Local editing started - IP: {client_ip}, phase: {current_phase}, total count: {updated_count}, prompt: {prompt.strip()}", flush=True)
         
-        # 调用局部图像编辑处理函数
+        # Call local image editing processing function
         result_url, message = process_local_image_edit(base_image, layers, prompt.strip(), progress_callback)
         
         if result_url:
             print(f"✅ Local editing completed successfully - IP: {client_ip}, result_url: {result_url}", flush=True)
+            
+            # Detect result image NSFW content (only in rate limit phases)
+            if nsfw_detector is not None and current_phase != 'free':
+                try:
+                    if progress is not None:
+                        progress(0.9, desc="Checking result image...")
+                    
+                    is_nsfw, nsfw_error = download_and_check_result_nsfw(result_url, nsfw_detector)
+                    
+                    if nsfw_error:
+                        print(f"⚠️ Local edit result image NSFW detection error - IP: {client_ip}, error: {nsfw_error}")
+                    elif is_nsfw:
+                        is_nsfw_task = True  # Mark task as NSFW
+                        print(f"🔍 Local edit result image NSFW detected in {current_phase} phase: ❌❌❌ - IP: {client_ip} (will blur result)")
+                    else:
+                        print(f"🔍 Local edit result image NSFW check passed: ✅✅✅ - IP: {client_ip}")
+                        
+                except Exception as e:
+                    print(f"⚠️ Local edit result image NSFW detection exception - IP: {client_ip}, error: {str(e)}")
+            
+            # Apply blur if this is an NSFW task in rate limit phases
+            should_blur = False
+            
+            if current_phase in ['rate_limit_1', 'rate_limit_2', 'rate_limit_3'] and is_nsfw_task:
+                should_blur = True
+            
+            # Apply blur processing
+            if should_blur:
+                if progress is not None:
+                    progress(0.95, desc="Applying content filter...")
+                
+                blurred_image = apply_gaussian_blur_to_image_url(result_url)
+                if blurred_image is not None:
+                    final_result = blurred_image  # Return PIL Image object
+                    final_message = f"⚠️ NSFW content detected, content filter applied. NSFW content is prohibited by Hugging Face, but you can generate unlimited content at our official website https://omnicreator.net/#generator"
+                    print(f"🔒 Local edit applied Gaussian blur for NSFW content - IP: {client_ip}")
+                else:
+                    # Blur failed, return original URL with warning
+                    final_result = result_url
+                    final_message = f"⚠️ NSFW content detected, but content filter failed. Please visit https://omnicreator.net/#generator for better experience"
+            else:
+                final_result = result_url
+                final_message = "✅ " + message
+            
             try:
                 if progress is not None:
                     progress(1.0, desc="Processing completed")
             except Exception as e:
                 print(f"⚠️ Local edit final progress update failed: {e}")
-            return result_url, "✅ " + message
+            
+            return final_result, final_message
         else:
             print(f"❌ Local editing processing failed - IP: {client_ip}, error: {message}", flush=True)
             return None, "❌ " + message
@@ -363,20 +506,17 @@ def create_app():
             width: 100%;
         }
         """,
-        # 改善并发性能的配置
+        # Improve concurrency performance configuration
         head="""
         <script>
-        // 减少客户端状态更新频率，避免过度的 SSE 连接
+        // Reduce client-side state update frequency, avoid excessive SSE connections
         if (window.gradio) {
-            window.gradio.update_frequency = 2000; // 2秒更新一次
+            window.gradio.update_frequency = 2000; // Update every 2 seconds
         }
         </script>
         """
     ) as app:
         
-        # 减少State组件，只保留必要的
-        # 移除了大部分State组件以减少状态管理复杂度
-        
         gr.Markdown(
             """
             # 🎨 AI Image Editor
@@ -384,6 +524,15 @@ def create_app():
             elem_classes=["main-container"]
         )
         
+        # Powered by line below title
+        gr.HTML("""
+        <div style="text-align: center; margin: -10px auto 30px auto;">
+            <p style="margin: 0; font-size: 16px; color: #999; font-weight: 400;">
+                powered by <a href="https://omnicreator.net/#generator" target="_blank" style="color: #667eea; text-decoration: none;">omnicreator.net</a>
+            </p>
+        </div>
+        """, padding=False)
+        
         with gr.Tabs():
             # Global editing tab
             with gr.Tab("🌍 Global Editing"):
@@ -419,7 +568,7 @@ def create_app():
                             elem_classes=["result-area"]
                         )
                         
-                        # 添加 "Use as Input" 按钮
+                        # Add "Use as Input" button
                         use_as_input_btn = gr.Button(
                             "🔄 Use as Input",
                             variant="secondary",
@@ -454,18 +603,18 @@ def create_app():
                             outputs=prompt_input
                         )
                 
-                # 绑定按钮点击事件 - 简��，移除状态管理
+                # Bind button click events - simplified, remove state management
                 edit_button.click(
                     fn=edit_image_interface,
                     inputs=[input_image, prompt_input],
                     outputs=[output_image, status_output],
                     show_progress=True,
-                    # 增加并发设置
-                    concurrency_limit=10,  # 限制并发数
+                    # Increase concurrency settings
+                    concurrency_limit=10,  # Limit concurrent requests
                     api_name="global_edit"
                 )
                 
-                # 简化 "Use as Input" 按钮，直接复制图片
+                # Simplify "Use as Input" button, directly copy image
                 def simple_use_as_input(output_img):
                     if output_img is not None:
                         return output_img
@@ -486,7 +635,7 @@ def create_app():
                             label="Upload image and draw mask",
                             type="pil",
                             height=512,
-                            brush=gr.Brush(colors=["#ff0000"], default_size=60),
+                            brush=gr.Brush(colors=["#ff0000"], default_size=180),
                             elem_classes=["upload-area"]
                         )
                         
@@ -512,7 +661,7 @@ def create_app():
                             elem_classes=["result-area"]
                         )
                         
-                        # 添加 "Use as Input" 按钮
+                        # Add "Use as Input" button
                         local_use_as_input_btn = gr.Button(
                             "🔄 Use as Input",
                             variant="secondary",
@@ -547,21 +696,21 @@ def create_app():
                             outputs=local_prompt_input
                         )
                 
-                # 绑定局部编辑按钮点击事件 - 简化，移除状态管理
+                # Bind local edit button click events - simplified, remove state management
                 local_edit_button.click(
                     fn=local_edit_interface,
                     inputs=[local_input_image, local_prompt_input],
                     outputs=[local_output_image, local_status_output],
                     show_progress=True,
-                    # 增加并发设置
-                    concurrency_limit=8,  # 局部编辑更复杂，限制更少的并发
+                    # Increase concurrency settings
+                    concurrency_limit=8,  # Local editing is more complex, allow fewer concurrent requests
                     api_name="local_edit"
                 )
                 
-                # 简化局部编辑 "Use as Input" 按钮
+                # Simplify local edit "Use as Input" button
                 def simple_local_use_as_input(output_img):
                     if output_img is not None:
-                        # 创建简单的 ImageEditor 格式
+                        # Create simple ImageEditor format
                         editor_data = {
                             "background": output_img, 
                             "layers": [],
@@ -575,22 +724,303 @@ def create_app():
                     inputs=[local_output_image],
                     outputs=[local_input_image]
                 )
+        
+        # SEO Content Section
+        gr.HTML("""
+    <div style="width: 100%; margin: 50px 0; padding: 0 20px;">
+        
+        <!-- Showcase Section -->
+        <div style="text-align: center; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; padding: 40px; border-radius: 20px; margin: 40px 0;">
+            <h2 style="margin: 0 0 20px 0; font-size: 2.2em; font-weight: 700;">
+                🎨 Unlimited AI Image Generation & Editing
+            </h2>
+            <p style="margin: 0 0 25px 0; font-size: 1.2em; opacity: 0.95; line-height: 1.6;">
+                Experience the ultimate freedom in AI image creation! Generate and edit unlimited images without restrictions, 
+                including NSFW content, with our premium AI image editing platform.
+            </p>
+            
+            <div style="display: flex; justify-content: center; gap: 25px; flex-wrap: wrap; margin: 30px 0;">
+                <a href="https://omnicreator.net/#generator" target="_blank" style="
+                    display: inline-flex; 
+                    align-items: center;
+                    justify-content: center;
+                    padding: 20px 40px; 
+                    background: linear-gradient(135deg, #ff6b6b 0%, #feca57 100%);
+                    color: white; 
+                    text-decoration: none; 
+                    border-radius: 15px; 
+                    font-weight: 700;
+                    font-size: 18px;
+                    text-align: center;
+                    min-width: 250px;
+                    box-shadow: 0 8px 25px rgba(255, 107, 107, 0.4);
+                    transition: all 0.3s ease;
+                    border: none;
+                    transform: scale(1);
+                " onmouseover="this.style.transform='scale(1.05)'" onmouseout="this.style.transform='scale(1)'">
+                    🚀 Get Unlimited Access Now
+                </a>
+                
+            </div>
+            
+            <p style="color: rgba(255,255,255,0.9); font-size: 1em; margin: 20px 0 0 0;">
+                Join thousands of creators who trust Omni Creator for unrestricted AI image generation!
+            </p>
+        </div>
+
+        <!-- Hero Description -->
+        <div style="text-align: center; margin: 25px auto; background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%); padding: 35px; border-radius: 20px; box-shadow: 0 10px 30px rgba(0,0,0,0.1);">
+            <h2 style="color: #2c3e50; margin: 0 0 20px 0; font-size: 1.9em; font-weight: 700;">
+                🌟 Professional AI Image Editor - No Restrictions
+            </h2>
+            <p style="color: #555; font-size: 1.1em; line-height: 1.6; margin: 0 0 20px 0; padding: 0 20px;">
+                Transform your creative vision into reality with our advanced AI image editing platform. Whether you're creating 
+                art, editing photos, designing content, or working with any type of imagery - our powerful AI removes all limitations 
+                and gives you complete creative freedom.
+            </p>
+        </div>
+
+        <!-- Features Grid -->
+        <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(300px, 1fr)); gap: 25px; margin: 40px 0;">
+            
+            <div style="background: white; padding: 30px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #e74c3c;">
+                <h3 style="color: #e74c3c; margin: 0 0 15px 0; font-size: 1.4em; font-weight: 600;">
+                    🎯 Unlimited Generation
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    Premium users enjoy unlimited image generation without daily limits, rate restrictions, or content barriers. 
+                    Create as many images as you need, whenever you need them.
+                </p>
+            </div>
+
+            <div style="background: white; padding: 30px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #3498db;">
+                <h3 style="color: #3498db; margin: 0 0 15px 0; font-size: 1.4em; font-weight: 600;">
+                    🔓 No Content Restrictions
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    Generate and edit any type of content without NSFW filters or content limitations. Complete creative 
+                    freedom for artists, designers, and content creators.
+                </p>
+            </div>
+
+            <div style="background: white; padding: 30px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #27ae60;">
+                <h3 style="color: #27ae60; margin: 0 0 15px 0; font-size: 1.4em; font-weight: 600;">
+                    ⚡ Lightning Fast Processing
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    Advanced AI infrastructure delivers high-quality results in seconds. No waiting in queues, 
+                    no processing delays - just instant, professional-grade image editing.
+                </p>
+            </div>
+
+            <div style="background: white; padding: 30px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #9b59b6;">
+                <h3 style="color: #9b59b6; margin: 0 0 15px 0; font-size: 1.4em; font-weight: 600;">
+                    🎨 Advanced Editing Tools
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    Global transformations, precision local editing, style transfer, object removal, background replacement, 
+                    and dozens of other professional editing capabilities.
+                </p>
+            </div>
+
+            <div style="background: white; padding: 30px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #f39c12;">
+                <h3 style="color: #f39c12; margin: 0 0 15px 0; font-size: 1.4em; font-weight: 600;">
+                    💎 Premium Quality
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    State-of-the-art AI models trained on millions of images deliver exceptional quality and realism. 
+                    Professional results suitable for commercial use and high-end projects.
+                </p>
+            </div>
+
+            <div style="background: white; padding: 30px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.08); border-left: 5px solid #34495e;">
+                <h3 style="color: #34495e; margin: 0 0 15px 0; font-size: 1.4em; font-weight: 600;">
+                    🌍 Multi-Modal Support
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    Support for all image formats, styles, and use cases. From photorealistic portraits to artistic creations, 
+                    product photography to digital art - we handle everything.
+                </p>
+            </div>
+
+        </div>
+
+        <!-- Premium Benefits Section -->
+        <div style="background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%); color: white; padding: 40px; border-radius: 20px; margin: 40px 0; text-align: center;">
+            <h2 style="margin: 0 0 25px 0; font-size: 1.8em; font-weight: 700;">
+                💎 Why Choose Omni Creator Premium?
+            </h2>
+            <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(250px, 1fr)); gap: 20px; margin: 30px 0;">
+                
+                <div style="background: rgba(255,255,255,0.15); padding: 20px; border-radius: 12px;">
+                    <h4 style="margin: 0 0 10px 0; font-size: 1.2em;">🚫 No Rate Limits</h4>
+                    <p style="margin: 0; opacity: 0.9; font-size: 0.95em;">Generate unlimited images without waiting periods or daily restrictions</p>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.15); padding: 20px; border-radius: 12px;">
+                    <h4 style="margin: 0 0 10px 0; font-size: 1.2em;">🎭 Unrestricted Content</h4>
+                    <p style="margin: 0; opacity: 0.9; font-size: 0.95em;">Create any type of content without NSFW filters or censorship</p>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.15); padding: 20px; border-radius: 12px;">
+                    <h4 style="margin: 0 0 10px 0; font-size: 1.2em;">⚡ Priority Processing</h4>
+                    <p style="margin: 0; opacity: 0.9; font-size: 0.95em;">Skip queues and get instant results with dedicated processing power</p>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.15); padding: 20px; border-radius: 12px;">
+                    <h4 style="margin: 0 0 10px 0; font-size: 1.2em;">🎨 Advanced Features</h4>
+                    <p style="margin: 0; opacity: 0.9; font-size: 0.95em;">Access to latest AI models and cutting-edge editing capabilities</p>
+                </div>
+
+            </div>
+            <div style="display: flex; justify-content: center; margin: 25px 0 0 0;">
+                <a href="https://omnicreator.net/#generator" target="_blank" style="
+                    display: inline-flex; 
+                    align-items: center;
+                    justify-content: center;
+                    padding: 18px 35px; 
+                    background: rgba(255,255,255,0.9);
+                    color: #333; 
+                    text-decoration: none; 
+                    border-radius: 15px; 
+                    font-weight: 700;
+                    font-size: 16px;
+                    text-align: center;
+                    min-width: 200px;
+                    box-shadow: 0 6px 20px rgba(0,0,0,0.3);
+                    transition: all 0.3s ease;
+                    border: none;
+                ">🌟 Start Creating Now</a>
+            </div>
+        </div>
+
+        <!-- Tips Section -->
+        <div style="background: linear-gradient(135deg, #ff9a9e 0%, #fecfef 50%, #fecfef 100%); padding: 30px; border-radius: 15px; margin: 40px 0;">
+            <h3 style="color: #8b5cf6; text-align: center; margin: 0 0 25px 0; font-size: 1.5em; font-weight: 700;">
+                💡 Pro Tips for Best Results
+            </h3>
+            <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(300px, 1fr)); gap: 18px;">
+                
+                <div style="background: rgba(255,255,255,0.85); padding: 18px; border-radius: 12px;">
+                    <strong style="color: #8b5cf6; font-size: 1.1em;">📝 Clear Descriptions:</strong>
+                    <p style="color: #555; margin: 5px 0 0 0; line-height: 1.5;">Use detailed, specific prompts for better results. Describe colors, styles, lighting, and composition clearly.</p>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.85); padding: 18px; border-radius: 12px;">
+                    <strong style="color: #8b5cf6; font-size: 1.1em;">🎯 Local Editing:</strong>
+                    <p style="color: #555; margin: 5px 0 0 0; line-height: 1.5;">Use precise brush strokes to select areas for local editing. Smaller, focused edits often yield better results.</p>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.85); padding: 18px; border-radius: 12px;">
+                    <strong style="color: #8b5cf6; font-size: 1.1em;">⚡ Iterative Process:</strong>
+                    <p style="color: #555; margin: 5px 0 0 0; line-height: 1.5;">Use "Use as Input" feature to refine results. Multiple iterations can achieve complex transformations.</p>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.85); padding: 18px; border-radius: 12px;">
+                    <strong style="color: #8b5cf6; font-size: 1.1em;">🖼️ Image Quality:</strong>
+                    <p style="color: #555; margin: 5px 0 0 0; line-height: 1.5;">Higher resolution input images (up to 10MB) generally produce better editing results and finer details.</p>
+                </div>
+
+            </div>
+        </div>
+
+        <!-- Use Cases Section -->
+        <div style="text-align: center; margin: 25px auto; background: linear-gradient(135deg, #a8edea 0%, #fed6e3 100%); padding: 35px; border-radius: 20px; box-shadow: 0 10px 30px rgba(0,0,0,0.1);">
+            <h2 style="color: #2c3e50; margin: 0 0 20px 0; font-size: 1.8em; font-weight: 700;">
+                🚀 Perfect For Every Creative Need
+            </h2>
+            <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(200px, 1fr)); gap: 20px; margin: 25px 0; text-align: left;">
+                
+                <div style="background: rgba(255,255,255,0.8); padding: 20px; border-radius: 12px;">
+                    <h4 style="color: #e74c3c; margin: 0 0 10px 0;">🎨 Digital Art</h4>
+                    <ul style="color: #555; margin: 0; padding-left: 18px; line-height: 1.6;">
+                        <li>Character design</li>
+                        <li>Concept art</li>
+                        <li>Style transfer</li>
+                        <li>Artistic effects</li>
+                    </ul>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.8); padding: 20px; border-radius: 12px;">
+                    <h4 style="color: #3498db; margin: 0 0 10px 0;">📸 Photography</h4>
+                    <ul style="color: #555; margin: 0; padding-left: 18px; line-height: 1.6;">
+                        <li>Background replacement</li>
+                        <li>Object removal</li>
+                        <li>Lighting adjustment</li>
+                        <li>Portrait enhancement</li>
+                    </ul>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.8); padding: 20px; border-radius: 12px;">
+                    <h4 style="color: #27ae60; margin: 0 0 10px 0;">🛍️ E-commerce</h4>
+                    <ul style="color: #555; margin: 0; padding-left: 18px; line-height: 1.6;">
+                        <li>Product photography</li>
+                        <li>Lifestyle shots</li>
+                        <li>Color variations</li>
+                        <li>Context placement</li>
+                    </ul>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.8); padding: 20px; border-radius: 12px;">
+                    <h4 style="color: #9b59b6; margin: 0 0 10px 0;">📱 Social Media</h4>
+                    <ul style="color: #555; margin: 0; padding-left: 18px; line-height: 1.6;">
+                        <li>Content creation</li>
+                        <li>Meme generation</li>
+                        <li>Brand visuals</li>
+                        <li>Viral content</li>
+                    </ul>
+                </div>
+
+            </div>
+            <div style="text-align: center; margin: 25px 0 0 0;">
+                <a href="https://omnicreator.net/#generator" target="_blank" style="
+                    display: inline-flex; 
+                    align-items: center;
+                    justify-content: center;
+                    padding: 18px 35px; 
+                    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                    color: white; 
+                    text-decoration: none; 
+                    border-radius: 15px; 
+                    font-weight: 700;
+                    font-size: 16px;
+                    text-align: center;
+                    min-width: 220px;
+                    box-shadow: 0 8px 25px rgba(102, 126, 234, 0.4);
+                    transition: all 0.3s ease;
+                    border: none;
+                ">🎯 Start Your Project Now</a>
+            </div>
+        </div>
+
+    </div>
+    
+    <!-- Powered by footer -->
+    <div style="text-align: center; margin: 30px auto 20px auto; padding: 20px;">
+        <p style="margin: 0 0 10px 0; font-size: 18px; color: #333; font-weight: 500;">
+            Powered by <a href="https://omnicreator.net/#generator" target="_blank" style="color: #667eea; text-decoration: none; font-weight: bold;">Omni Creator</a>
+        </p>
+        <p style="margin: 0; font-size: 14px; color: #999; font-weight: 400;">
+            The ultimate AI image generation and editing platform • Unlimited creativity, zero restrictions
+        </p>
+    </div>
+    """, padding=False)
     
     return app
 
 if __name__ == "__main__":
     app = create_app()
-    # 改善队列配置以处理高并发和防止 SSE 连接问题
+    # Improve queue configuration to handle high concurrency and prevent SSE connection issues
     app.queue(
-        default_concurrency_limit=20,  # 默认并发限制
-        max_size=50,  # 队列最大大小
-        api_open=False  # 关闭 API 访问，减少资源消耗
+        default_concurrency_limit=20,  # Default concurrency limit
+        max_size=50,  # Maximum queue size
+        api_open=False  # Close API access to reduce resource consumption
     )
     app.launch(
         server_name="0.0.0.0",
-        show_error=True,  # 显示详细错误信息
-        quiet=False,  # 保持日志输出
-        max_threads=40,  # 增加线程池大小
+        show_error=True,  # Show detailed error information
+        quiet=False,  # Keep log output
+        max_threads=40,  # Increase thread pool size
         height=800,
-        favicon_path=None  # 减少资源加载
+        favicon_path=None  # Reduce resource loading
     )

util.py

@@ -77,8 +77,8 @@ class R2Api:
 
     def __init__(self, session=None):
         super().__init__()
-        self.R2_BUCKET = "trump-ai-voice"
-        self.domain = "https://www.trumpaivoice.net/"
+        self.R2_BUCKET = "omni-creator"
+        self.domain = "https://www.omnicreator.net/"
         self.R2_ACCESS_KEY = R2_ACCESS_KEY
         self.R2_SECRET_KEY = R2_SECRET_KEY
         self.R2_ENDPOINT = R2_ENDPOINT
@@ -290,7 +290,7 @@ def upload_mask_image_r2(client_ip, time_id, mask_image):
 
 def submit_image_edit_task(user_image_url, prompt, task_type="80", mask_image_url=""):
     """
-    Submit image editing task with improved error handling
+    Submit image editing task with improved error handling using API v2
     """
     headers = {
         'Content-Type': 'application/json',
@@ -299,11 +299,12 @@ def submit_image_edit_task(user_image_url, prompt, task_type="80", mask_image_ur
     
     data = {
         "user_image": user_image_url,
-        "mask_image": mask_image_url,
-        "task_type": task_type,
-        "prompt": prompt,
-        "secret_key": "219ngu",
-        "is_private": "0"
+        "user_mask": mask_image_url,
+        "type": task_type,
+        "text": prompt,
+        "user_uuid": APIKEY,
+        "priority": 0,
+        "secret_key": "219ngu"
     }
     
     retry_count = 0
@@ -312,7 +313,7 @@ def submit_image_edit_task(user_image_url, prompt, task_type="80", mask_image_ur
     while retry_count < max_retries:
         try:
             response = requests.post(
-                f'{UKAPIURL}/public_image_edit',
+                f'{UKAPIURL}/public_image_edit_v2',
                 headers=headers,
                 json=data,
                 timeout=30  # 增加超时时间
@@ -351,7 +352,7 @@ def submit_image_edit_task(user_image_url, prompt, task_type="80", mask_image_ur
 
 def check_task_status(task_id):
     """
-    Query task status with improved error handling
+    Query task status with improved error handling using API v2
     """
     headers = {
         'Content-Type': 'application/json',
@@ -368,7 +369,7 @@ def check_task_status(task_id):
     while retry_count < max_retries:
         try:
             response = requests.post(
-                f'{UKAPIURL}/status_image_edit',
+                f'{UKAPIURL}/status_image_edit_v2',
                 headers=headers,
                 json=data,
                 timeout=15  # 状态查询超时时间短一些
@@ -378,7 +379,7 @@ def check_task_status(task_id):
                 result = response.json()
                 if result.get('code') == 0:
                     task_data = result['data']
-                    return task_data['status'], task_data.get('output1'), task_data
+                    return task_data['status'], task_data.get('image_url'), task_data
                 else:
                     return 'error', None, result.get('message', 'Unknown error')
             elif response.status_code in [502, 503, 504]:  # 服务器错误，可以重试
@@ -577,6 +578,43 @@ def process_local_image_edit(base_image, layers, prompt, progress_callback=None)
         return None, f"error occurred during processing: {str(e)}"
 
 
+def download_and_check_result_nsfw(image_url, nsfw_detector=None):
+    """
+    下载结果图片并进行NSFW检测
+    
+    Args:
+        image_url (str): 结果图片URL
+        nsfw_detector: NSFW检测器实例
+        
+    Returns:
+        tuple: (is_nsfw, error_message)
+    """
+    if nsfw_detector is None:
+        return False, None
+    
+    try:
+        # 下载图片
+        response = requests.get(image_url, timeout=30)
+        if response.status_code != 200:
+            return False, f"Failed to download result image: HTTP {response.status_code}"
+        
+        # 将图片数据转换为PIL Image
+        image_data = io.BytesIO(response.content)
+        result_image = Image.open(image_data)
+        
+        # 进行NSFW检测
+        nsfw_result = nsfw_detector.predict_pil_label_only(result_image)
+        
+        is_nsfw = nsfw_result.lower() == "nsfw"
+        print(f"🔍 结果图片NSFW检测: {'❌❌❌ ' + nsfw_result if is_nsfw else '✅✅✅ ' + nsfw_result}")
+        
+        return is_nsfw, None
+        
+    except Exception as e:
+        print(f"⚠️ 结果图片NSFW检测失败: {e}")
+        return False, f"Failed to check result image: {str(e)}"
+
+
 if __name__ == "__main__":
     
     pass