Update app.py

app.py

@@ -49,7 +49,7 @@ from urllib.parse import urlparse, quote
 
 def download_image_fast(url, out_path, fail_log, code, suf):
     def log(msg):
-        with open(fail_log, "a") as f:
+        thread_log_failure(f"{code}{suf}: {msg}")
             f.write(f"{code}{suf}: {msg}\n")
 
     parts     = urlparse(url)
@@ -289,6 +289,8 @@ def handle_batch(workbook_path, fmt, w, h):
     outdir = os.path.join(tmpdir, 'out')
     os.makedirs(outdir, exist_ok=True)
     fail_log = os.path.join(tmpdir, 'failed.txt')
+    fail_lock = threading.Lock()
+    failures = []
 
     try:
         jobs = read_links_from_workbook(workbook_path.name)
@@ -310,16 +312,15 @@ def handle_batch(workbook_path, fmt, w, h):
 
     # Download all images
     future_to_task = {}
-    with ThreadPoolExecutor(max_workers=min(32, CPU_CORES * 2)) as ex:
+     with ThreadPoolExecutor(max_workers=min(32, CPU_CORES * 2)) as ex:
+        futures = []
         for code, url, dl_path, suf in tasks:
-            fut = ex.submit(download_image_fast, url, dl_path, fail_log, code, suf)
-            future_to_task[fut] = (code, url, dl_path, suf)
-
-        for fut in as_completed(future_to_task):
+            fut = ex.submit(download_image_fast_mod, url, dl_path, code, suf)
+            futures.append((fut, code, suf))
+        for fut, code, suf in futures:
             success = fut.result()
-            code, url, dl_path, suf = future_to_task[fut]
             if not success:
-                log_failure(fail_log, f"{code}{suf}: download failed")
+                thread_log_failure(f"{code}{suf}: download failed")
 
     processed = []
     for code, url, dl_path, suf in tasks:
@@ -379,11 +380,12 @@ def handle_single_upload(uploaded_file, fmt, w, h):
     outp=os.path.join(tmpdir,f"single.{fmt.lower()}")
     if fmt.upper()=="JPEG":
         final_jpg = convert_to_jpeg(img, target=w)
-        final_jpg.save(out_p, "JPEG", quality=95, optimize=True, progressive=True)
+        final_jpg.save(outp, "JPEG", quality=95, optimize=True, progressive=True)  # fixed outp instead of out_p
     else:
         final.save(outp, fmt.upper())
     return final, outp, "✅ Single processed."
 
+
 def handle_single_url(url, fmt, w, h):
     if not url.strip():
         return None, None, "❌ No URL provided."
@@ -424,13 +426,15 @@ def handle_single_url(url, fmt, w, h):
 
     # 5) Save final image
     outp = os.path.join(tmpdir, f"single.{fmt.lower()}")
+        # ...
     try:
         if fmt.upper() == "JPEG":
             final_jpg = convert_to_jpeg(img, target=w)
-            final_jpg.save(out_p, "JPEG", quality=95, optimize=True, progressive=True)
+            final_jpg.save(outp, "JPEG", quality=95, optimize=True, progressive=True)  # fixed outp instead of out_p
         else:
             final.save(outp, fmt.upper())
     except Exception as e:
+        # ...
         reason = str(e)
         shutil.rmtree(tmpdir)
         return None, None, f"❌ Save failed: {reason}"
@@ -821,11 +825,12 @@ with gr.Blocks(css=css) as demo:
             return files, zip_path, msg
         else:
             if uploads:
-                _, fpath, msg = handle_single_upload(uploads, fmt, w, h)
+                files, fpath, msg = handle_single_upload(uploads, fmt, w, h)
             else:
-                _, fpath, msg = handle_single_url(url, fmt, w, h)
+                files, fpath, msg = handle_single_url(url, fmt, w, h)
             return ([fpath] if fpath else []), None, msg
 
+
     # ─── Wire up the Process button here ───────────────────
     process_btn_url.click(
         fn=dispatch,