update: app

app.py

@@ -3,12 +3,27 @@ from pathlib import Path
 from PIL import Image
 
 
-def convert_to_webp(input_image: str = None, quality=80):
-    file_path = Path("caches") / "{}.{}".format(Path(input_image).stem, "webp")
+def get_supported_formats():
+    supported_formats = Image.registered_extensions()
+    return supported_formats
+
+
+SUPPORTED_FORMATS = get_supported_formats()
+
+
+def convert_format(input_image: str = None, ext=".webp", quality=80):
+    file_path = Path("caches") / "{}{}".format(Path(input_image).stem, ext)
     file_path.parent.mkdir(parents=True, exist_ok=True)
     img = Image.open(input_image)
-    img = img.convert("RGBA")
-    img.save(file_path, "WEBP", quality=quality)
+    # img = img.convert("RGBA")
+    format = None
+    if ext in SUPPORTED_FORMATS:
+        format = SUPPORTED_FORMATS[ext]
+    if format is None:
+        gr.Error(
+            f"Unsupported image format. Supported formats: {', '.join(SUPPORTED_FORMATS)}"
+        )
+    img.save(file_path, format, quality=quality)
 
     # reopen and check
     img_reopen = Image.open(file_path)
@@ -16,11 +31,11 @@ def convert_to_webp(input_image: str = None, quality=80):
     return img_reopen, str(file_path)
 
 
-def process(input_list, quality=80):
+def process(input_list, ext=".webp", quality=80):
     out_files = []
     out_images = []
     for path in input_list:
-        img_reopen, file_path = convert_to_webp(path[0], quality)
+        img_reopen, file_path = convert_format(path[0], ext, quality)
         out_files.append(file_path)
         out_images.append(img_reopen)
     return out_files, out_images
@@ -34,6 +49,12 @@ def swap_to_gallery(images):
     )
 
 
+def download_files(files):
+    for file in files:
+        breakpoint()
+        gr.DownloadButton(visible=True, value=file)
+
+
 def run(server_name="127.0.0.1", server_port=7860):
     with gr.Blocks() as app:
         gr.Markdown(
@@ -53,38 +74,59 @@ def run(server_name="127.0.0.1", server_port=7860):
                 uploaded_files = gr.Gallery(
                     label="Your images", visible=False, columns=4, height=250
                 )
-                inputs = [
-                    uploaded_files,
-                    gr.Slider(
+                with gr.Row():
+                    quality_slider = gr.Slider(
                         minimum=1,
                         maximum=100,
                         value=80,
                         step=1,
                         label="Image Quality",
-                    ),
-                ]
-                btn = gr.Button("Run Convert", variant="primary")
+                    )
+                    extension_dropdown = gr.Dropdown(
+                        label="Output Format",
+                        choices=[
+                            ".webp",
+                            ".png",
+                            ".jpg",
+                            ".jpeg",
+                            ".gif",
+                            ".bmp",
+                            ".tiff",
+                            ".tif",
+                        ],
+                        value=".webp",
+                    )
+                proc_btn = gr.Button("Run Convert", variant="primary")
 
             with gr.Column():
-                outputs = [
-                    gr.File(label="Converted WebP"),
-                    gr.Gallery(
-                        label="Re-check converted images",
-                        show_label=False,
-                        elem_id="gallery",
-                        object_fit="contain",
-                        height="auto",
-                        columns=4,
-                        # height=125,
-                    ),
-                ]
+                output_file = gr.File(label="Converted WebP")
+                output_gallery = gr.Gallery(
+                    label="Re-check converted images",
+                    show_label=False,
+                    elem_id="gallery",
+                    object_fit="contain",
+                    height="auto",
+                    columns=4,
+                )
+
+        # collect inputs and outputs
+        inputs = [
+            uploaded_files,
+            extension_dropdown,
+            quality_slider,
+        ]
+        outputs = [
+            output_file,
+            output_gallery,
+        ]
+
+        # actions
         files.upload(
             fn=swap_to_gallery,
             inputs=files,
-            outputs=[uploaded_files, btn, files],
+            outputs=[uploaded_files, proc_btn, files],
         )
-        btn.click(process, inputs=inputs, outputs=outputs)
-    #
+        proc_btn.click(process, inputs=inputs, outputs=outputs)
     app.queue().launch(
         server_name=server_name, server_port=server_port, share=False
     )