Use WB scaled-down images

app.py

@@ -83,10 +83,10 @@ def get_additional_images(id_product_money: str, marketplace: str) -> List[str]:
             if result and result.more_images:
                 print(f"Wildberries raw more_images: {result.more_images}")
                 try:
-                    urls = json.loads(result.more_images)
-                    if isinstance(urls, list) and len(urls) > 0:
-                        # Split the URLs by semicolons
-                        return urls[0].split(';')
+                    data = json.loads(result.more_images)
+                    if isinstance(data, list):
+                        # Extract image URLs from the JSON structure
+                        return [item.get('image_url') for item in data if item.get('image_url')]
                     return []
                 except Exception as e:
                     print(f"Error parsing JSON: {str(e)}")
@@ -97,25 +97,33 @@ def get_additional_images(id_product_money: str, marketplace: str) -> List[str]:
 
 def try_scaled_image_url(client: httpx.Client, url: str, marketplace: str, max_retries: int = 3) -> str:
     """Try to get a scaled version of the image URL, fall back to original if not available."""
+    scaled_url = url
+    
     if marketplace == 'lamoda':
         scaled_url = url.replace('product', 'img600x866')
-        for attempt in range(max_retries):
-            try:
-                response = client.get(scaled_url, timeout=5.0)
-                if response.status_code == 200:
-                    print(f"Using scaled image: {scaled_url}")
-                    return scaled_url
-                else:
-                    print(f"Scaled image not available (status {response.status_code}), using original: {url}")
-                    return url
-            except httpx.TimeoutException:
-                print(f"Timeout checking scaled image (attempt {attempt + 1}/{max_retries})")
-                if attempt == max_retries - 1:
-                    print(f"Max retries reached, using original: {url}")
-                    return url
-            except Exception as e:
-                print(f"Error checking scaled image: {type(e).__name__}: {str(e)}")
+    elif marketplace == 'wildberries':
+        scaled_url = url.replace('/big/', '/c516x688/')
+    else:
+        return url  # No scaling for other marketplaces
+        
+    for attempt in range(max_retries):
+        try:
+            response = client.get(scaled_url, timeout=5.0)
+            if response.status_code == 200:
+                print(f"Using scaled image: {scaled_url}")
+                return scaled_url
+            else:
+                print(f"Scaled image not available (status {response.status_code}), using original: {url}")
+                return url
+        except httpx.TimeoutException:
+            print(f"Timeout checking scaled image (attempt {attempt + 1}/{max_retries})")
+            if attempt == max_retries - 1:
+                print(f"Max retries reached, using original: {url}")
                 return url
+        except Exception as e:
+            print(f"Error checking scaled image: {type(e).__name__}: {str(e)}")
+            return url
+    
     return url
 
 def download_and_encode_images(image_urls: List[str], marketplace: str) -> List[Dict]:

test.py

@@ -82,10 +82,10 @@ def get_additional_images(id_product_money: str, marketplace: str) -> List[str]:
             if result and result.more_images:
                 print(f"Wildberries raw more_images: {result.more_images}")
                 try:
-                    urls = json.loads(result.more_images)
-                    if isinstance(urls, list) and len(urls) > 0:
-                        # Split the URLs by semicolons
-                        return urls[0].split(';')
+                    data = json.loads(result.more_images)
+                    if isinstance(data, list):
+                        # Extract image URLs from the JSON structure
+                        return [item.get('image_url') for item in data if item.get('image_url')]
                     return []
                 except Exception as e:
                     print(f"Error parsing JSON: {str(e)}")
@@ -96,25 +96,33 @@ def get_additional_images(id_product_money: str, marketplace: str) -> List[str]:
 
 def try_scaled_image_url(client: httpx.Client, url: str, marketplace: str, max_retries: int = 3) -> str:
     """Try to get a scaled version of the image URL, fall back to original if not available."""
+    scaled_url = url
+    
     if marketplace == 'lamoda':
         scaled_url = url.replace('product', 'img600x866')
-        for attempt in range(max_retries):
-            try:
-                response = client.get(scaled_url, timeout=5.0)
-                if response.status_code == 200:
-                    print(f"Using scaled image: {scaled_url}")
-                    return scaled_url
-                else:
-                    print(f"Scaled image not available (status {response.status_code}), using original: {url}")
-                    return url
-            except httpx.TimeoutException:
-                print(f"Timeout checking scaled image (attempt {attempt + 1}/{max_retries})")
-                if attempt == max_retries - 1:
-                    print(f"Max retries reached, using original: {url}")
-                    return url
-            except Exception as e:
-                print(f"Error checking scaled image: {type(e).__name__}: {str(e)}")
+    elif marketplace == 'wildberries':
+        scaled_url = url.replace('/big/', '/c516x688/')
+    else:
+        return url  # No scaling for other marketplaces
+        
+    for attempt in range(max_retries):
+        try:
+            response = client.get(scaled_url, timeout=5.0)
+            if response.status_code == 200:
+                print(f"Using scaled image: {scaled_url}")
+                return scaled_url
+            else:
+                print(f"Scaled image not available (status {response.status_code}), using original: {url}")
+                return url
+        except httpx.TimeoutException:
+            print(f"Timeout checking scaled image (attempt {attempt + 1}/{max_retries})")
+            if attempt == max_retries - 1:
+                print(f"Max retries reached, using original: {url}")
                 return url
+        except Exception as e:
+            print(f"Error checking scaled image: {type(e).__name__}: {str(e)}")
+            return url
+    
     return url
 
 def download_and_encode_images(image_urls: List[str], marketplace: str) -> List[Dict]: