init

__lib__/app.py

@@ -73,13 +73,13 @@ def set_active_task(client_ip, active=True):
     """
     if active:
         util.IP_Active_Tasks[client_ip] = time.time()
-        print(f"🔒 Task locked for IP: {client_ip}")
+        # print(f"🔒 Task locked for IP: {client_ip}")
     else:
         if client_ip in util.IP_Active_Tasks:
             del util.IP_Active_Tasks[client_ip]
         # 记录任务完成时间
         util.IP_Last_Task_Time[client_ip] = time.time()
-        print(f"🔓 Task unlocked for IP: {client_ip}")
+        # print(f"🔓 Task unlocked for IP: {client_ip}")
 
 
 
@@ -691,6 +691,9 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
         
         task_priority = 1 if current_count < rate_limit_config.HIGH_PRIORITY_COUNT else 0
         record_generation_attempt(client_ip, current_phase)
+        updated_count = get_ip_generation_count(client_ip)
+        
+        print(f"✅ [Single Edit] Processing started - IP: {client_ip}, Country: {country}, Phase: {current_phase}, Count: {updated_count}, Priority: {task_priority}, Prompt: {prompt.strip()[:50]}...", flush=True)
         
         input_image_url, result_url, message, task_uuid = process_image_edit(input_image, prompt.strip(), None, progress_callback, priority=task_priority, client_ip=client_ip)
         
@@ -715,13 +718,16 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
             result_html = create_result_image_html(result_url)
             action_html = create_action_buttons_html(task_uuid, input_image_url, result_url, prompt, lang, show_like_tip)
             set_active_task(client_ip, False)  # 解锁任务
+            print(f"✅ [Single Edit] Completed - IP: {client_ip}, TaskUUID: {task_uuid}, Result: {result_url[:80]}...", flush=True)
             return result_html, result_url, t("status_completed_message", lang).format(message=message), gr.update(value=action_html, visible=True)
         else:
             set_active_task(client_ip, False)  # 解锁任务
+            print(f"❌ [Single Edit] Failed - IP: {client_ip}, Error: {message}", flush=True)
             return "", None, t("error_processing_failed", lang).format(message=message), gr.update(visible=False)
     
     except Exception as e:
         set_active_task(client_ip, False)  # 解锁任务
+        print(f"❌ [Single Edit] Exception - IP: {client_ip}, Error: {str(e)}", flush=True)
         return "", None, t("error_processing_exception", lang).format(error=str(e)), gr.update(visible=False)
 
 
@@ -808,6 +814,9 @@ def multi_image_edit_interface(input_image1, input_image2, input_image3, prompt,
         set_active_task(client_ip, True)
         
         record_generation_attempt(client_ip, current_phase)
+        updated_count = get_ip_generation_count(client_ip)
+        
+        print(f"✅ [Multi Edit] Processing started - IP: {client_ip}, Country: {country}, Phase: {current_phase}, Count: {updated_count}, Priority: {task_priority}, Images: {len(images)}, Prompt: {prompt.strip()[:50]}...", flush=True)
         
         input_url, result_url, message, task_uuid = process_multi_image_edit(images, prompt.strip(), progress_callback, priority=task_priority, client_ip=client_ip, is_sr=is_sr)
         
@@ -839,13 +848,16 @@ def multi_image_edit_interface(input_image1, input_image2, input_image3, prompt,
             status_message = f"{mode_tip}\n{t('status_completed_message', lang).format(message=message)}"
             
             set_active_task(client_ip, False)  # 解锁任务
+            print(f"✅ [Multi Edit] Completed - IP: {client_ip}, TaskUUID: {task_uuid}, Result: {result_url[:80]}...", flush=True)
             return result_html, status_message, gr.update(value=action_html, visible=True)
         else:
             set_active_task(client_ip, False)  # 解锁任务
+            print(f"❌ [Multi Edit] Failed - IP: {client_ip}, Error: {message}", flush=True)
             return "", t("error_processing_failed", lang).format(message=message), gr.update(visible=False)
     
     except Exception as e:
         set_active_task(client_ip, False)  # 解锁任务
+        print(f"❌ [Multi Edit] Exception - IP: {client_ip}, Error: {str(e)}", flush=True)
         return "", t("error_processing_exception", lang).format(error=str(e)), gr.update(visible=False)
 
 
@@ -921,6 +933,9 @@ def text_to_image_interface(prompt, aspect_ratio, lang, request: gr.Request, pro
         set_active_task(client_ip, True)
         
         record_generation_attempt(client_ip, current_phase)
+        updated_count = get_ip_generation_count(client_ip)
+        
+        print(f"✅ [Text2Image] Processing started - IP: {client_ip}, Country: {country}, Phase: {current_phase}, Count: {updated_count}, Priority: {task_priority}, Ratio: {aspect_ratio}, Prompt: {prompt.strip()[:50]}...", flush=True)
         
         result_url, message, task_uuid = process_text_to_image(prompt.strip(), aspect_ratio, progress_callback, priority=task_priority, client_ip=client_ip, is_sr=is_sr, target_area=target_area)
         
@@ -986,13 +1001,16 @@ def text_to_image_interface(prompt, aspect_ratio, lang, request: gr.Request, pro
             status_message = f"{mode_tip}\n{t('status_completed_message', lang).format(message=message)}"
             
             set_active_task(client_ip, False)  # 解锁任务
+            print(f"✅ [Text2Image] Completed - IP: {client_ip}, TaskUUID: {task_uuid}, Result: {result_url[:80]}...", flush=True)
             return result_html, status_message, gr.update(value=action_html, visible=True)
         else:
             set_active_task(client_ip, False)  # 解锁任务
+            print(f"❌ [Text2Image] Failed - IP: {client_ip}, Error: {message}", flush=True)
             return "", t("error_processing_failed", lang).format(message=message), gr.update(visible=False)
     
     except Exception as e:
         set_active_task(client_ip, False)  # 解锁任务
+        print(f"❌ [Text2Image] Exception - IP: {client_ip}, Error: {str(e)}", flush=True)
         return "", t("error_processing_exception", lang).format(error=str(e)), gr.update(visible=False)
 
 
@@ -1051,6 +1069,9 @@ def image_upscale_interface(input_image, lang, request: gr.Request, progress=gr.
         
         task_priority = 1 if current_count < rate_limit_config.HIGH_PRIORITY_COUNT else 0
         record_generation_attempt(client_ip, current_phase)
+        updated_count = get_ip_generation_count(client_ip)
+        
+        print(f"✅ [Upscale] Processing started - IP: {client_ip}, Country: {country}, Phase: {current_phase}, Count: {updated_count}, Priority: {task_priority}", flush=True)
         
         input_url, result_url, message, task_uuid = process_image_upscale(input_image, progress_callback, priority=task_priority, client_ip=client_ip)
         
@@ -1099,13 +1120,16 @@ def image_upscale_interface(input_image, lang, request: gr.Request, progress=gr.
             if show_like_tip or lang in ["zh", "hi", "ru"]:
                 action_html += create_like_tip_html()
             set_active_task(client_ip, False)  # 解锁任务
+            print(f"✅ [Upscale] Completed - IP: {client_ip}, TaskUUID: {task_uuid}, Result: {result_url[:80]}...", flush=True)
             return result_html, t("status_completed_message", lang).format(message=message), gr.update(value=action_html, visible=bool(action_html))
         else:
             set_active_task(client_ip, False)  # 解锁任务
+            print(f"❌ [Upscale] Failed - IP: {client_ip}, Error: {message}", flush=True)
             return "", t("error_processing_failed", lang).format(message=message), gr.update(visible=False)
     
     except Exception as e:
         set_active_task(client_ip, False)  # 解锁任务
+        print(f"❌ [Upscale] Exception - IP: {client_ip}, Error: {str(e)}", flush=True)
         return "", t("error_processing_exception", lang).format(error=str(e)), gr.update(visible=False)
 
 
@@ -1186,6 +1210,10 @@ def face_swap_interface(target_editor, source_face_image, lang, request: gr.Requ
         set_active_task(client_ip, True)
         task_priority = 1 if current_count < rate_limit_config.HIGH_PRIORITY_COUNT else 0
         record_generation_attempt(client_ip, current_phase)
+        updated_count = get_ip_generation_count(client_ip)
+        
+        has_mask = "Yes" if mask_image is not None else "No"
+        print(f"✅ [FaceSwap] Processing started - IP: {client_ip}, Country: {country}, Phase: {current_phase}, Count: {updated_count}, Priority: {task_priority}, HasMask: {has_mask}", flush=True)
         
         input_url, result_url, message, task_uuid = process_face_swap(target_image, source_face_image, mask_image, progress_callback, priority=task_priority, client_ip=client_ip)
         
@@ -1234,13 +1262,16 @@ def face_swap_interface(target_editor, source_face_image, lang, request: gr.Requ
             if show_like_tip or lang in ["zh", "hi", "ru"]:
                 action_html += create_like_tip_html()
             set_active_task(client_ip, False)
+            print(f"✅ [FaceSwap] Completed - IP: {client_ip}, TaskUUID: {task_uuid}, Result: {result_url[:80]}...", flush=True)
             return result_html, t("status_completed_message", lang).format(message=message), gr.update(value=action_html, visible=bool(action_html))
         else:
             set_active_task(client_ip, False)
+            print(f"❌ [FaceSwap] Failed - IP: {client_ip}, Error: {message}", flush=True)
             return "", t("error_processing_failed", lang).format(message=message), gr.update(visible=False)
     
     except Exception as e:
         set_active_task(client_ip, False)
+        print(f"❌ [FaceSwap] Exception - IP: {client_ip}, Error: {str(e)}", flush=True)
         return "", t("error_processing_exception", lang).format(error=str(e)), gr.update(visible=False)
 
 
@@ -1297,6 +1328,10 @@ def watermark_removal_interface(input_image, force_removal, lang, request: gr.Re
         set_active_task(client_ip, True)
         task_priority = 1 if current_count < rate_limit_config.HIGH_PRIORITY_COUNT else 0
         record_generation_attempt(client_ip, current_phase)
+        updated_count = get_ip_generation_count(client_ip)
+        
+        force_mode = "Yes" if force_removal else "No"
+        print(f"✅ [Watermark] Processing started - IP: {client_ip}, Country: {country}, Phase: {current_phase}, Count: {updated_count}, Priority: {task_priority}, ForceMode: {force_mode}", flush=True)
         
         input_url, result_url, message, task_uuid = process_watermark_removal(input_image, progress_callback, priority=task_priority, client_ip=client_ip, force_removal=force_removal)
         
@@ -1340,13 +1375,16 @@ def watermark_removal_interface(input_image, force_removal, lang, request: gr.Re
             if show_like_tip or lang in ["zh", "hi", "ru"]:
                 action_html += create_like_tip_html()
             set_active_task(client_ip, False)
+            print(f"✅ [Watermark] Completed - IP: {client_ip}, TaskUUID: {task_uuid}, Result: {result_url[:80]}...", flush=True)
             return result_html, t("status_completed_message", lang).format(message=message), gr.update(value=action_html, visible=bool(action_html))
         else:
             set_active_task(client_ip, False)
+            print(f"❌ [Watermark] Failed - IP: {client_ip}, Error: {message}", flush=True)
             return "", t("error_processing_failed", lang).format(message=message), gr.update(visible=False)
     
     except Exception as e:
         set_active_task(client_ip, False)
+        print(f"❌ [Watermark] Exception - IP: {client_ip}, Error: {str(e)}", flush=True)
         return "", t("error_processing_exception", lang).format(error=str(e)), gr.update(visible=False)