Update app.py

app.py

@@ -1,41 +1,25 @@
 import gradio as gr
 from PIL import Image
 import io
-import os
 
-def convert_image(input_image, output_format):
-    # Convert image using Pillow (CPU-efficient)
-    img = Image.open(input_image)
-    output_buffer = io.BytesIO()
+def convert_image(image, output_format):
+    img = Image.open(image)
+    img = img.convert("RGB")  # Ensuring compatibility for formats like TIFF
     
-    # Optimize for web formats
-    if output_format == "JPEG":
-        img.convert("RGB").save(output_buffer, format=output_format, quality=85)
-    else:
-        img.save(output_buffer, format=output_format)
+    output = io.BytesIO()
+    img.save(output, format=output_format.upper())
+    output.seek(0)
     
-    # Return converted image bytes
-    return output_buffer.getvalue()
+    return output
 
-# Gradio Interface
-interface = gr.Interface(
-    fn=convert_image,
-    inputs=[
-        gr.Image(type="filepath", label="Upload Image"),
-        gr.Dropdown(
-            ["PNG", "JPEG", "WEBP", "BMP"], 
-            value="PNG", 
-            label="Output Format"
-        )
-    ],
-    outputs=gr.File(label="Download Converted Image"),
-    examples=[
-        ["example.jpg", "PNG"],
-        ["example.png", "JPEG"]
-    ],
-    title="Lightweight Image Converter",
-    description="Convert images between formats with minimal CPU usage"
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# Simple Image Converter")
+    with gr.Row():
+        input_image = gr.File(label="Upload Image", type="file")
+        output_format = gr.Radio(["JPG", "PNG"], label="Convert To")
+    convert_button = gr.Button("Convert")
+    output_image = gr.File(label="Download Converted Image")
+    
+    convert_button.click(convert_image, inputs=[input_image, output_format], outputs=output_image)
 
-if __name__ == "__main__":
-    interface.launch()
+demo.launch()