File size: 2,212 Bytes
3aba635 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import cv2
import gradio
import numpy
import os
import PIL
import pillow_heif
pillow_heif.register_heif_opener()
class HEIC2Any():
def __init__(self):
self.title = gradio.Markdown("# HEIC to Any Converter", render = False)
self.app = gradio.Blocks()
self.input_file = gradio.File(label = "Select HEIC file", file_types = [".heic"], type = "filepath", render = False)
self.image_display = gradio.Image(show_download_button = False, render = False)
self.convert_button = gradio.Button("Convert", render = False)
self.download_button = gradio.DownloadButton("Download", interactive = False, render = False)
self.type_mode = gradio.Radio(["jpg", "png"], label = "Select output file type", render = False)
self.__render()
def __render(self):
with self.app:
self.title.render()
self.input_file.render()
self.type_mode.render()
self.convert_button.render()
self.image_display.render()
self.download_button.render()
self.__set_behavior()
def __set_behavior(self):
self.convert_button.click(self.convert, inputs = [self.input_file, self.type_mode], outputs = [self.image_display, self.download_button])
def convert(self, input_file, type_mode):
image = PIL.Image.open(input_file)
print(type_mode)
if image is None:
self.download_button.interactive = False
self.download_button.render()
return None
image = image.convert("RGB")
image_np = numpy.array(image)
image_cv = cv2.cvtColor(image_np, cv2.COLOR_RGB2BGR)
file_dir = os.path.dirname(input_file)
file_name = f"{os.path.basename(input_file).split('.')[0]}.{type_mode}"
cv2.imwrite(f"{os.path.join(file_dir, file_name)}", image_cv)
download_button = gradio.DownloadButton("Download", value = os.path.join(file_dir, file_name), interactive = True)
return [image_np, download_button]
def launch(self):
self.app.launch(width = 400)
app = HEIC2Any()
app.launch() |