Update app.py

app.py

@@ -299,14 +299,25 @@ with tab_download:
 
     
     mode = st.selectbox("Snapshot Mode", ["Single Page (High-Res)", "Two Pages (High-Res)", "All Pages (High-Res)"], key="download_mode")  # 🎛️ Selectbox: Choose your snapshot resolution!
-    if st.button("Snapshot Selected 📸"):  # 📸 Button: Time to snap some PDF snapshots!
-        selected_pdfs = [path for path in get_gallery_files() if path.endswith('.pdf') and st.session_state['asset_checkboxes'].get(path, False)]  # 📄 Filter: Pick only the PDFs marked for snapshotting!
+    if st.button("Snapshot Selected 📸"):
+        selected_pdfs = [path for path in get_gallery_files() 
+                         if path.endswith('.pdf') and st.session_state['asset_checkboxes'].get(path, False)]
         if selected_pdfs:
             for pdf_path in selected_pdfs:
-                mode_key = {"Single Page (High-Res)": "single", "Two Pages (High-Res)": "twopage", "All Pages (High-Res)": "allpages"}[mode]; snapshots = asyncio.run(process_pdf_snapshot(pdf_path, mode_key))  # 🔄 Processing: Generate snapshots per selected mode!
-                for snapshot in snapshots: st.image(Image.open(snapshot), caption=snapshot, use_container_width=True); st.session_state['asset_checkboxes'][snapshot] = True  # 🖼️ Display: Show each snapshot and mark it for potential SFT!
-            update_gallery()  # 🔄 Refresh gallery: Let the new snapshots shine!
-        else: st.warning("No PDFs selected for snapshotting! Check some boxes in the sidebar.")  # ⚠️ Warning: No PDFs chosen—select some boxes to make magic happen!
+                if not os.path.exists(pdf_path):
+                    st.warning(f"File not found: {pdf_path}. Skipping.")
+                    continue
+                mode_key = {"Single Page (High-Res)": "single", 
+                            "Two Pages (High-Res)": "twopage", 
+                            "All Pages (High-Res)": "allpages"}[mode]
+                snapshots = asyncio.run(process_pdf_snapshot(pdf_path, mode_key))
+                for snapshot in snapshots:
+                    st.image(Image.open(snapshot), caption=snapshot, use_container_width=True)
+                    st.session_state['asset_checkboxes'][snapshot] = True
+            update_gallery()
+        else:
+            st.warning("No PDFs selected for snapshotting! Check some boxes in the sidebar.")
+
 
 # === Tab: Test OCR ===
 with tab_ocr: