init

app.py

@@ -31,9 +31,9 @@ def t(key, lang="en"):
 # Configuration parameters
 
 TIP_TRY_N = 4      # Show like button tip after 12 tries
-FREE_TRY_N = 8        # Free phase: first 15 tries without restrictions
-SLOW_TRY_N = 15       # Slow phase start: 25 tries
-SLOW2_TRY_N = 20        # Slow phase start: 32 tries
+FREE_TRY_N = 7        # Free phase: first 15 tries without restrictions
+SLOW_TRY_N = 12       # Slow phase start: 25 tries
+SLOW2_TRY_N = 18        # Slow phase start: 32 tries
 RATE_LIMIT_60 = 25     # Full restriction: blocked after 40 tries
 
 # Time window configuration (minutes)
@@ -527,18 +527,25 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
     current_phase = get_ip_phase(client_ip)
     current_count = get_ip_generation_count(client_ip)
     geo_info = IP_Country_Cache.get(client_ip, {"country": "Unknown", "region": "Unknown", "city": "Unknown"})
+    is_restricted = is_restricted_country_ip(client_ip)
     
-    print(f"📊 User phase info - IP: {client_ip}, Location: {geo_info['country']}/{geo_info['region']}/{geo_info['city']}, Phase: {current_phase}, Count: {current_count}")
+    print(f"📊 User phase info - IP: {client_ip}, Location: {geo_info['country']}/{geo_info['region']}/{geo_info['city']}, Phase: {current_phase}, Count: {current_count}, Restricted: {is_restricted}")
     
-    # Check if user reached the like button tip threshold  
-    show_like_tip = (current_count >= TIP_TRY_N)
+    # Check if user reached the like button tip threshold
+    # For restricted countries, show like tip from the first attempt
+    show_like_tip = (current_count >= 1) if is_restricted else (current_count >= TIP_TRY_N)
     
     # Check if completely blocked
     if current_phase == 'blocked':
-        # Generate blocked limit button
+        # Generate blocked limit button with different URL for restricted countries
+        if is_restricted:
+            blocked_url = 'https://raphael.app/ai-image-editor'
+        else:
+            blocked_url = 'https://omnicreator.net/#generator'
+            
         blocked_button_html = f"""
         <div style='display: flex; justify-content: center; gap: 15px; margin: 10px 0 5px 0; padding: 0px;'>
-            <a href='https://omnicreator.net/#generator' target='_blank' style='
+            <a href='{blocked_url}' target='_blank' style='
                 display: inline-flex; 
                 align-items: center;
                 justify-content: center;
@@ -568,10 +575,15 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
         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
-            # Generate rate limit button
+            # Generate rate limit button with different URL for restricted countries
+            if is_restricted:
+                rate_limit_url = 'https://raphael.app/ai-image-editor'
+            else:
+                rate_limit_url = 'https://omnicreator.net/#generator'
+                
             rate_limit_button_html = f"""
             <div style='display: flex; justify-content: center; gap: 15px; margin: 10px 0 5px 0; padding: 0px;'>
-                <a href='https://omnicreator.net/#generator' target='_blank' style='
+                <a href='{rate_limit_url}' target='_blank' style='
                     display: inline-flex; 
                     align-items: center;
                     justify-content: center;
@@ -688,10 +700,15 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
                     final_result = result_url
                     final_message = t("warning_content_review", lang)
                 
-                # Generate NSFW button for blurred content
+                # Generate NSFW button for blurred content with different URL for restricted countries
+                if is_restricted:
+                    nsfw_url = 'https://raphael.app/ai-image-editor'
+                else:
+                    nsfw_url = 'https://omnicreator.net/#generator'
+                    
                 nsfw_action_buttons_html = f"""
                 <div style='display: flex; justify-content: center; gap: 15px; margin: 10px 0 5px 0; padding: 0px;'>
-                    <a href='https://omnicreator.net/#generator' target='_blank' style='
+                    <a href='{nsfw_url}' target='_blank' style='
                         display: inline-flex; 
                         align-items: center;
                         justify-content: center;
@@ -721,71 +738,96 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
             except Exception as e:
                 print(f"⚠️ Final progress update failed: {e}")
             
-            # Generate action buttons HTML like Trump AI Voice
+            # Generate action buttons HTML
             action_buttons_html = ""
-            if task_uuid:
-                # Create task detail URL for downloading HD image
-                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='https://omnicreator.net/#generator' 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;
-                    '>&#128640; Unlimited Generation</a>
-                    <a href='{task_detail_url}' 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;
-                    '>&#128190; Download HD Image</a>
-                </div>
-                """
             
-            # Add popup script if needed (using different approach)
-            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, #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>
-                """
+            # 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, #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 normal buttons
+                if task_uuid:
+                    # Create task detail URL for downloading HD image
+                    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='https://omnicreator.net/#generator' 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;
+                        '>&#128640; Unlimited Generation</a>
+                        <a href='{task_detail_url}' 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;
+                        '>&#128190; Download HD Image</a>
+                    </div>
+                    """
+                
+                # Add popup script if needed (using different approach)
+                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, #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>
+                    """
             
             return final_result, final_message, gr.update(value=action_buttons_html, visible=True)
         else: