progress

app.py

@@ -31,7 +31,9 @@ def process_pdf(pdf_file, sample_size, temp_dir, progress=gr.Progress()):
         )
 
         images = []
-        for i, page_num in enumerate(selected_pages):
+        for page_num in progress.tqdm(
+            selected_pages, desc=f"Converting {os.path.basename(pdf_path)}", unit="page"
+        ):
             page = doc[page_num]
             pix = page.get_pixmap()
             image_path = os.path.join(
@@ -39,10 +41,6 @@ def process_pdf(pdf_file, sample_size, temp_dir, progress=gr.Progress()):
             )
             pix.save(image_path)
             images.append(image_path)
-            progress(
-                (i + 1) / len(selected_pages),
-                desc=f"Converting {os.path.basename(pdf_path)}",
-            )
 
         doc.close()
         return images, None
@@ -58,23 +56,15 @@ def pdf_to_images(pdf_files, sample_size, temp_dir, progress=gr.Progress()):
     all_images = []
     skipped_pdfs = []
 
-    with ThreadPoolExecutor(max_workers=os.cpu_count()) as executor:
-        future_to_pdf = {
-            executor.submit(
-                process_pdf, pdf_file, sample_size, temp_dir, progress
-            ): pdf_file
-            for pdf_file in pdf_files
-        }
-
-        for i, future in enumerate(as_completed(future_to_pdf)):
-            pdf_file = future_to_pdf[future]
-            images, error = future.result()
-            if error:
-                skipped_pdfs.append(pdf_file.name)
-                gr.Info(error)
-            else:
-                all_images.extend(images)
-            progress((i + 1) / len(pdf_files), desc="Converting PDFs")
+    for i, pdf_file in enumerate(
+        progress.tqdm(pdf_files, desc="Converting PDFs", unit="PDF")
+    ):
+        images, error = process_pdf(pdf_file, sample_size, temp_dir, progress)
+        if error:
+            skipped_pdfs.append(pdf_file.name)
+            gr.Info(error)
+        else:
+            all_images.extend(images)
 
     message = f"Saved {len(all_images)} images to temporary directory"
     if skipped_pdfs: