init

app.py

@@ -5,7 +5,7 @@ import shutil
 import tempfile
 import time
 import json
-from util import process_image_edit, download_and_check_result_nsfw
+from util import process_image_edit, download_and_check_result_nsfw, GoodWebsiteUrl
 from nfsw import NSFWDetector
 
 # i18n
@@ -539,7 +539,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
     if current_phase == 'blocked':
         # Generate blocked limit button with different URL for restricted countries
         if is_restricted:
-            blocked_url = 'https://raphael.app/ai-image-editor'
+            blocked_url = GoodWebsiteUrl
         else:
             blocked_url = 'https://omnicreator.net/#generator'
             
@@ -577,7 +577,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
             wait_minutes_int = int(wait_minutes) + 1
             # Generate rate limit button with different URL for restricted countries
             if is_restricted:
-                rate_limit_url = 'https://raphael.app/ai-image-editor'
+                rate_limit_url = GoodWebsiteUrl
             else:
                 rate_limit_url = 'https://omnicreator.net/#generator'
                 
@@ -702,7 +702,7 @@ def edit_image_interface(input_image, prompt, lang, request: gr.Request, progres
                 
                 # Generate NSFW button for blurred content with different URL for restricted countries
                 if is_restricted:
-                    nsfw_url = 'https://raphael.app/ai-image-editor'
+                    nsfw_url = GoodWebsiteUrl
                 else:
                     nsfw_url = 'https://omnicreator.net/#generator'
                     

util.py

@@ -31,6 +31,7 @@ LLMKEY = OneKey[3]
 R2_ACCESS_KEY = OneKey[4]
 R2_SECRET_KEY = OneKey[5]
 R2_ENDPOINT = OneKey[6]
+GoodWebsiteUrl = OneKey[7]
 
 
 # tmpFolder is no longer needed since we upload directly from memory