init

app.py

@@ -9,9 +9,9 @@ from nfsw import NSFWDetector
 
 # Configuration parameters
 FREE_TRY_N = 20        # Free phase: first 20 tries without restrictions
-SLOW_TRY_N = 30        # Slow phase start: 30 tries
+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
+RATE_LIMIT_60 = 60     # Full restriction: blocked after 60 tries
 
 # Time window configuration (minutes)
 PHASE_1_WINDOW = 5     # 20-30 tries: 3 minutes
@@ -192,17 +192,17 @@ def edit_image_interface(input_image, prompt, request: gr.Request, progress=gr.P
         IP_Dict[client_ip] += 1
 
         if input_image is None:
-            return None, "Please upload an image first"
+            return None, "Please upload an image first", gr.update(visible=False)
         
         if not prompt or prompt.strip() == "":
-            return None, "Please enter editing prompt"
+            return None, "Please enter editing prompt", gr.update(visible=False)
         
         # Check if prompt length is greater than 3 characters
         if len(prompt.strip()) <= 3:
-            return None, "❌ Editing prompt must be more than 3 characters"
+            return None, "❌ Editing prompt must be more than 3 characters", gr.update(visible=False)
     except Exception as e:
         print(f"⚠️ Request preprocessing error: {e}")
-        return None, "❌ Request processing error"
+        return None, "❌ Request processing error", gr.update(visible=False)
     
     # Get user current phase
     current_phase = get_ip_phase(client_ip)
@@ -212,14 +212,14 @@ def edit_image_interface(input_image, prompt, request: gr.Request, progress=gr.P
     
     # 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"
+        return None, f"❌ You have reached Hugging Face's free generation limit. Please visit https://omnicreator.net/#generator for unlimited generation", gr.update(visible=False)
     
     # 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"
+            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", gr.update(visible=False)
     
     # Handle NSFW detection based on phase
     is_nsfw_task = False  # Track if this task involves NSFW content
@@ -260,10 +260,10 @@ def edit_image_interface(input_image, prompt, request: gr.Request, progress=gr.P
         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)
+        result_url, message, task_uuid = 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)
+            print(f"✅ Processing 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':
@@ -314,14 +314,57 @@ def edit_image_interface(input_image, prompt, request: gr.Request, progress=gr.P
             except Exception as e:
                 print(f"⚠️ Final progress update failed: {e}")
             
-            return final_result, final_message
+            # 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/#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"❌ Processing failed - IP: {client_ip}, error: {message}", flush=True)
-            return None, "❌ " + message
+            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)}"
+        return None, f"❌ Error occurred during processing: {str(e)}", gr.update(visible=False)
 
 def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Progress()):
     """
@@ -338,27 +381,27 @@ def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Pr
         IP_Dict[client_ip] += 1
 
         if image_dict is None:
-            return None, "Please upload an image and draw the area to edit"
+            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"
+            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"
+            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"
+            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"
+            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"
+        return None, "❌ Request processing error", gr.update(visible=False)
     
     # Get user current phase
     current_phase = get_ip_phase(client_ip)
@@ -368,14 +411,14 @@ def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Pr
     
     # 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"
+        return None, f"❌ You have reached Hugging Face's free generation limit. Please visit https://omnicreator.net/#generator for unlimited generation", gr.update(visible=False)
     
     # 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"
+            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", gr.update(visible=False)
     
     # Handle NSFW detection based on phase
     is_nsfw_task = False  # Track if this task involves NSFW content
@@ -416,10 +459,10 @@ def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Pr
         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)
+        result_url, message, task_uuid = 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)
+            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':
@@ -470,14 +513,57 @@ def local_edit_interface(image_dict, prompt, request: gr.Request, progress=gr.Pr
             except Exception as e:
                 print(f"⚠️ Local edit final progress update failed: {e}")
             
-            return final_result, final_message
+            # 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/#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
+            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)}"
+        return None, f"❌ Error occurred during processing: {str(e)}", gr.update(visible=False)
 
 # Create Gradio interface
 def create_app():
@@ -517,16 +603,18 @@ def create_app():
         """
     ) as app:
         
-        gr.Markdown(
-            """
-            # 🎨 AI Image Editor
-            """,
-            elem_classes=["main-container"]
-        )
+        # Main title - styled like Trump AI Voice
+        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 Image Editor
+            </h1>
+        </div>
+        """, padding=False)
         
-        # Powered by line below title
+        # Powered by line below title - styled like Trump AI Voice
         gr.HTML("""
-        <div style="text-align: center; margin: -10px auto 30px auto;">
+        <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/#generator" target="_blank" style="color: #667eea; text-decoration: none;">omnicreator.net</a>
             </p>
@@ -582,6 +670,9 @@ def create_app():
                             max_lines=3,
                             interactive=False
                         )
+                        
+                        # Action buttons that will show after task completion
+                        action_buttons = gr.HTML(visible=False)
                 
                 # Example area
                 gr.Markdown("### 💡 Prompt Examples")
@@ -607,7 +698,7 @@ def create_app():
                 edit_button.click(
                     fn=edit_image_interface,
                     inputs=[input_image, prompt_input],
-                    outputs=[output_image, status_output],
+                    outputs=[output_image, status_output, action_buttons],
                     show_progress=True,
                     # Increase concurrency settings
                     concurrency_limit=10,  # Limit concurrent requests
@@ -675,6 +766,9 @@ def create_app():
                             max_lines=3,
                             interactive=False
                         )
+                        
+                        # Action buttons that will show after task completion
+                        local_action_buttons = gr.HTML(visible=False)
                 
                 # Local editing examples
                 gr.Markdown("### 💡 Local Editing Prompt Examples")
@@ -700,7 +794,7 @@ def create_app():
                 local_edit_button.click(
                     fn=local_edit_interface,
                     inputs=[local_input_image, local_prompt_input],
-                    outputs=[local_output_image, local_status_output],
+                    outputs=[local_output_image, local_status_output, local_action_buttons],
                     show_progress=True,
                     # Increase concurrency settings
                     concurrency_limit=8,  # Local editing is more complex, allow fewer concurrent requests

util.py

@@ -436,7 +436,7 @@ def process_image_edit(img_input, prompt, progress_callback=None):
         # 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"
+            return None, "image upload failed", None
         
         # Extract actual image URL from upload URL
         if "?" in uploaded_url:
@@ -448,23 +448,28 @@ def process_image_edit(img_input, prompt, progress_callback=None):
         # Submit image editing task
         task_id, error = submit_image_edit_task(uploaded_url, prompt)
         if error:
-            return None, 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"
+                    return output_url, "image edit completed", task_uuid
                 else:
-                    return None, "Task completed but no result image returned"
+                    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}"
+                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})")
@@ -474,10 +479,10 @@ def process_image_edit(img_input, prompt, progress_callback=None):
                     progress_callback(f"unknown status: {status}")
                 time.sleep(1)
         
-        return None, "task processing timeout"
+        return None, "task processing timeout", task_uuid
         
     except Exception as e:
-        return None, f"error occurred during processing: {str(e)}"
+        return None, f"error occurred during processing: {str(e)}", None
 
 
 def process_local_image_edit(base_image, layers, prompt, progress_callback=None):
@@ -504,7 +509,7 @@ def process_local_image_edit(base_image, layers, prompt, progress_callback=None)
         # 检查mask是否有内容
         mask_array = np.array(mask_image)
         if np.max(mask_array) == 0:
-            return None, "please draw mask"
+            return None, "please draw mask", None
         
         print(f"📝 创建mask图片成功，绘制区域像素数: {np.sum(mask_array > 0)}")
         
@@ -514,7 +519,7 @@ def process_local_image_edit(base_image, layers, prompt, progress_callback=None)
         # 直接从内存上传原始图片
         uploaded_url = upload_user_img_r2(client_ip, time_id, base_image)
         if not uploaded_url:
-            return None, "original image upload failed"
+            return None, "original image upload failed", None
         
         # 从上传 URL 中提取实际的图片 URL
         if "?" in uploaded_url:
@@ -526,7 +531,7 @@ def process_local_image_edit(base_image, layers, prompt, progress_callback=None)
         # 直接从内存上传mask图片
         mask_url = upload_mask_image_r2(client_ip, time_id, mask_image)
         if not mask_url:
-            return None, "mask image upload failed"
+            return None, "mask image upload failed", None
         
         # 从上传 URL 中提取实际的图片 URL
         if "?" in mask_url:
@@ -542,7 +547,7 @@ def process_local_image_edit(base_image, layers, prompt, progress_callback=None)
         # 提交局部图片编辑任务 (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
+            return None, error, None
         
         if progress_callback:
             progress_callback(f"task submitted, ID: {task_id}, processing...")
@@ -551,17 +556,22 @@ def process_local_image_edit(base_image, layers, prompt, progress_callback=None)
         
         # 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"
+                    return output_url, "local image edit completed", task_uuid
                 else:
-                    return None, "task completed but no result image returned"
+                    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}"
+                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})")
@@ -571,11 +581,11 @@ def process_local_image_edit(base_image, layers, prompt, progress_callback=None)
                     progress_callback(f"unknown status: {status}")
                 time.sleep(1)
         
-        return None, "task processing timeout"
+        return None, "task processing timeout", task_uuid
         
     except Exception as e:
         print(f"❌ 局部编辑处理异常: {str(e)}")
-        return None, f"error occurred during processing: {str(e)}"
+        return None, f"error occurred during processing: {str(e)}", None
 
 
 def download_and_check_result_nsfw(image_url, nsfw_detector=None):