fixing the url issue

generator_function/image_function.py

@@ -32,6 +32,7 @@ def _encode_image_to_base64(image_path):
 
 logger = logging.getLogger("imagegen_service")
 logging.basicConfig(level=logging.INFO)
+CANONICAL_STATIC_BASE = "https://adgen.static.lookfinity.com"
 
 REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
 MAX_WORKERS = min(32, (os.cpu_count() or 1) + 4)
@@ -39,6 +40,11 @@ REQUEST_TIMEOUT = 30
 RETRY_ATTEMPTS = 3
 
 
+def _public_base_url() -> str:
+    base = (os.getenv("NEW_BASE") or "").strip().rstrip("/")
+    return base or CANONICAL_STATIC_BASE
+
+
 def get_cost_per_image(
     model_key: str,
     width: int | None = None,
@@ -183,7 +189,7 @@ def _get_model_config_cached(model_key: str) -> Optional[Dict[str, Any]]:
 
 def _s3():
     if not hasattr(_thread_local, "s3"):
-        needed = ["R2_ENDPOINT","R2_ACCESS_KEY","R2_SECRET_KEY","R2_BUCKET_NAME","NEW_BASE"]
+        needed = ["R2_ENDPOINT", "R2_ACCESS_KEY", "R2_SECRET_KEY", "R2_BUCKET_NAME"]
         if any(not os.getenv(k) for k in needed):
             _thread_local.s3 = None
             return None
@@ -214,7 +220,7 @@ def _upload_to_r2(image_bytes: bytes) -> Optional[str]:
                 Body=image_bytes,
                 ContentType="image/png",
             )
-            return f"{os.getenv('NEW_BASE').rstrip('/')}/{key}"
+            return f"{_public_base_url()}/{key}"
         except Exception as e:
             if attempt == RETRY_ATTEMPTS - 1:
                 logger.error(f"R2 upload failed: {e}")

generator_function/video_hashid_function.py

@@ -7,6 +7,12 @@ import boto3
 
 load_dotenv()
 logger = logging.getLogger(__name__)
+CANONICAL_STATIC_BASE = "https://adgen.static.lookfinity.com"
+
+
+def _public_base_url() -> str:
+    base = (os.getenv("NEW_BASE") or "").strip().rstrip("/")
+    return base or CANONICAL_STATIC_BASE
 
 # ---------- HASH UTILS ----------
 def _sha256(path: Path, chunk: int = 1 << 20) -> str:
@@ -112,7 +118,7 @@ def upload_video_to_r2(video_bytes: bytes, file_name: str) -> str:
 
     s3.put_object(Bucket=bucket, Key=key, Body=video_bytes, ContentType="video/mp4")
 
-    return f"{os.getenv('NEW_BASE').rstrip('/')}/{key}"
+    return f"{_public_base_url()}/{key}"
 
 # ---------- PROCESS + UPLOAD ----------
 def process_video_with_hash_info(input_path: str, output_path: Optional[str] = None, variation_idx: int = 0) -> dict:

helpers_function/helpers.py

@@ -7,6 +7,7 @@ import base64
 import logging
 
 logger = logging.getLogger(__name__)
+CANONICAL_STATIC_BASE = "https://adgen.static.lookfinity.com"
 
 def get_video_thumbnail_base64(video_path: str, time_sec: int = 1) -> str:
     try:
@@ -25,6 +26,11 @@ def get_video_thumbnail_base64(video_path: str, time_sec: int = 1) -> str:
 
 log = logging.getLogger(__name__)
 
+
+def _public_base_url() -> str:
+    base = (os.getenv("NEW_BASE") or "").strip().rstrip("/")
+    return base or CANONICAL_STATIC_BASE
+
 def upload_image_to_r2(image_bytes, folder_name="search_arb", app_type="bulk_generation"):
     try:
         s3 = boto3.client(
@@ -37,7 +43,7 @@ def upload_image_to_r2(image_bytes, folder_name="search_arb", app_type="bulk_gen
         filename = f"{uuid4().hex}.png"
         file_key = f"hug_face/{app_type.strip('/')}/{folder_name.strip('/')}/{filename}" if folder_name else f"hug_face/{app_type.strip('/')}/{filename}"
         s3.put_object(Bucket=os.getenv('R2_BUCKET_NAME'), Key=file_key, Body=image_bytes, ContentType="image/png")
-        return f"{os.getenv('NEW_BASE').rstrip('/')}/{file_key}"
+        return f"{_public_base_url()}/{file_key}"
     except Exception as e:
         log.error(e)
         return None