| 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() |