add try except

app.py

@@ -19,37 +19,45 @@ def pdf_to_images(pdf_files, sample_size, temp_dir, progress=gr.Progress()):
         os.makedirs(temp_dir)
     progress(0, desc="Starting conversion")
     all_images = []
+    skipped_pdfs = []
     for pdf_file in progress.tqdm(pdf_files, desc="Converting PDFs"):
-        pdf_path = pdf_file.name
-        pdf = PdfReader(pdf_path)
-        total_pages = len(pdf.pages)
-
-        # Determine the number of pages to convert
-        pages_to_convert = (
-            total_pages if sample_size == 0 else min(sample_size, total_pages)
-        )
-
-        # Select random pages if sampling
-        if sample_size > 0 and sample_size < total_pages:
-            selected_pages = sorted(
-                random.sample(range(1, total_pages + 1), pages_to_convert)
+        try:
+            pdf_path = pdf_file.name
+            pdf = PdfReader(pdf_path)
+            total_pages = len(pdf.pages)
+
+            # Determine the number of pages to convert
+            pages_to_convert = (
+                total_pages if sample_size == 0 else min(sample_size, total_pages)
             )
-        else:
-            selected_pages = range(1, total_pages + 1)
 
-        # Convert selected PDF pages to images
-        for page_num in selected_pages:
-            images = convert_from_path(
-                pdf_path, first_page=page_num, last_page=page_num
-            )
-            for image in images:
-                image_path = os.path.join(
-                    temp_dir, f"{os.path.basename(pdf_path)}_page_{page_num}.jpg"
+            # Select random pages if sampling
+            if sample_size > 0 and sample_size < total_pages:
+                selected_pages = sorted(
+                    random.sample(range(1, total_pages + 1), pages_to_convert)
                 )
-                image.save(image_path, "JPEG")
-                all_images.append(image_path)
+            else:
+                selected_pages = range(1, total_pages + 1)
 
-    return all_images, f"Saved {len(all_images)} images to temporary directory"
+            # Convert selected PDF pages to images
+            for page_num in selected_pages:
+                images = convert_from_path(
+                    pdf_path, first_page=page_num, last_page=page_num
+                )
+                for image in images:
+                    image_path = os.path.join(
+                        temp_dir, f"{os.path.basename(pdf_path)}_page_{page_num}.jpg"
+                    )
+                    image.save(image_path, "JPEG")
+                    all_images.append(image_path)
+        except Exception as e:
+            skipped_pdfs.append(pdf_file.name)
+            gr.Info(f"Skipped PDF {pdf_file.name} due to error: {str(e)}")
+
+    message = f"Saved {len(all_images)} images to temporary directory"
+    if skipped_pdfs:
+        message += f"\nSkipped {len(skipped_pdfs)} PDFs due to errors: {', '.join(skipped_pdfs)}"
+    return all_images, message
 
 
 def get_size_category(num_images):