Create app.py

app.py

@@ -0,0 +1,40 @@
+import fitz
+import io
+import base64
+from PIL import Image
+import gradio as gr
+import cv2
+import tempfile
+import os
+
+def pdf_to_img(pdf_path):
+    pdf_document = fitz.open(pdf_path)
+    counter = 1
+    img_list = []
+    for page_number in range(len(pdf_document)):
+        page = pdf_document[page_number]
+        image_list = page.get_images()
+        for image in image_list:
+            base_img = pdf_document.extract_image(image[0])
+            image_data = base_img["image"]
+            img = Image.open(io.BytesIO(image_data))
+            extention = base_img['ext']
+            img.save(open(f"image{counter}.{extention}","wb"))
+            img_list.append(f"image{counter}.{extention}")
+            counter += 1
+    return (img_list)
+
+title = "Extract Image and Text"
+
+
+with gr.Blocks(theme=gr.themes.Glass(primary_hue=gr.themes.colors.slate)) as demo:
+    gr.Markdown(f'<h1 style="text-align: center;">{title}</h1>')
+    with gr.Row():
+        with gr.Row():
+            with gr.Column():
+                file_input = gr.File(type="filepath", label="Upload .pdf file")
+                upload_button = gr.Button(value="Show Images")
+                img_gallery = gr.Gallery(label="Generated images", show_label=True, elem_id="gallery", object_fit="contain", height="auto",allow_preview=True)
+
+    upload_button.click(pdf_to_img, inputs=file_input, outputs=[img_gallery])
+demo.launch()