Update app.py

app.py

@@ -2,16 +2,22 @@ import gradio as gr
 from PIL import Image
 import io
 
-def convert_image(image, output_format):
-    img = Image.open(image)
-    img = img.convert("RGB")  # Ensuring compatibility for formats like TIFF
-    
-    output = io.BytesIO()
-    img.save(output, format=output_format.upper())
-    output.seek(0)
-    
-    return output
+# Function to convert image
+def convert_image(image_path, output_format):
+    try:
+        img = Image.open(image_path)
+        img = img.convert("RGB")  # Ensuring compatibility for formats like TIFF, HEIC, and JFIF
+        
+        output = io.BytesIO()
+        save_format = "JPEG" if output_format.upper() == "JPG" else "PNG"
+        img.save(output, format=save_format)
+        output.seek(0)
+        
+        return output
+    except Exception as e:
+        return str(e)
 
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Simple Image Converter")
     with gr.Row():
@@ -22,4 +28,4 @@ with gr.Blocks() as demo:
     
     convert_button.click(convert_image, inputs=[input_image, output_format], outputs=output_image)
 
-demo.launch()
+demo.launch()