init

.gitattributes

@@ -33,3 +33,11 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+# Image files
+*.jpg filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.gif filter=lfs diff=lfs merge=lfs -text
+*.bmp filter=lfs diff=lfs merge=lfs -text
+*.webp filter=lfs diff=lfs merge=lfs -text
+__pycache__/

.gitignore

@@ -0,0 +1,7 @@
+# Ignore sample images - users will upload their own
+samples/
+*.DS_Store
+__pycache__/
+*.pyc
+*.pyo
+*.egg-info/

app.py

@@ -0,0 +1,899 @@
+import gradio as gr
+import threading
+import os
+import shutil
+import tempfile
+import time
+from util import process_image_edit, OPENROUTER_API_KEY
+
+# Configuration parameters for watermark removal
+FREE_TRY_N = 8         # Free phase: first 8 tries without restrictions
+SLOW_TRY_N = 20        # Slow phase start: 20 tries  
+RATE_LIMIT_20 = 20     # Full restriction: blocked after 20 tries
+
+# Time window configuration (minutes)
+PHASE_1_WINDOW = 5     # 8-20 tries: 5 minutes
+MAX_IMAGES_PER_WINDOW = 2  # Max images per time window
+
+IP_Dict = {}
+# 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 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:
+        str: 'free', 'rate_limit_1', 'blocked'
+    """
+    count = get_ip_generation_count(client_ip)
+    
+    if count < FREE_TRY_N:           # 0-7 tries
+        return 'free'
+    elif count < SLOW_TRY_N:         # 8-19 tries
+        return 'rate_limit_1'        # 5 minutes 2 images
+    else:                            # 20+ tries
+        return 'blocked'             # Generation blocked
+
+def check_rate_limit_for_phase(client_ip, phase):
+    """
+    Check rate limit for specific phase
+    
+    Returns:
+        tuple: (is_limited, wait_time_minutes, current_count)
+    """
+    if phase not in ['rate_limit_1']:
+        return False, 0, 0
+    
+    # Determine time window
+    window_minutes = PHASE_1_WINDOW    # 5 minutes
+    
+    current_time = time.time()
+    window_key = f"{client_ip}_{phase}"
+    
+    # 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
+        }
+    
+    track_data = IP_Rate_Limit_Track[window_key]
+    
+    # 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, track_data['count']
+
+def record_generation_attempt(client_ip, phase):
+    """
+    Record generation attempt
+    """
+    # Increment total count
+    increment_ip_generation_count(client_ip)
+    
+    # Record time window count
+    if phase in ['rate_limit_1']:
+        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
+            }
+
+# Load sample images from samples directory
+def load_sample_images():
+    """Load sample images from samples directory"""
+    samples_dir = "samples"
+    sample_images = []
+    
+    if os.path.exists(samples_dir):
+        for filename in os.listdir(samples_dir):
+            if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
+                file_path = os.path.join(samples_dir, filename)
+                sample_images.append((file_path, filename))
+    
+    return sample_images
+
+# Pre-processed results for sample images (local files)
+SAMPLE_RESULTS = {
+    "sample01.jpeg": "samples/sample01_rm.jpeg",
+    "sample02.jpeg": "samples/sample02_rm.jpeg", 
+    "1.jpg": "samples/1_rm.jpg",
+    "4.jpg": "samples/4_rm.jpg",
+    "10.jpg": "samples/10_rm.jpg"
+}
+
+def generate_smart_watermark_prompt(image_url):
+    """Generate smart watermark removal prompt using OpenRouter API"""
+    import requests
+    import json
+    
+    try:
+        api_key = OPENROUTER_API_KEY if 'OPENROUTER_API_KEY' in globals() else ""
+        if not api_key:
+            print("Warning: OPENROUTER_API_KEY not found")
+            return "remove watermark in the image"
+        
+        headers = {
+            'Content-Type': 'application/json',
+            'Authorization': f'Bearer {api_key}'
+        }
+        
+        data = {
+            "model": "google/gemini-2.5-flash-lite",
+            "messages": [
+                {"role": "user", "content": "现在用户输入了一张带水印的图片，需要生成修复prompt来引导图像生成模型去除水印。你需要找到水印，然后输出去除水印的英文prompt，指出水印的位置和特点，让模型知道即可，尽量简短。直接输出prompt即可。举例1: 'Remove the blue watermark in the lower right corner of the document'. 举例2: 'Remove the white watermark over the whole picture'"},
+                {"role": "assistant", "content": "好的"},
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "text", "text": "Generate an English prompt to remove the watermark in this image."},
+                        {"type": "image_url", "image_url": {"url": image_url}}
+                    ]
+                }
+            ]
+        }
+        
+        response = requests.post(
+            'https://openrouter.ai/api/v1/chat/completions',
+            headers=headers,
+            json=data,
+            timeout=30
+        )
+        
+        if response.status_code == 200:
+            result = response.json()
+            if result.get("choices") and len(result["choices"]) > 0:
+                prompt = result["choices"][0]["message"]["content"].strip()
+                print(f"✅ Generated smart prompt: {prompt}")
+                return prompt
+        
+        print(f"⚠️ OpenRouter API failed with status {response.status_code}")
+        return "remove watermark in the image"
+        
+    except Exception as e:
+        print(f"⚠️ Error generating smart prompt: {e}")
+        return "remove watermark in the image"
+
+def remove_watermark_interface(input_image, force_removal, request: gr.Request, progress=gr.Progress()):
+    """
+    Interface function for watermark removal with phase-based limitations
+    """
+    try:
+        # Extract user IP
+        client_ip = request.client.host
+        x_forwarded_for = dict(request.headers).get('x-forwarded-for')
+        if x_forwarded_for:
+            client_ip = x_forwarded_for   
+        if client_ip not in IP_Dict:
+            IP_Dict[client_ip] = 0
+        IP_Dict[client_ip] += 1
+
+        if input_image is None:
+            return None, "Please upload an image first", gr.update(visible=False)
+        
+        # Set initial prompt for watermark removal
+        prompt = "remove watermark in the image"
+    except Exception as e:
+        print(f"⚠️ Request preprocessing error: {e}")
+        return None, "❌ Request processing error", gr.update(visible=False)
+    
+    # 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':
+        # Generate blocked limit button
+        blocked_button_html = f"""
+        <div style='display: flex; justify-content: center; gap: 15px; margin: 10px 0 5px 0; padding: 0px;'>
+            <a href='https://omnicreator.net/remove-watermark#generator' target='_blank' style='
+                display: inline-flex; 
+                align-items: center;
+                justify-content: center;
+                padding: 16px 32px; 
+                background: linear-gradient(135deg, #e74c3c 0%, #c0392b 100%);
+                color: white; 
+                text-decoration: none; 
+                border-radius: 12px; 
+                font-weight: 600;
+                font-size: 16px;
+                text-align: center;
+                min-width: 200px;
+                box-shadow: 0 4px 15px rgba(231, 76, 60, 0.4);
+                transition: all 0.3s ease;
+                border: none;
+            '>🚀 Remove More Watermarks</a>
+        </div>
+        """
+        return None, f"❌ You have reached Hugging Face's free watermark removal limit. Please visit https://omnicreator.net/remove-watermark#generator for unlimited removal", gr.update(value=blocked_button_html, visible=True)
+    
+    # Check rate limit (applies to rate_limit phases)
+    if current_phase in ['rate_limit_1']:
+        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
+            # Generate rate limit button
+            rate_limit_button_html = f"""
+            <div style='display: flex; justify-content: center; gap: 15px; margin: 10px 0 5px 0; padding: 0px;'>
+                <a href='https://omnicreator.net/remove-watermark#generator' target='_blank' style='
+                    display: inline-flex; 
+                    align-items: center;
+                    justify-content: center;
+                    padding: 16px 32px; 
+                    background: linear-gradient(135deg, #f39c12 0%, #e67e22 100%);
+                    color: white; 
+                    text-decoration: none; 
+                    border-radius: 12px; 
+                    font-weight: 600;
+                    font-size: 16px;
+                    text-align: center;
+                    min-width: 200px;
+                    box-shadow: 0 4px 15px rgba(243, 156, 18, 0.4);
+                    transition: all 0.3s ease;
+                    border: none;
+                '>⏰ Skip Wait - Unlimited Watermark Removal</a>
+            </div>
+            """
+            return None, f"❌ You have reached Hugging Face's free watermark removal limit. Please visit https://omnicreator.net/remove-watermark#generator for unlimited removal, or wait {wait_minutes_int} minutes before generating again", gr.update(value=rate_limit_button_html, visible=True)
+    
+    # No NSFW detection needed for watermark removal
+    
+    result_url = None
+    status_message = ""
+    
+    def progress_callback(message):
+        try:
+            nonlocal status_message
+            status_message = message
+            # 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:
+        # 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)
+        
+        # Force removal mode: Generate smart prompt using OpenRouter API
+        if force_removal:
+            if progress is not None:
+                progress(0.3, desc="💪 AI Force Mode: Analyzing watermark pattern...")
+            
+            try:
+                # Upload image to get URL for OpenRouter API
+                from util import upload_user_img_r2
+                time_id = int(time.time())
+                image_url = upload_user_img_r2(client_ip, time_id, input_image)
+                
+                if image_url:
+                    # Generate smart prompt using OpenRouter API
+                    smart_prompt = generate_smart_watermark_prompt(image_url)
+                    if smart_prompt and smart_prompt.strip() != "remove watermark in the image":
+                        prompt = smart_prompt
+                        print(f"🚀 Using AI Force Mode prompt: {prompt}")
+                    else:
+                        print("⚠️ Using fallback prompt")
+                else:
+                    print("⚠️ Failed to upload image for AI analysis, using default prompt")
+                        
+            except Exception as e:
+                print(f"⚠️ Force Mode prompt generation failed: {e}, using default prompt")
+            
+            if progress is not None:
+                progress(0.5, desc="Starting force watermark removal...")
+        
+        # Call image editing processing function
+        result_url, message, task_uuid = process_image_edit(input_image, prompt.strip(), progress_callback)
+        
+        if result_url:
+            print(f"✅ Watermark removal completed successfully - IP: {client_ip}, result_url: {result_url}, task_uuid: {task_uuid}", flush=True)
+            
+            # No NSFW detection needed for watermark removal
+            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}")
+            
+            # Generate action buttons HTML like Trump AI Voice
+            action_buttons_html = ""
+            if task_uuid:
+                task_detail_url = f"https://omnicreator.net/my-creations/task/{task_uuid}"
+                action_buttons_html = f"""
+                <div style='display: flex; justify-content: center; gap: 15px; margin: 10px 0 5px 0; padding: 0px;'>
+                    <a href='{task_detail_url}' target='_blank' style='
+                        display: inline-flex; 
+                        align-items: center;
+                        justify-content: center;
+                        padding: 16px 32px; 
+                        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                        color: white; 
+                        text-decoration: none; 
+                        border-radius: 12px; 
+                        font-weight: 600;
+                        font-size: 16px;
+                        text-align: center;
+                        min-width: 160px;
+                        box-shadow: 0 4px 15px rgba(102, 126, 234, 0.4);
+                        transition: all 0.3s ease;
+                        border: none;
+                    '>🖼️ Download HD Image</a>
+                    <a href='https://omnicreator.net/remove-watermark#generator' target='_blank' style='
+                        display: inline-flex; 
+                        align-items: center;
+                        justify-content: center;
+                        padding: 16px 32px; 
+                        background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%);
+                        color: white; 
+                        text-decoration: none; 
+                        border-radius: 12px; 
+                        font-weight: 600;
+                        font-size: 16px;
+                        text-align: center;
+                        min-width: 160px;
+                        box-shadow: 0 4px 15px rgba(17, 153, 142, 0.4);
+                        transition: all 0.3s ease;
+                        border: none;
+                    '>🚀 Remove More Watermarks</a>
+                </div>
+                
+                <div style='text-align: center; margin: 15px 0; padding: 15px; background: linear-gradient(135deg, #fff5f5 0%, #fed7e2 100%); border-radius: 12px; border-left: 4px solid #f56565;'>
+                    <p style='margin: 0 0 10px 0; color: #4a5568; font-size: 14px; font-weight: 500;'>
+                        Not satisfied with the removal result?
+                    </p>
+                    <a href='https://omnicreator.net/remove-watermark' target='_blank' style='
+                display: inline-flex; 
+                align-items: center;
+                justify-content: center;
+                        padding: 12px 24px; 
+                        background: linear-gradient(135deg, #f56565 0%, #e53e3e 100%);
+                color: white; 
+                text-decoration: none; 
+                        border-radius: 8px; 
+                font-weight: 600;
+                        font-size: 14px;
+                text-align: center;
+                        box-shadow: 0 4px 12px rgba(245, 101, 101, 0.4);
+                transition: all 0.3s ease;
+                border: none;
+                    '>💪 Try Force Mode on Website</a>
+                </div>
+                """
+            
+            return final_result, final_message, gr.update(value=action_buttons_html, visible=True)
+        else:
+            print(f"❌ Processing failed - IP: {client_ip}, error: {message}", flush=True)
+            return None, "❌ " + message, gr.update(visible=False)
+            
+    except Exception as e:
+        print(f"❌ Processing exception - IP: {client_ip}, error: {str(e)}")
+        return None, f"❌ Error occurred during processing: {str(e)}", gr.update(visible=False)
+
+# Create Gradio interface
+def create_app():
+    with gr.Blocks(
+        title="AI Watermark Remover",
+        theme=gr.themes.Soft(),
+        css="""
+        .main-container {
+            max-width: 1200px;
+            margin: 0 auto;
+        }
+        .upload-area {
+            border: 2px dashed #ccc;
+            border-radius: 10px;
+            padding: 20px;
+            text-align: center;
+        }
+        .result-area {
+            margin-top: 20px;
+            padding: 20px;
+            border-radius: 10px;
+            background-color: #f8f9fa;
+        }
+        .use-as-input-btn {
+            margin-top: 10px;
+            width: 100%;
+        }
+        """,
+        # Improve concurrency performance configuration
+        head="""
+        <script>
+        // Reduce client-side state update frequency, avoid excessive SSE connections
+        if (window.gradio) {
+            window.gradio.update_frequency = 2000; // Update every 2 seconds
+        }
+        </script>
+        """
+    ) as app:
+        
+        # Main title - styled for watermark removal
+        gr.HTML("""
+        <div style="text-align: center; margin: 5px auto 0px auto; max-width: 800px;">
+            <h1 style="color: #2c3e50; margin: 0; font-size: 3.5em; font-weight: 800; letter-spacing: 3px; text-shadow: 2px 2px 4px rgba(0,0,0,0.1);">
+                💧 AI Watermark Remover
+            </h1>
+        </div>
+        """, padding=False)
+        
+        # Powered by line below title - styled like Trump AI Voice
+        gr.HTML("""
+        <div style="text-align: center; margin: 0px auto -5px auto;">
+            <p style="margin: 0; font-size: 16px; color: #999; font-weight: 400;">
+                powered by <a href="https://omnicreator.net/remove-watermark#generator" target="_blank" style="color: #667eea; text-decoration: none;">omnicreator.net</a>
+            </p>
+        </div>
+        """, padding=False)
+        
+        # Load sample images
+        sample_images = load_sample_images()
+        
+        # Watermark removal interface (New 2-column layout)
+        with gr.Row():
+            # --- LEFT COLUMN (INPUT & CONTROLS) ---
+            with gr.Column(scale=1):
+                gr.Markdown("### 📸 Upload Image with Watermark")
+                input_image = gr.Image(
+                    label="Select image to remove watermark",
+                    type="pil",
+                    height=450,
+                    elem_classes=["upload-area"]
+                )
+                
+                force_removal = gr.Checkbox(
+                    label="💪 Force Removal",
+                    info="Ultimate AI-powered watermark removal. AI analyzes the watermark for better results.",
+                    value=False
+                )
+                
+                remove_button = gr.Button(
+                    "💧 Remove Watermark",
+                    variant="primary",
+                    size="lg"
+                )
+            
+            # --- RIGHT COLUMN (OUTPUT) ---
+            with gr.Column(scale=1):
+                gr.Markdown("### 🎯 Watermark Removal Result")
+                output_image = gr.Image(
+                    label="Image without watermark",
+                    height=450,
+                    elem_classes=["result-area"]
+                )
+                        
+                use_as_input_btn = gr.Button(
+                    "🔄 Use as Input",
+                    variant="secondary",
+                    size="sm",
+                    elem_classes=["use-as-input-btn"]
+                )
+                
+                status_output = gr.Textbox(
+                    label="Processing status",
+                    lines=2,
+                    max_lines=3,
+                    interactive=False
+                )
+                
+                action_buttons = gr.HTML(visible=False)
+                
+        # Sample images gallery (moved below main interface)
+        sample_buttons = []
+        if sample_images:
+            gr.Markdown("### 💡 Try with Sample Images")
+            with gr.Row():
+                for idx, (img_path, img_name) in enumerate(sample_images[:4]):  # Show first 4 samples
+                    with gr.Column(scale=1):
+                        sample_img = gr.Image(
+                            value=img_path,
+                            label=img_name,
+                            height=150,
+                            interactive=False
+                        )
+                        use_sample_btn = gr.Button(
+                            f"Use {img_name}",
+                            size="sm",
+                            variant="secondary"
+                        )
+                        sample_buttons.append((use_sample_btn, img_path))
+            
+            # Add sample image button functionality with demo processing
+            def create_sample_handler(img_path, img_name):
+                def handler():
+                    import time
+                    import os
+                    from PIL import Image
+                    
+                    # Wait 3 seconds as requested
+                    time.sleep(3)
+                    
+                    # Get the pre-processed result file path
+                    result_file_path = SAMPLE_RESULTS.get(img_name)
+                    
+                    if result_file_path and os.path.exists(result_file_path):
+                        # Load the result image from local file
+                        try:
+                            result_image = Image.open(result_file_path)
+                            return img_path, result_image, "✅ 示例水印移除完成！这是一个预处理的演示结果。"
+                        except Exception as e:
+                            print(f"Failed to load sample result from {result_file_path}: {e}")
+                    
+                    # Fallback: just return the input image
+                    return img_path, None, "✅ 示例图片已加载。点击 '移除水印' 开始处理。"
+                
+                return handler
+            
+            for sample_btn, img_path in sample_buttons:
+                img_name = os.path.basename(img_path)
+                sample_btn.click(
+                    fn=create_sample_handler(img_path, img_name),
+                    outputs=[input_image, output_image, status_output],
+                    show_progress=True
+                )
+        
+        # Bind button click events
+        remove_button.click(
+            fn=remove_watermark_interface,
+            inputs=[input_image, force_removal],
+            outputs=[output_image, status_output, action_buttons],
+            show_progress=True,
+            concurrency_limit=10,
+            api_name="remove_watermark"
+        )
+                
+        # "Use as Input" button functionality
+        def simple_use_as_input(output_img):
+            if output_img is not None:
+                return output_img
+            return None
+                
+        use_as_input_btn.click(
+            fn=simple_use_as_input,
+            inputs=[output_image],
+            outputs=[input_image]
+        )
+        
+        # Watermark Removal SEO Content Section
+        gr.HTML("""
+    <div style="width: 100%; margin: 50px 0; padding: 0 20px;">
+        
+        <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 Watermark Removal
+            </h2>
+            <p style="margin: 0 0 25px 0; font-size: 1.2em; opacity: 0.95; line-height: 1.6;">
+                Remove watermarks from any image instantly with advanced AI technology! Clean your photos and documents 
+                without compromising image quality using our professional watermark removal tool.
+            </p>
+            
+            <div style="display: flex; justify-content: center; gap: 25px; flex-wrap: wrap; margin: 30px 0;">
+                <a href="https://omnicreator.net/remove-watermark#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 Watermark Removal
+                </a>
+                
+            </div>
+            
+            <p style="color: rgba(255,255,255,0.9); font-size: 1em; margin: 20px 0 0 0;">
+                Join thousands of users who trust Omni Creator for professional watermark removal!
+            </p>
+        </div>
+
+        <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 Watermark Remover - Fast & Accurate
+            </h2>
+            <p style="color: #555; font-size: 1.1em; line-height: 1.6; margin: 0 0 20px 0; padding: 0 20px;">
+                Clean your images effortlessly with our advanced AI watermark removal technology. Whether you're dealing with 
+                text watermarks, logo stamps, or transparent overlays - our intelligent algorithm removes them while preserving 
+                the original image quality and details.
+            </p>
+        </div>
+
+        <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 Removal
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    Premium users enjoy unlimited watermark removal without daily limits or processing restrictions. 
+                    Clean as many images as you need, whenever you need them, with no quality loss.
+                </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;">
+                    🎯 Smart Detection
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    AI automatically identifies and removes various types of watermarks including text, logos, stamps, and 
+                    transparent overlays without affecting the main image content.
+                </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 watermark-free results in seconds. No waiting in queues, 
+                    no processing delays - just instant, professional-grade watermark removal.
+                </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;">
+                    🎨 Quality Preservation
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    Intelligent inpainting technology reconstructs watermark areas seamlessly while maintaining original 
+                    image quality, colors, textures, and details perfectly.
+                </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 specifically for watermark removal deliver exceptional results. 
+                    Professional quality output 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;">
+                    🌍 Universal Compatibility
+                </h3>
+                <p style="color: #666; margin: 0; line-height: 1.6; font-size: 1em;">
+                    Works with all image formats (JPG, PNG, GIF, BMP) and removes any type of watermark. From photos to 
+                    documents, product images to artwork - we clean them all perfectly.
+                </p>
+            </div>
+
+        </div>
+
+        <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 Premium Watermark Removal?
+            </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 Processing Limits</h4>
+                    <p style="margin: 0; opacity: 0.9; font-size: 0.95em;">Remove watermarks from 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;">💧 Advanced Removal</h4>
+                    <p style="margin: 0; opacity: 0.9; font-size: 0.95em;">Remove any type of watermark including complex, transparent, and multi-layered marks</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 watermark removal 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;">🎨 Perfect Quality</h4>
+                    <p style="margin: 0; opacity: 0.9; font-size: 0.95em;">Access to latest AI models for highest quality watermark removal results</p>
+                </div>
+
+            </div>
+            <div style="display: flex; justify-content: center; margin: 25px 0 0 0;">
+                <a href="https://omnicreator.net/remove-watermark#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>
+
+        <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 Watermark Removal
+            </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;">📝 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 watermark removal results and finer details.</p>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.85); padding: 18px; border-radius: 12px;">
+                    <strong style="color: #8b5cf6; font-size: 1.1em;">🎯 Clean Images:</strong>
+                    <p style="color: #555; margin: 5px 0 0 0; line-height: 1.5;">Images with simple backgrounds work best. Complex patterns behind watermarks may require multiple processing attempts.</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 remove stubborn or complex watermarks completely.</p>
+                </div>
+
+                <div style="background: rgba(255,255,255,0.85); padding: 18px; border-radius: 12px;">
+                    <strong style="color: #8b5cf6; font-size: 1.1em;">🖼️ File Formats:</strong>
+                    <p style="color: #555; margin: 5px 0 0 0; line-height: 1.5;">JPG and PNG formats work best. Avoid heavily compressed images as they may affect watermark detection accuracy.</p>
+                </div>
+
+            </div>
+        </div>
+
+        <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 Watermark Removal 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;">📄 Documents</h4>
+                    <ul style="color: #555; margin: 0; padding-left: 18px; line-height: 1.6;">
+                        <li>PDFs with watermarks</li>
+                        <li>Scanned documents</li>
+                        <li>Legal papers</li>
+                        <li>Certificates</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>Stock photos</li>
+                        <li>Portrait photos</li>
+                        <li>Event photography</li>
+                        <li>Nature images</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 images</li>
+                        <li>Catalog photos</li>
+                        <li>Supplier images</li>
+                        <li>Brand cleanup</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>Profile pictures</li>
+                        <li>Marketing visuals</li>
+                        <li>Clean presentations</li>
+                    </ul>
+                </div>
+
+            </div>
+            <div style="text-align: center; margin: 25px 0 0 0;">
+                <a href="https://omnicreator.net/remove-watermark#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 Removing Watermarks</a>
+            </div>
+        </div>
+
+    </div>
+    
+    <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/remove-watermark#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 watermark removal platform • Professional results, unlimited processing
+        </p>
+    </div>
+    """, padding=False)
+    
+    return app
+
+if __name__ == "__main__":
+    app = create_app()
+    # Improve queue configuration to handle high concurrency and prevent SSE connection issues
+    app.queue(
+        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,  # Show detailed error information
+        quiet=False,  # Keep log output
+        max_threads=40,  # Increase thread pool size
+        height=800,
+        favicon_path=None  # Reduce resource loading
+    )

push.sh

@@ -0,0 +1,13 @@
+
+# 设置仓库级别用户名
+git config user.name "selfitcamera"
+git config user.email "[email protected]"
+
+# 验证
+git config user.name
+git config user.email
+
+
+git add .
+git commit -m "init"
+git push

requirements.txt

@@ -0,0 +1,10 @@
+gradio==5.42.0
+opencv-python>=4.8.0
+requests>=2.28.0
+func-timeout>=4.3.5
+numpy>=1.24.0
+boto3
+botocore
+onnxruntime
+huggingface_hub>=0.16.0
+Pillow>=9.0.0

util.py

@@ -0,0 +1,600 @@
+
+import os
+import sys
+import cv2
+import json
+import random
+import time
+import datetime
+import requests
+import func_timeout
+import numpy as np
+import gradio as gr
+import boto3
+import tempfile
+import io
+import uuid
+from botocore.client import Config
+from PIL import Image
+
+
+# TOKEN = os.environ['TOKEN']
+# APIKEY = os.environ['APIKEY']
+# UKAPIURL = os.environ['UKAPIURL']
+
+OneKey = os.environ['OneKey'].strip()
+OneKey = OneKey.split("#")
+TOKEN = OneKey[0]
+APIKEY = OneKey[1]
+UKAPIURL = OneKey[2]
+LLMKEY = OneKey[3]
+R2_ACCESS_KEY = OneKey[4]
+R2_SECRET_KEY = OneKey[5]
+R2_ENDPOINT = OneKey[6]
+OPENROUTER_API_KEY = OneKey[7]
+
+
+# tmpFolder is no longer needed since we upload directly from memory
+# tmpFolder = "tmp"
+# os.makedirs(tmpFolder, exist_ok=True)
+
+
+class R2Api:
+
+    def __init__(self, session=None):
+        super().__init__()
+        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
+
+        self.client = boto3.client(
+            "s3",
+            endpoint_url=self.R2_ENDPOINT,
+            aws_access_key_id=self.R2_ACCESS_KEY,
+            aws_secret_access_key=self.R2_SECRET_KEY,
+            config=Config(signature_version="s3v4")
+        )
+
+        self.session = requests.Session() if session is None else session
+
+    def upload_from_memory(self, image_data, filename, content_type='image/jpeg'):
+        """
+        Upload image data directly from memory to R2
+        
+        Args:
+            image_data (bytes): Image data in bytes
+            filename (str): Filename for the uploaded file
+            content_type (str): MIME type of the image
+            
+        Returns:
+            str: URL of the uploaded file
+        """
+        t1 = time.time()
+        headers = {"Content-Type": content_type}
+        
+        cloud_path = f"QwenImageEdit/Uploads/{str(datetime.date.today())}/{filename}"
+        url = self.client.generate_presigned_url(
+            "put_object",
+            Params={"Bucket": self.R2_BUCKET, "Key": cloud_path, "ContentType": content_type},
+            ExpiresIn=604800
+        )
+
+        retry_count = 0
+        while retry_count < 3:
+            try:
+                response = self.session.put(url, data=image_data, headers=headers, timeout=15)
+                if response.status_code == 200:
+                    break
+                else:
+                    print(f"⚠️ Upload failed with status code: {response.status_code}")
+                    retry_count += 1
+            except (requests.exceptions.Timeout, requests.exceptions.RequestException) as e:
+                print(f"⚠️ Upload retry {retry_count + 1}/3 failed: {e}")
+                retry_count += 1
+                if retry_count == 3:
+                    raise Exception(f'Failed to upload file to R2 after 3 retries! Last error: {str(e)}')
+                time.sleep(1)  # 等待1秒后重试
+                continue
+        print("upload_from_memory time is ====>", time.time() - t1)
+        return f"{self.domain}{cloud_path}"
+    
+    def upload_file(self, local_path, cloud_path):
+        t1 = time.time()
+        head_dict = {
+            'jpg': 'image/jpeg',
+            'jpeg': 'image/jpeg',
+            'png': 'image/png',
+            'gif': 'image/gif',
+            'bmp': 'image/bmp',
+            'webp': 'image/webp',
+            'ico': 'image/x-icon'
+        }
+        ftype = os.path.basename(local_path).split(".")[-1].lower()
+        ctype = head_dict.get(ftype, 'application/octet-stream')
+        headers = {"Content-Type": ctype}
+
+
+        cloud_path = f"QwenImageEdit/Uploads/{str(datetime.date.today())}/{os.path.basename(local_path)}"
+        url = self.client.generate_presigned_url(
+            "put_object",
+            Params={"Bucket": self.R2_BUCKET, "Key": cloud_path, "ContentType": ctype},
+            ExpiresIn=604800
+        )
+
+        retry_count = 0
+        while retry_count < 3:
+            try:
+                with open(local_path, 'rb') as f:
+                    self.session.put(url, data=f.read(), headers=headers, timeout=8)
+                break
+            except (requests.exceptions.Timeout, requests.exceptions.RequestException):
+                retry_count += 1
+                if retry_count == 3:
+                    raise Exception('Failed to upload file to R2 after 3 retries!')
+                continue
+        print("upload_file time is ====>", time.time() - t1)
+        return f"{self.domain}{cloud_path}"
+    
+def upload_user_img_r2(clientIp, timeId, pil_image):
+    """
+    Upload PIL Image directly to R2 without saving to local file
+    
+    Args:
+        clientIp (str): Client IP address
+        timeId (int): Timestamp
+        pil_image (PIL.Image): PIL Image object
+        
+    Returns:
+        str: Uploaded URL
+    """
+    # Generate unique filename using UUID to prevent file conflicts in concurrent environment
+    unique_id = str(uuid.uuid4())
+    fileName = f"user_img_{unique_id}_{timeId}.jpg"
+    
+    # Convert PIL Image to bytes
+    img_buffer = io.BytesIO()
+    if pil_image.mode != 'RGB':
+        pil_image = pil_image.convert('RGB')
+    pil_image.save(img_buffer, format='JPEG', quality=95)
+    img_data = img_buffer.getvalue()
+    
+    # Upload directly from memory
+    res = R2Api().upload_from_memory(img_data, fileName, 'image/jpeg')
+    return res
+
+
+
+def create_mask_from_layers(base_image, layers):
+    """
+    Create mask image from ImageEditor layers
+    
+    Args:
+        base_image (PIL.Image): Original image
+        layers (list): ImageEditor layer data
+        
+    Returns:
+        PIL.Image: Black and white mask image
+    """
+    from PIL import Image, ImageDraw
+    import numpy as np
+    
+    # Create blank mask with same size as original image
+    mask = Image.new('L', base_image.size, 0)  # 'L' mode is grayscale, 0 is black
+    
+    if not layers:
+        return mask
+    
+    # Iterate through all layers, set drawn areas to white
+    for layer in layers:
+        if layer is not None:
+            # Convert layer to numpy array
+            layer_array = np.array(layer)
+            
+            # Check layer format
+            if len(layer_array.shape) == 3:  # RGB/RGBA format
+                # If RGBA, check alpha channel
+                if layer_array.shape[2] == 4:
+                    # Use alpha channel as mask
+                    alpha_channel = layer_array[:, :, 3]
+                    # Set non-transparent areas (alpha > 0) to white
+                    mask_array = np.where(alpha_channel > 0, 255, 0).astype(np.uint8)
+                else:
+                    # RGB format, check if not pure black (0,0,0)
+                    # Assume drawn areas are non-black
+                    non_black = np.any(layer_array > 0, axis=2)
+                    mask_array = np.where(non_black, 255, 0).astype(np.uint8)
+            elif len(layer_array.shape) == 2:  # Grayscale
+                # Use grayscale values directly, set non-zero areas to white
+                mask_array = np.where(layer_array > 0, 255, 0).astype(np.uint8)
+            else:
+                continue
+            
+            # Convert mask_array to PIL image and merge into total mask
+            layer_mask = Image.fromarray(mask_array, mode='L')
+            # Resize to match original image
+            if layer_mask.size != base_image.size:
+                layer_mask = layer_mask.resize(base_image.size, Image.LANCZOS)
+            
+            # Merge masks (use maximum value to ensure all drawn areas are included)
+            mask_array_current = np.array(mask)
+            layer_mask_array = np.array(layer_mask)
+            combined_mask_array = np.maximum(mask_array_current, layer_mask_array)
+            mask = Image.fromarray(combined_mask_array, mode='L')
+    
+    return mask
+
+
+def upload_mask_image_r2(client_ip, time_id, mask_image):
+    """
+    Upload mask image to R2 directly from memory
+    
+    Args:
+        client_ip (str): Client IP
+        time_id (int): Timestamp
+        mask_image (PIL.Image): Mask image
+        
+    Returns:
+        str: Uploaded URL
+    """
+    # Generate unique filename using UUID to prevent file conflicts in concurrent environment
+    unique_id = str(uuid.uuid4())
+    file_name = f"mask_img_{unique_id}_{time_id}.png"
+    
+    try:
+        # Convert mask image to bytes
+        img_buffer = io.BytesIO()
+        mask_image.save(img_buffer, format='PNG')
+        img_data = img_buffer.getvalue()
+        
+        # Upload directly from memory
+        res = R2Api().upload_from_memory(img_data, file_name, 'image/png')
+        
+        return res
+    except Exception as e:
+        print(f"Failed to upload mask image: {e}")
+        return None
+
+
+
+def submit_image_edit_task(user_image_url, prompt, task_type="80", mask_image_url=""):
+    """
+    Submit image editing task with improved error handling using API v2
+    """
+    headers = {
+        'Content-Type': 'application/json',
+        'Authorization': f'Bearer {APIKEY}'
+    }
+    
+    data = {
+        "user_image": user_image_url,
+        "user_mask": mask_image_url,
+        "type": task_type,
+        "text": prompt,
+        "user_uuid": APIKEY,
+        "priority": 0,
+        "secret_key": "219ngu"
+    }
+    
+    retry_count = 0
+    max_retries = 3
+    
+    while retry_count < max_retries:
+        try:
+            response = requests.post(
+                f'{UKAPIURL}/public_image_edit_v2',
+                headers=headers,
+                json=data,
+                timeout=30  # 增加超时时间
+            )
+            
+            if response.status_code == 200:
+                result = response.json()
+                if result.get('code') == 0:
+                    return result['data']['task_id'], None
+                else:
+                    return None, f"API Error: {result.get('message', 'Unknown error')}"
+            elif response.status_code in [502, 503, 504]:  # 服务器错误，可以重试
+                retry_count += 1
+                if retry_count < max_retries:
+                    print(f"⚠️ Server error {response.status_code}, retrying {retry_count}/{max_retries}")
+                    time.sleep(2)  # 等待2秒后重试
+                    continue
+                else:
+                    return None, f"HTTP Error after {max_retries} retries: {response.status_code}"
+            else:
+                return None, f"HTTP Error: {response.status_code}"
+                
+        except (requests.exceptions.Timeout, requests.exceptions.ConnectionError) as e:
+            retry_count += 1
+            if retry_count < max_retries:
+                print(f"⚠️ Network error, retrying {retry_count}/{max_retries}: {e}")
+                time.sleep(2)
+                continue
+            else:
+                return None, f"Network error after {max_retries} retries: {str(e)}"
+        except Exception as e:
+            return None, f"Request Exception: {str(e)}"
+    
+    return None, f"Failed after {max_retries} retries"
+
+
+def check_task_status(task_id):
+    """
+    Query task status with improved error handling using API v2
+    """
+    headers = {
+        'Content-Type': 'application/json',
+        'Authorization': f'Bearer {APIKEY}'
+    }
+    
+    data = {
+        "task_id": task_id
+    }
+    
+    retry_count = 0
+    max_retries = 2  # 状态查询重试次数少一些
+    
+    while retry_count < max_retries:
+        try:
+            response = requests.post(
+                f'{UKAPIURL}/status_image_edit_v2',
+                headers=headers,
+                json=data,
+                timeout=15  # 状态查询超时时间短一些
+            )
+            
+            if response.status_code == 200:
+                result = response.json()
+                if result.get('code') == 0:
+                    task_data = result['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]:  # 服务器错误，可以重试
+                retry_count += 1
+                if retry_count < max_retries:
+                    print(f"⚠️ Status check server error {response.status_code}, retrying {retry_count}/{max_retries}")
+                    time.sleep(1)  # 状态查询重试间隔短一些
+                    continue
+                else:
+                    return 'error', None, f"HTTP Error after {max_retries} retries: {response.status_code}"
+            else:
+                return 'error', None, f"HTTP Error: {response.status_code}"
+                
+        except (requests.exceptions.Timeout, requests.exceptions.ConnectionError) as e:
+            retry_count += 1
+            if retry_count < max_retries:
+                print(f"⚠️ Status check network error, retrying {retry_count}/{max_retries}: {e}")
+                time.sleep(1)
+                continue
+            else:
+                return 'error', None, f"Network error after {max_retries} retries: {str(e)}"
+        except Exception as e:
+            return 'error', None, f"Request Exception: {str(e)}"
+    
+    return 'error', None, f"Failed after {max_retries} retries"
+
+
+def process_image_edit(img_input, prompt, progress_callback=None):
+    """
+    Complete process for image editing
+    
+    Args:
+        img_input: Can be file path (str) or PIL Image object
+        prompt: Editing instructions
+        progress_callback: Progress callback function
+    """
+    try:
+        # Generate client IP and timestamp
+        client_ip = "127.0.0.1"  # Default IP
+        time_id = int(time.time())
+        
+        # Process input image - supports PIL Image and file path
+        if hasattr(img_input, 'save'):  # PIL Image object
+            pil_image = img_input
+            print(f"💾 Using PIL Image directly from memory")
+        else:
+            # Load from file path
+            pil_image = Image.open(img_input)
+            print(f"📁 Loaded image from file: {img_input}")
+        
+        if progress_callback:
+            progress_callback("uploading image...")
+        
+        # Upload user image directly from memory
+        uploaded_url = upload_user_img_r2(client_ip, time_id, pil_image)
+        if not uploaded_url:
+            return None, "image upload failed", None
+        
+        # Extract actual image URL from upload URL
+        if "?" in uploaded_url:
+            uploaded_url = uploaded_url.split("?")[0]
+        
+        if progress_callback:
+            progress_callback("submitting edit task...")
+        
+        # Submit image editing task
+        task_id, error = submit_image_edit_task(uploaded_url, prompt)
+        if error:
+            return None, error, None
+        
+        if progress_callback:
+            progress_callback(f"task submitted, ID: {task_id}, processing...")
+        
+        # Wait for task completion
+        max_attempts = 60  # Wait up to 10 minutes
+        task_uuid = None
+        for attempt in range(max_attempts):
+            status, output_url, task_data = check_task_status(task_id)
+            
+            # Extract task_uuid from task_data
+            if task_data and isinstance(task_data, dict):
+                task_uuid = task_data.get('uuid', None)
+            
+            if status == 'completed':
+                if output_url:
+                    return output_url, "image edit completed", task_uuid
+                else:
+                    return None, "Task completed but no result image returned", task_uuid
+            elif status == 'error' or status == 'failed':
+                return None, f"task processing failed: {task_data}", task_uuid
+            elif status in ['queued', 'processing', 'running', 'created', 'working']:
+                if progress_callback:
+                    progress_callback(f"task processing... (status: {status})")
+                time.sleep(1)
+            else:
+                if progress_callback:
+                    progress_callback(f"unknown status: {status}")
+                time.sleep(1)
+        
+        return None, "task processing timeout", task_uuid
+        
+    except Exception as e:
+        return None, f"error occurred during processing: {str(e)}", None
+
+
+def process_local_image_edit(base_image, layers, prompt, progress_callback=None):
+    """
+    处理局部图片编辑的完整流程
+    
+    Args:
+        base_image (PIL.Image): 原始图片
+        layers (list): ImageEditor的层数据
+        prompt (str): 编辑指令
+        progress_callback: 进度回调函数
+    """
+    try:
+        # Generate client IP and timestamp
+        client_ip = "127.0.0.1"  # Default IP
+        time_id = int(time.time())
+        
+        if progress_callback:
+            progress_callback("creating mask image...")
+        
+        # 从layers创建mask图片
+        mask_image = create_mask_from_layers(base_image, layers)
+        
+        # 检查mask是否有内容
+        mask_array = np.array(mask_image)
+        if np.max(mask_array) == 0:
+            return None, "please draw mask", None
+        
+        print(f"📝 创建mask图片成功，绘制区域像素数: {np.sum(mask_array > 0)}")
+        
+        if progress_callback:
+            progress_callback("uploading original image...")
+        
+        # 直接从内存上传原始图片
+        uploaded_url = upload_user_img_r2(client_ip, time_id, base_image)
+        if not uploaded_url:
+            return None, "original image upload failed", None
+        
+        # 从上传 URL 中提取实际的图片 URL
+        if "?" in uploaded_url:
+            uploaded_url = uploaded_url.split("?")[0]
+        
+        if progress_callback:
+            progress_callback("uploading mask image...")
+        
+        # 直接从内存上传mask图片
+        mask_url = upload_mask_image_r2(client_ip, time_id, mask_image)
+        if not mask_url:
+            return None, "mask image upload failed", None
+        
+        # 从上传 URL 中提取实际的图片 URL
+        if "?" in mask_url:
+            mask_url = mask_url.split("?")[0]
+        
+        print(f"📤 图片上传成功:")
+        print(f"   原始图片: {uploaded_url}")
+        print(f"   Mask图片: {mask_url}")
+        
+        if progress_callback:
+            progress_callback("submitting local edit task...")
+        
+        # 提交局部图片编辑任务 (task_type=81)
+        task_id, error = submit_image_edit_task(uploaded_url, prompt, task_type="81", mask_image_url=mask_url)
+        if error:
+            return None, error, None
+        
+        if progress_callback:
+            progress_callback(f"task submitted, ID: {task_id}, processing...")
+        
+        print(f"🚀 局部编辑任务已提交，任务ID: {task_id}")
+        
+        # Wait for task completion
+        max_attempts = 60  # Wait up to 10 minutes
+        task_uuid = None
+        for attempt in range(max_attempts):
+            status, output_url, task_data = check_task_status(task_id)
+            
+            # Extract task_uuid from task_data
+            if task_data and isinstance(task_data, dict):
+                task_uuid = task_data.get('uuid', None)
+            
+            if status == 'completed':
+                if output_url:
+                    print(f"✅ 局部编辑任务完成，结果: {output_url}")
+                    return output_url, "local image edit completed", task_uuid
+                else:
+                    return None, "task completed but no result image returned", task_uuid
+            elif status == 'error' or status == 'failed':
+                return None, f"task processing failed: {task_data}", task_uuid
+            elif status in ['queued', 'processing', 'running', 'created', 'working']:
+                if progress_callback:
+                    progress_callback(f"processing... (status: {status})")
+                time.sleep(1)  # Wait 1 second before retry
+            else:
+                if progress_callback:
+                    progress_callback(f"unknown status: {status}")
+                time.sleep(1)
+        
+        return None, "task processing timeout", task_uuid
+        
+    except Exception as e:
+        print(f"❌ 局部编辑处理异常: {str(e)}")
+        return None, f"error occurred during processing: {str(e)}", None
+
+
+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