init

__lib__/app.py

@@ -33,7 +33,7 @@ def t(key, lang="en"):
 
 # Configuration parameters
 # a = b
-TIP_TRY_N = 1      # Show like button tip after 12 tries
+TIP_TRY_N = 1      # Show like button tip after x tries
 FREE_TRY_N = 4        # Free phase: first 15 tries without restrictions
 SLOW_TRY_N = 6       # Slow phase start: 25 tries
 SLOW2_TRY_N = 10       # Slow phase start: 32 tries
@@ -623,7 +623,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
                 '>⏰ Skip Wait - Unlimited Generation</a>
             </div>
             """
-            return None, t("error_free_limit_wait", lang).format(wait_minutes_int=wait_minutes_int), gr.update(value=rate_limit_button_html, visible=True)
+            return None, t("error_free_limit_wait", lang).format(wait_minutes_int=wait_minutes_int), gr.update(value=rate_limit_button_html, visible=True), use_as_input_state
     
     # Handle NSFW detection based on phase
     is_nsfw_task = False  # Track if this task involves NSFW content
@@ -635,6 +635,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
             
             if nsfw_result.lower() == "nsfw":
                 is_nsfw_task = True
+                use_as_input_state = gr.update(interactive=False)
                 print(f"🔍 Input NSFW detected in {current_phase} phase: ❌❌❌ {nsfw_result} - IP: {client_ip} (will blur result)")
             else:
                 print(f"🔍 Input NSFW check passed: ✅✅✅ {nsfw_result} - IP: {client_ip}")
@@ -645,6 +646,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
     
     result_url = None
     status_message = ""
+    use_as_input_state = gr.update(interactive=True)
     
     def progress_callback(message):
         try:
@@ -716,7 +718,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
             # Use translated message or default
             limit_message = error_message if error_message else t("error_free_limit_reached", lang)
             
-            return None, limit_message, gr.update(value=hf_limit_button_html, visible=True)
+            return None, limit_message, gr.update(value=hf_limit_button_html, visible=True), use_as_input_state
         
         if result_url:
             print(f"✅ Processing completed successfully - IP: {client_ip}, result_url: {result_url}, task_uuid: {task_uuid}", flush=True)
@@ -761,12 +763,27 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
                     final_result = result_url
                     final_message = t("warning_content_review", lang)
                 
+                # Disable use-as-input when NSFW content is detected
+                use_as_input_state = gr.update(interactive=False)
+                
                 # Generate NSFW button for blurred content with different URL for restricted countries
                 if is_restricted or lang in ["hi", "ru", "zh"]:
                     nsfw_url = GOOGLE_GEMINI_URL
                 else:
                     nsfw_url = 'https://omnicreator.net/#generator'
                     
+                banner_html = """
+                <div style='margin: 14px auto 0; max-width: 640px; background: linear-gradient(120deg, #f0f4ff 0%, #e5edff 50%, #f7fbff 100%); border: 1px solid #cbd5ff; border-radius: 14px; padding: 14px 18px; box-shadow: 0 10px 25px rgba(88, 101, 242, 0.18); text-align: center;'>
+                    <div style='font-size: 15px; font-weight: 800; color: #1f2a44; display: flex; align-items: center; justify-content: center; gap: 8px;'>
+                        🚀 Omni Image Editor 2.0 is live!
+                    </div>
+                    <a href='https://huggingface.co/spaces/selfit-camera/Omni-Image-Editor' target='_blank' style='display: inline-flex; align-items: center; justify-content: center; margin-top: 6px; padding: 10px 18px; background: #5865f2; color: white; border-radius: 10px; font-weight: 800; text-decoration: none; box-shadow: 0 6px 18px rgba(88, 101, 242, 0.35);'>
+                        Try the Hugging Face Space demo (free)
+                    </a>
+                    <div style='font-size: 13px; color: #4a5568; margin-top: 6px; font-weight: 600;'>This is a free HF Space demo for Omni Image Editor 2.0.</div>
+                </div>
+                """
+                
                 nsfw_action_buttons_html = f"""
                 <div style='text-align: center; margin: 18px 0 10px 0;'>
                     <a href='{nsfw_url}' target='_blank' style='
@@ -786,17 +803,9 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
                         border: none;
                     '>🔥 Unlimited Creative Generation</a>
                 </div>
-                <div style='margin: 14px auto 0; max-width: 640px; background: linear-gradient(120deg, #f0f4ff 0%, #e5edff 50%, #f7fbff 100%); border: 1px solid #cbd5ff; border-radius: 14px; padding: 14px 18px; box-shadow: 0 10px 25px rgba(88, 101, 242, 0.18); text-align: center;'>
-                    <div style='font-size: 15px; font-weight: 800; color: #1f2a44; display: flex; align-items: center; justify-content: center; gap: 8px;'>
-                        🚀 Omni Image Editor 2.0 is live!
-                    </div>
-                    <a href='https://huggingface.co/spaces/selfit-camera/Omni-Image-Editor' target='_blank' style='display: inline-flex; align-items: center; justify-content: center; margin-top: 6px; padding: 10px 18px; background: #5865f2; color: white; border-radius: 10px; font-weight: 800; text-decoration: none; box-shadow: 0 6px 18px rgba(88, 101, 242, 0.35);'>
-                        Try the Hugging Face Space demo (free)
-                    </a>
-                    <div style='font-size: 13px; color: #4a5568; margin-top: 6px; font-weight: 600;'>This is a free HF Space demo for Omni Image Editor 2.0.</div>
-                </div>
+                {banner_html}
                 """
-                return final_result, final_message, gr.update(value=nsfw_action_buttons_html, visible=True)
+                return final_result, final_message, gr.update(value=nsfw_action_buttons_html, visible=True), use_as_input_state
             else:
                 final_result = result_url
                 final_message = t("status_completed_message", lang).format(message=message)
@@ -808,40 +817,54 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
                 print(f"⚠️ Final progress update failed: {e}")
             
             # Generate action buttons HTML
+            banner_html = """
+            <div style='margin: 14px auto 0; max-width: 640px; background: linear-gradient(120deg, #f0f4ff 0%, #e5edff 50%, #f7fbff 100%); border: 1px solid #cbd5ff; border-radius: 14px; padding: 14px 18px; box-shadow: 0 10px 25px rgba(88, 101, 242, 0.18); text-align: center;'>
+                <div style='font-size: 15px; font-weight: 800; color: #1f2a44; display: flex; align-items: center; justify-content: center; gap: 8px;'>
+                    🚀 Omni Image Editor 2.0 is live!
+                </div>
+                <a href='https://huggingface.co/spaces/selfit-camera/Omni-Image-Editor' target='_blank' style='display: inline-flex; align-items: center; justify-content: center; margin-top: 6px; padding: 10px 18px; background: #5865f2; color: white; border-radius: 10px; font-weight: 800; text-decoration: none; box-shadow: 0 6px 18px rgba(88, 101, 242, 0.35);'>
+                    Try the Hugging Face Space demo (free)
+                </a>
+                <div style='font-size: 13px; color: #4a5568; margin-top: 6px; font-weight: 600;'>This is a free HF Space demo for Omni Image Editor 2.0.</div>
+            </div>
+            """
+            
             action_buttons_html = ""
             
-            # For restricted countries, only show like tip (no action buttons)
-            if is_restricted:
-                if show_like_tip:
-                    action_buttons_html = """
-                    <div style='display: flex; justify-content: center; margin: 15px 0 5px 0; padding: 0px;'>
-                        <div style='
-                            display: inline-flex; 
-                            align-items: center;
-                            justify-content: center;
-                            padding: 12px 24px; 
-                            background: linear-gradient(135deg, #7c3aed 0%, #6366f1 100%);
-                            color: white; 
-                            text-decoration: none; 
-                            border-radius: 10px; 
-                            font-weight: 600;
-                            font-size: 14px;
-                            text-align: center;
-                            max-width: 400px;
-                            box-shadow: 0 3px 12px rgba(255, 107, 107, 0.3);
-                            border: none;
-                        '>👉 Click the ❤️ Like button to unlock more free trial attempts!</div>
-                    </div>
-                    """
+            # 根据 TIP_TRY_N（受限地区从第一次起就触发）展示点赞提示
+            if show_like_tip:
+                action_buttons_html = """
+                <div style='display: flex; justify-content: center; margin: 15px 0 5px 0; padding: 0px;'>
+                    <div style='
+                        display: inline-flex; 
+                        align-items: center;
+                        justify-content: center;
+                        padding: 12px 24px; 
+                        background: linear-gradient(135deg, #7c3aed 0%, #6366f1 100%);
+                        color: white; 
+                        text-decoration: none; 
+                        border-radius: 10px; 
+                        font-weight: 600;
+                        font-size: 14px;
+                        text-align: center;
+                        max-width: 400px;
+                        box-shadow: 0 3px 12px rgba(255, 107, 107, 0.3);
+                        border: none;
+                    '>👉 Click the ❤️ Like button to unlock more free trial attempts!</div>
+                </div>
+                """
+            
+            # Always show the Omni Image Editor 2.0 banner under the result image
+            action_buttons_html = f"{action_buttons_html}{banner_html}"
             
-            return final_result, final_message, gr.update(value=action_buttons_html, visible=True)
+            return final_result, final_message, gr.update(value=action_buttons_html, visible=True), use_as_input_state
         else:
             print(f"❌ Processing failed - IP: {client_ip}, error: {message}", flush=True)
-            return None, t("error_processing_failed", lang).format(message=message), gr.update(visible=False)
+            return None, t("error_processing_failed", lang).format(message=message), gr.update(visible=False), use_as_input_state
             
     except Exception as e:
         print(f"❌ Processing exception - IP: {client_ip}, error: {str(e)}")
-        return None, t("error_processing_exception", lang).format(error=str(e)), gr.update(visible=False)
+        return None, t("error_processing_exception", lang).format(error=str(e)), gr.update(visible=False), use_as_input_state
     finally:
         # Task finished (success or failure) — clear active marker to allow next submission immediately
         Active_Tasks.pop(client_ip, None)