added image conversion

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from video_watermark_remover import *
 from video_convertor import *
-
+from image_convertor import *
 
 css = """
     #remove_btn {
@@ -77,6 +77,26 @@ with demo:
         format_select.change(lambda x: gr.Radio.update(choices=video_format if x == "Video" else audio_format, interactive=True), inputs=[format_select], outputs=[format])
         format.change(lambda x: gr.Button.update(interactive=True), None, outputs=[convert_btn])
         convert_btn.click(convert_video, inputs=[input_video, format], outputs=[output, status])
+
+    with gr.Tab("Image Convertor"):
+        gr.Markdown("""
+        # <center>🖼️ Image Convertor</center>
+        """)
+        image_format = ['jpg', 'jpeg', 'png', 'bmp', 'tiff', 'gif', 'webp', 'ico', 'heic', 'heiv', 'heif']
+        with gr.Row():
+            with gr.Column():
+                input_image = gr.Video(label="Upload a Video")
+            with gr.Column():
+                with gr.Row():
+                    with gr.Row():
+                        image_format = gr.Radio(image_format, label="Select Format", interactive=False)
+                    with gr.Row():
+                        image_convert_btn = gr.Button("Convert Image", interactive=False, elem_id="convert_btn")
+        with gr.Row():
+            output_image = gr.File(label="Output File", interactive=False)
+        image_status = gr.Textbox(label="Status", interactive=False)
+        image_format.change(lambda x: gr.Button.update(interactive=True), None, outputs=[image_convert_btn])
+        image_convert_btn.click(convert_image, inputs=[input_image, image_format], outputs=[output_image, image_status])
     gr.Markdown("""## <center style="margin:20px;">Developed by Muhammad Ahmed<img src="https://avatars.githubusercontent.com/u/63394104?v=4" style="height:50px;width:50px;border-radius:50%;margin:5px;"></img></center>
     """)
 demo.launch(show_api=False)

image_convertor.py

@@ -0,0 +1,64 @@
+import os
+from PIL import Image
+import pyheif
+import io
+
+class ImageConverter:
+    def __init__(self):
+        self.supported_formats = {
+            'jpg': 'JPEG',
+            'jpeg': 'JPEG',
+            'png': 'PNG',
+            'bmp': 'BMP',
+            'tiff': 'TIFF',
+            'gif': 'GIF',
+            'webp': 'WEBP',
+            'ico': 'ICO',
+            'heic': 'HEIC',
+            'heiv': 'HEIC',
+            'heif': 'HEIC',
+        }
+
+    def open_heif_image(self, input_image):
+        heif_file = pyheif.read(input_image)
+        return Image.frombytes(
+            heif_file.mode, 
+            heif_file.size, 
+            heif_file.data,
+            "raw",
+            heif_file.mode,
+            heif_file.stride,
+        )
+
+    def convert_image(self, input_image, output_format, output_path=None):
+        try:
+            if not os.path.exists(input_image):
+                raise FileNotFoundError(f"The input image '{input_image}' does not exist.")
+
+            input_extension = input_image.split('.')[-1].lower()
+
+            if input_extension not in self.supported_formats:
+                raise ValueError(f"The input format '{input_extension}' is not supported.")
+
+            if output_format.lower() not in self.supported_formats:
+                raise ValueError(f"The output format '{output_format}' is not supported.")
+
+            if input_extension in ['heic', 'heiv', 'heif']:
+                image = self.open_heif_image(input_image)
+            else:
+                image = Image.open(input_image)
+
+            if output_path is None:
+                output_image = '.'.join(input_image.split('.')[:-1]) + f'.{output_format}'
+            else:
+                output_image = output_path
+
+            image.save(output_image, self.supported_formats[output_format.lower()])
+            print(f"Image converted and saved as {output_image}")
+            return output_image, "Image converted and saved as {output_image}"
+        except Exception as e:
+            None, print(f"Error: {e}")
+
+def convert_image(input_image, output_format):
+    converter = ImageConverter()
+    return converter.convert_image(input_image, output_format)

requirements.txt

@@ -2,4 +2,5 @@ gradio==3.22.1
 ffmpeg-python
 moviepy
 pydub
-opencv-python
+opencv-python
+pyheif