init

app.py

@@ -468,8 +468,8 @@ def edit_image_interface(input_image1, input_image2, input_image3, prompt, aspec
         
         print(f"✅ Multi-image processing started - IP: {client_ip}, phase: {current_phase}, total count: {updated_count}, images: {len(valid_images)}, size: {width}x{height}, prompt: {prompt.strip()}", flush=True)
         
-        # Determine priority (first few tasks for each IP get priority)
-        task_priority = 1 if current_count < 2 else 0
+        # Determine priority (only the first task for each IP gets priority)
+        task_priority = 1 if current_count == 0 else 0
         
         # Call multi-image editing processing function
         result_url, message, task_uuid = process_multi_image_edit(valid_images, prompt.strip(), width, height, progress_callback, priority=task_priority, client_ip=client_ip)
@@ -604,310 +604,7 @@ def edit_image_interface(input_image1, input_image2, input_image3, prompt, aspec
     finally:
         # Always clear active task status when done (success or failure)
         set_active_task(client_ip, active=False)
-
-def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Progress()):
-    """
-    Handle local editing requests (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 image_dict is None:
-            return None, "Please upload an image and draw the area to edit", gr.update(visible=False)
-        
-        # Check if background and layers exist
-        if "background" not in image_dict or "layers" not in image_dict:
-            return None, "Please draw the area to edit on the image", gr.update(visible=False)
-        
-        base_image = image_dict["background"]
-        layers = image_dict["layers"]
-        
-        if not layers:
-            return None, "Please draw the area to edit on the image", gr.update(visible=False)
-            
-        if not prompt or prompt.strip() == "":
-            return None, "Please enter editing prompt", gr.update(visible=False)
-        
-        # Check prompt length
-        if len(prompt.strip()) <= 3:
-            return None, "❌ Editing prompt must be more than 3 characters", gr.update(visible=False)
-    except Exception as e:
-        print(f"⚠️ Local edit request preprocessing error: {e}")
-        return None, "❌ Request processing error", gr.update(visible=False)
-    
-    # Check if user has an active task in progress
-    if has_active_task(client_ip):
-        concurrent_button_html = f"""
-        <div style='display: flex; justify-content: center; gap: 15px; margin: 10px 0 5px 0; padding: 0px;'>
-            <a href='https://omnicreator.net/multi-image-edit#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;
-            '>⏰ Process Multiple Tasks Simultaneously</a>
-        </div>
-        """
-        return None, "❌ You have a task in progress. Please wait for it to complete before submitting a new task. Or visit https://omnicreator.net/multi-image-edit#generator to process multiple tasks simultaneously.", gr.update(value=concurrent_button_html, visible=True)
-    
-    # Get user current phase
-    current_phase = get_ip_phase(client_ip)
-    current_count = get_ip_generation_count(client_ip)
-    geo_info = IP_Country_Cache.get(client_ip, {"country": "Unknown", "region": "Unknown", "city": "Unknown"})
-    is_restricted = is_restricted_country_ip(client_ip)
-    
-    print(f"📊 Local edit user phase info - IP: {client_ip}, Location: {geo_info['country']}/{geo_info['region']}/{geo_info['city']}, Phase: {current_phase}, Count: {current_count}, Restricted: {is_restricted}, Active task: {has_active_task(client_ip)}")
-    
-    # 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/multi-image-edit#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;
-            '>🚀 Unlimited Generation</a>
-        </div>
-        """
-        return None, f"❌ You have reached Hugging Face's free generation limit. Please visit https://omnicreator.net/multi-image-edit#generator for unlimited generation", gr.update(value=blocked_button_html, visible=True)
-    
-    # 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
-            # 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/multi-image-edit#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 Generation</a>
-            </div>
-            """
-            return None, f"❌ You have reached Hugging Face's free generation limit. Please visit https://omnicreator.net/multi-image-edit#generator for unlimited generation, or wait {wait_minutes_int} minutes before generating again", gr.update(value=rate_limit_button_html, visible=True)
-    
-    # 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":
-                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"🔍 Local edit input NSFW check passed: ✅✅✅ {nsfw_result} - IP: {client_ip}")
-                
-        except Exception as e:
-            print(f"⚠️ Local edit input NSFW detection failed: {e}")
-            # Allow continuation when detection fails
-
-    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"⚠️ Local edit 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)
-        
-        # Mark this IP as having an active task
-        set_active_task(client_ip, active=True)
         
-        print(f"✅ Local editing started - IP: {client_ip}, phase: {current_phase}, total count: {updated_count}, prompt: {prompt.strip()}", flush=True)
-        
-        # Determine priority (first few tasks for each IP get priority)
-        task_priority = 1 if current_count < 2 else 0
-        
-        # Call local image editing processing function
-        result_url, message, task_uuid = process_local_image_edit(base_image, layers, prompt.strip(), progress_callback, priority=task_priority, client_ip=client_ip)
-        
-        if result_url:
-            print(f"✅ Local editing completed successfully - IP: {client_ip}, result_url: {result_url}, task_uuid: {task_uuid}", 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/multi-image-edit#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/multi-image-edit#generator for better experience"
-                
-                # Generate NSFW button for blurred content
-                nsfw_action_buttons_html = f"""
-                <div style='display: flex; justify-content: center; gap: 15px; margin: 10px 0 5px 0; padding: 0px;'>
-                    <a href='https://omnicreator.net/multi-image-edit#generator' target='_blank' style='
-                        display: inline-flex; 
-                        align-items: center;
-                        justify-content: center;
-                        padding: 16px 32px; 
-                        background: linear-gradient(135deg, #ff6b6b 0%, #feca57 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(255, 107, 107, 0.4);
-                        transition: all 0.3s ease;
-                        border: none;
-                    '>🔥 Unlimited NSFW Generation</a>
-                </div>
-                """
-                return final_result, final_message, gr.update(value=nsfw_action_buttons_html, visible=True)
-            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}")
-            
-            # 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/multi-image-edit#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;
-                    '>🚀 Unlimited Generation</a>
-                </div>
-                """
-            
-            return final_result, final_message, gr.update(value=action_buttons_html, visible=True)
-        else:
-            print(f"❌ Local editing processing failed - IP: {client_ip}, error: {message}", flush=True)
-            return None, "❌ " + message, gr.update(visible=False)
-            
-    except Exception as e:
-        print(f"❌ Local editing exception - IP: {client_ip}, error: {str(e)}")
-        return None, f"❌ Error occurred during processing: {str(e)}", gr.update(visible=False)
-    finally:
-        # Always clear active task status when done (success or failure)
-        set_active_task(client_ip, active=False)
-
 # Create Gradio interface
 def create_app():
     with gr.Blocks(