Update app.py

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
 from PIL import Image
-from PIL import ImageFile
-ImageFile.LOAD_TRUNCATED_IMAGES = True
 import os
 import zipfile
 import pandas as pd
@@ -97,9 +95,12 @@ handle_process = lambda mode, sd, ups, pu, fmt, w, h: (
     process_single_url_image(pu, fmt, w, h) if pu.strip() else
     ([], None, "⚠️ No valid input provided", None)
 )
+from PIL import ImageFile
+ImageFile.LOAD_TRUNCATED_IMAGES = True
+
 def fetch_with_retries(url, headers, timeout=15, retries=2, backoff=1):
     """
-    Try up to `retries+1` times to GET `url` with `timeout` seconds.
+    Try up to retries+1 times to GET url with timeout seconds.
     Returns a requests.Response or None.
     """
     for attempt in range(retries + 1):
@@ -112,56 +113,36 @@ def fetch_with_retries(url, headers, timeout=15, retries=2, backoff=1):
         time.sleep(backoff * (attempt + 1))
     return None
 
-
-import cloudscraper
-
 def download_image(url, save_path):
+    # Skip cached
     if os.path.exists(save_path) and os.path.getsize(save_path) > 1000:
-        print(f"✅ Skipping cached: {save_path}")
         return save_path
 
-    if any(blocked in url for blocked in BLOCKED_SITES):
+    # Skip blocked
+    if any(b in url for b in BLOCKED_SITES):
         log_failure(url, "Skipped known slow site")
         return None
 
     referer = f"{urlparse(url).scheme}://{urlparse(url).netloc}/"
     headers = random_headers(referer)
 
-    # 1. Try direct
-    try:
-        print("🚀 Trying direct...")
-        resp = session.get(url, headers=headers, stream=True, timeout=8)
-        if resp.status_code == 200 and "image" in resp.headers.get("Content-Type", ""):
-            with open(save_path, 'wb') as f:
-                for chunk in resp.iter_content(8192):
-                    f.write(chunk)
-            print("✅ Direct worked")
-            return save_path
-    except Exception as e:
-        print(f"⚠️ Direct failed: {e}")
-
-    # 2. Try proxies in parallel
-    if PROXY_LIST:
-        headers = random_headers(referer)
-        with ThreadPoolExecutor(max_workers=5) as executor:
-            futures = {
-                executor.submit(try_download, url, headers, {"http": p, "https": p}): p
-                for p in random.sample(PROXY_LIST, min(5, len(PROXY_LIST)))
-            }
-            for future in as_completed(futures):
-                result = future.result()
-                if result:
-                    with open(save_path, 'wb') as f:
-                        for chunk in result.iter_content(8192):
-                            f.write(chunk)
-                    print("✅ Proxy worked:", futures[future])
-                    return save_path
-
-    # 3. cloudscraper fallback
+    # 1) Direct with retries
+    print(f"🚀 Fetching with retries: {url}")
+    resp = fetch_with_retries(url, headers, timeout=15, retries=2)
+    if resp:
+        with open(save_path, 'wb') as f:
+            for chunk in resp.iter_content(8192):
+                f.write(chunk)
+        print("✅ Direct fetch succeeded")
+        return save_path
+    else:
+        print("❌ Direct fetch failed after retries")
+
+    # 2) cloudscraper fallback
+    print("🟠 cloudscraper fallback...")
     try:
-        print("🟠 cloudscraper fallback...")
         scraper = cloudscraper.create_scraper(sess=session)
-        resp = scraper.get(url, headers=headers, stream=True, timeout=12)
+        resp = scraper.get(url, headers=headers, stream=True, timeout=15)
         if resp.status_code == 200 and "image" in resp.headers.get("Content-Type", ""):
             with open(save_path, 'wb') as f:
                 for chunk in resp.iter_content(8192):
@@ -171,7 +152,7 @@ def download_image(url, save_path):
     except Exception as e:
         print(f"❌ cloudscraper failed: {e}")
 
-    # 4. Final fallback: Selenium
+    # 3) Selenium fallback
     try:
         print("🧪 Headless browser fallback...")
         chrome_opts = Options()
@@ -181,7 +162,10 @@ def download_image(url, save_path):
         chrome_opts.add_argument("--disable-dev-shm-usage")
         chrome_opts.add_argument(f"user-agent={headers['User-Agent']}")
 
-        driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_opts)
+        driver = webdriver.Chrome(
+            service=Service(ChromeDriverManager().install()),
+            options=chrome_opts
+        )
         driver.get(url)
         time.sleep(4)
         final_url = driver.current_url
@@ -192,22 +176,20 @@ def download_image(url, save_path):
             for k, v in cookies.items():
                 session.cookies.set(k, v)
             r = session.get(final_url, headers=headers, stream=True, timeout=15)
-            if r.status_code == 200 and 'image' in r.headers.get("Content-Type", ""):
+            if r.status_code == 200 and "image" in r.headers.get("Content-Type", ""):
                 with open(save_path, 'wb') as f:
                     for chunk in r.iter_content(8192):
                         f.write(chunk)
                 print("✅ Selenium + cookies worked")
                 return save_path
-            else:
-                log_failure(url, f"Selenium bad response: {r.status_code}")
     except Exception as e:
         print(f"❌ Selenium failed: {e}")
-        log_failure(url, f"Selenium exception: {e}")
 
-    # Fail
+    # 4) Give up
     log_failure(url, "All methods failed")
     return None
 
+
 lamp_html = """
 <style>
   * {