Upload 3 files

app.py

@@ -0,0 +1,52 @@
+import gradio as gr
+import fitz
+import os
+import zipfile
+
+def pdf_to_images(pdf_file):
+
+    doc = fitz.open(pdf_file)
+    images = []
+
+    for page_id in range(doc.page_count):
+        page = doc[page_id]
+
+        pix = page.get_pixmap()
+        img_bytes = pix.tobytes("png")
+        images.append((img_bytes, f"{page_id+1}.png"))
+
+
+    doc.close()
+
+
+    temp_dir = "temp_images"
+    os.makedirs(temp_dir, exist_ok=True)
+
+
+    for img_bytes, img_name in images:
+        with open(os.path.join(temp_dir, img_name), "wb") as f:
+            f.write(img_bytes)
+
+
+    zip_path = "images.zip"
+    with zipfile.ZipFile(zip_path, "w") as zipf:
+        for img_name in os.listdir(temp_dir):
+            zipf.write(os.path.join(temp_dir, img_name), img_name)
+
+  
+    for img_name in os.listdir(temp_dir):
+        os.remove(os.path.join(temp_dir, img_name))
+    os.rmdir(temp_dir)
+
+    return zip_path
+
+
+iface = gr.Interface(
+    fn=pdf_to_images,
+    inputs=gr.File(label="Upload PDF File"),
+    outputs=gr.File(label="Download ZIP File"),
+    title="PDF to Images Converter",
+    description="Upload a PDF file and download a ZIP file containing all the pages as images."
+)
+iface.launch()
+

packages.txt

@@ -0,0 +1,2 @@
+poppler-utils
+libpoppler-cpp-dev

requirements.txt

@@ -0,0 +1,2 @@
+gradio==3.50
+pymupdf