init

__lib__/app.py

@@ -59,6 +59,8 @@ PRINT_STATS_INTERVAL = 10  # Print stats every N requests
 
 # Async IP query tracking
 IP_Query_Results = {}  # Track async query results
+# Active task tracking (within recent time window)
+Active_Tasks = {}  # {client_ip: {"start": timestamp}}
 
 # Restricted countries list (these countries have lower usage limits)
 RESTRICTED_COUNTRIES = ["印度", "巴基斯坦", "俄罗斯", "中国", "伊朗"]
@@ -525,8 +527,22 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
         if len(prompt.strip()) <= 3:
             return None, t("error_prompt_too_short", lang), gr.update(visible=False)
     except Exception as e:
-        print(f"⚠️ Request preprocessing error: {e}")
-        return None, t("error_request_processing", lang), gr.update(visible=False)
+        print(f"⚠️ Unexpected error: {e}", flush=True)
+        return None, t("error_processing_failed", lang), gr.update(visible=False)
+    
+    # Concurrency guard: block if there is an active task within last 3 minutes
+    try:
+        now_ts = time.time()
+        active_info = Active_Tasks.get(client_ip)
+        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)
+            else:
+                # Cleanup stale record
+                Active_Tasks.pop(client_ip, None)
+    except Exception as e:
+        print(f"⚠️ Concurrency guard check failed: {e}")
     
     # Get user current phase
     current_phase = get_ip_phase(client_ip)
@@ -657,6 +673,9 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
         # First high_priority_n tasks for each IP get priority=1
         task_priority = 1 if current_count < high_priority_n else 0
         
+        # Record active task start (for concurrency guard)
+        Active_Tasks[client_ip] = {"start": time.time()}
+        
         # 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)
@@ -801,71 +820,9 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
                             align-items: center;
                             justify-content: center;
                             padding: 12px 24px; 
-                            background: linear-gradient(135deg, #ff6b6b 0%, #feca57 100%);
-                            color: white; 
-                            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>
-                    """
-            else:
-                # For non-restricted countries, show action buttons
-                # Start with HF Space promo block so CTA is prominent above other actions
-                action_buttons_html = """
-                <div style='margin: 16px auto 0; max-width: 700px; background: linear-gradient(120deg, #f0f4ff 0%, #e5edff 50%, #f7fbff 100%); border: 1px solid #cbd5ff; border-radius: 14px; padding: 16px 18px; box-shadow: 0 12px 28px rgba(88, 101, 242, 0.18); text-align: center;'>
-                    <div style='font-size: 16px; font-weight: 900; 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: 8px; 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: 700;'>This is a free HF Space demo for Omni Image Editor 2.0.</div>
-                </div>
-                """
-
-                if task_uuid and lang not in ["zh", "hi", "ru"]:
-                    # Create i2v URL with input and output images and prompt
-                    from urllib.parse import quote
-                    encoded_prompt = quote(prompt.strip())
-                    i2v_url = f"https://omnicreator.net/image-to-video?input_image={input_image_url}&end_image={result_url}&prompt={encoded_prompt}"
-                    action_buttons_html += f"""
-                    <div style='display: flex; justify-content: center; gap: 12px; margin: 10px 0 5px 0; padding: 0px;'>
-                        <a href='{i2v_url}' target='_blank' style='
-                            display: inline-flex; 
-                            align-items: center;
-                            justify-content: center;
-                            padding: 16px 32px; 
-                            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                            background: linear-gradient(135deg, #7c3aed 0%, #6366f1 100%);
                             color: white; 
                             text-decoration: none; 
-                            border-radius: 12px; 
-                            font-weight: 700;
-                            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;
-                        '>&#127909; Convert to Video</a>
-                    </div>
-                    """
-                
-                # Add popup script if needed (using different approach)
-                if show_like_tip or lang in ["zh", "hi", "ru"]:
-                    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, #ff6b6b 0%, #feca57 100%);
-                            color: white; 
                             border-radius: 10px; 
                             font-weight: 600;
                             font-size: 14px;
@@ -885,6 +842,9 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
     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)
+    finally:
+        # Task finished (success or failure) — clear active marker to allow next submission immediately
+        Active_Tasks.pop(client_ip, None)
 
 # Create Gradio interface
 def create_app():