fix file to image implementation
Browse files
app.py
CHANGED
@@ -1,6 +1,10 @@
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
from pdfitdown.pdfconversion import Converter
|
|
|
|
|
|
|
|
|
4 |
|
5 |
converter = Converter()
|
6 |
|
@@ -19,7 +23,7 @@ def convert_file_to_pdf(filename:str) -> str:
|
|
19 |
converter.convert(filename.name, output_path)
|
20 |
return output_path
|
21 |
|
22 |
-
def convert_file_to_img(image_file) ->
|
23 |
"""
|
24 |
Convert an image file to PDF format.
|
25 |
|
@@ -32,9 +36,15 @@ def convert_file_to_img(image_file) -> str:
|
|
32 |
str: The file path of the generated PDF file. The output filename will be
|
33 |
the same as the input filename but with a .pdf extension.
|
34 |
"""
|
35 |
-
|
|
|
36 |
converter.convert(image_file.name, output_path)
|
37 |
-
|
|
|
|
|
|
|
|
|
|
|
38 |
|
39 |
|
40 |
|
@@ -50,9 +60,9 @@ file_to_pdf = gr.Interface(
|
|
50 |
image_to_pdf = gr.Interface(
|
51 |
fn=convert_file_to_img,
|
52 |
inputs=gr.File(label="Upload Image", file_types=["image"]),
|
53 |
-
outputs=gr.
|
54 |
-
title="
|
55 |
-
description="Convert your images to
|
56 |
)
|
57 |
|
58 |
# Create tabbed interface
|
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
from pdfitdown.pdfconversion import Converter
|
4 |
+
import fitz
|
5 |
+
from typing import List
|
6 |
+
from PIL import Image
|
7 |
+
from loadimg import load_img
|
8 |
|
9 |
converter = Converter()
|
10 |
|
|
|
23 |
converter.convert(filename.name, output_path)
|
24 |
return output_path
|
25 |
|
26 |
+
def convert_file_to_img(image_file:str) -> List[Image.Image] :
|
27 |
"""
|
28 |
Convert an image file to PDF format.
|
29 |
|
|
|
36 |
str: The file path of the generated PDF file. The output filename will be
|
37 |
the same as the input filename but with a .pdf extension.
|
38 |
"""
|
39 |
+
img_list = []
|
40 |
+
output_path = image_file.name.rsplit('.', 1)[0] + '.pdf'
|
41 |
converter.convert(image_file.name, output_path)
|
42 |
+
doc = fitz.open(output_path)
|
43 |
+
for page in doc:
|
44 |
+
img_list.append(load_img(page.get_pixmap()).convert("RGB"))
|
45 |
+
doc.close()
|
46 |
+
return img_list
|
47 |
+
|
48 |
|
49 |
|
50 |
|
|
|
60 |
image_to_pdf = gr.Interface(
|
61 |
fn=convert_file_to_img,
|
62 |
inputs=gr.File(label="Upload Image", file_types=["image"]),
|
63 |
+
outputs=gr.Gallery(label="Converted Images"),
|
64 |
+
title="File to Images Converter",
|
65 |
+
description="Convert your images to an image format"
|
66 |
)
|
67 |
|
68 |
# Create tabbed interface
|