init

__lib__/app.py

@@ -507,6 +507,8 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
     """
     Interface function for processing image editing with phase-based limitations
     """
+    # 默认禁用“Use as Input”按钮，待成功生成后再开启
+    use_as_input_state = gr.update(interactive=False)
     try:
         # Extract user IP
         client_ip = request.client.host
@@ -518,17 +520,17 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
         IP_Dict[client_ip] += 1
 
         if input_image is None:
-            return None, t("error_upload_first", lang), gr.update(visible=False)
+            return None, t("error_upload_first", lang), gr.update(visible=False), use_as_input_state
         
         if not prompt or prompt.strip() == "":
-            return None, t("error_enter_prompt", lang), gr.update(visible=False)
+            return None, t("error_enter_prompt", lang), gr.update(visible=False), use_as_input_state
         
         # Check if prompt length is greater than 3 characters
         if len(prompt.strip()) <= 3:
-            return None, t("error_prompt_too_short", lang), gr.update(visible=False)
+            return None, t("error_prompt_too_short", lang), gr.update(visible=False), use_as_input_state
     except Exception as e:
         print(f"⚠️ Unexpected error: {e}", flush=True)
-        return None, t("error_processing_failed", lang), gr.update(visible=False)
+        return None, t("error_processing_failed", lang), gr.update(visible=False), use_as_input_state
     
     # Concurrency guard: block if there is an active task within last 3 minutes
     try:
@@ -537,7 +539,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
         if active_info:
             start_ts = active_info.get("start", 0)
             if now_ts - start_ts <= 180:
-                return None, "You already have a task in progress. Please wait for it to finish before submitting a new one.", gr.update(visible=False, value=None)
+                return None, "You already have a task in progress. Please wait for it to finish before submitting a new one.", gr.update(visible=False, value=None), use_as_input_state
             else:
                 # Cleanup stale record
                 Active_Tasks.pop(client_ip, None)
@@ -589,7 +591,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
         # Use same message for all users to avoid discrimination perception
         blocked_message = t("error_free_limit_reached", lang)
         
-        return None, blocked_message, gr.update(value=blocked_button_html, visible=True)
+        return None, blocked_message, gr.update(value=blocked_button_html, visible=True), use_as_input_state
     
     # Check rate limit (applies to rate_limit phases)
     if current_phase in ['rate_limit_1', 'rate_limit_2', 'rate_limit_3']:
@@ -1105,7 +1107,7 @@ def create_app():
                 edit_button.click(
                     fn=edit_image_interface,
                     inputs=[input_image, prompt_input, lang_state],
-                    outputs=[output_image, status_output, action_buttons],
+                    outputs=[output_image, status_output, action_buttons, use_as_input_btn],
                     show_progress=True,
                     concurrency_limit=20
                 )