Update app.py

app.py

@@ -4,15 +4,6 @@ import io
 import tempfile
 
 def compress_image(input_image, output_format, quality=85):
-    """
-    Compress an image and return the compressed image as a file.
-    Parameters:
-    - input_image (PIL.Image.Image): The input image.
-    - output_format (str): The format of the output image ('JPEG', 'JPG', 'PNG', or 'WEBP').
-    - quality (int, optional): The quality of the output image (1-100). Default is 85.
-    Returns:
-    - str: The path to the temporary file containing the compressed image.
-    """
     img_byte_arr = io.BytesIO()
     if output_format.upper() in ['JPEG', 'JPG']:
         input_image.save(img_byte_arr, format='JPEG', quality=quality)
@@ -21,47 +12,70 @@ def compress_image(input_image, output_format, quality=85):
     elif output_format.upper() == 'WEBP':
         input_image.save(img_byte_arr, format='WEBP', quality=quality)
     
-    # Save the compressed image to a temporary file
-    img_byte_arr = img_byte_arr.getvalue()  # Convert to bytes
+    img_byte_arr = img_byte_arr.getvalue()
     with tempfile.NamedTemporaryFile(delete=False, suffix=f'.{output_format.lower()}') as temp_file:
         temp_file.write(img_byte_arr)
-        temp_file_path = temp_file.name  # Get the temporary file path
-
+        temp_file_path = temp_file.name
+    
     return temp_file_path
 
 def image_compression_interface(uploaded_image, output_format, quality):
     try:
-        # Open the image using PIL
         image = Image.open(uploaded_image)
-
-        # Convert image to RGB if it has an alpha channel and the output format is JPEG, JPG, or WEBP
         if image.mode in ("RGBA", "P") and output_format.upper() in ['JPEG', 'JPG', 'WEBP']:
             image = image.convert("RGB")
         
-        # Compress the image and save to a temporary file
         compressed_image_path = compress_image(image, output_format, quality)
-
-        # Return the path to the compressed image file
         return compressed_image_path
     
     except Exception as e:
         return str(e)
 
-# Create the Gradio interface
+def get_current_size(compressed_image_path):
+    try:
+        with open(compressed_image_path, 'rb') as f:
+            size = len(f.read())
+        return f"Current Size: {size} bytes"
+    except FileNotFoundError:
+        return "Image not found"
+
+def update_ui(uploaded_image, output_format, quality):
+    try:
+        image = Image.open(uploaded_image)
+        if image.mode in ("RGBA", "P") and output_format.upper() in ['JPEG', 'JPG', 'WEBP']:
+            image = image.convert("RGB")
+        
+        compressed_image_path = compress_image(image, output_format, quality)
+        current_size = get_current_size(compressed_image_path)
+        return [compressed_image_path, current_size]
+    
+    except Exception as e:
+        return [str(e), None]
+
 def create_gradio_interface():
-    # Input components for Gradio (Updated for Gradio v3+)
-    image_input = gr.File(label="Upload Image")  # Changed to gr.File to handle raw file upload
+    image_input = gr.File(label="Upload Image")
     output_format_input = gr.Dropdown(choices=["JPEG", "JPG", "PNG", "WEBP"], label="Select Output Format")
     quality_input = gr.Slider(minimum=1, maximum=100, step=1, value=85, label="Select Compression Quality")
+    size_display = gr.Textbox(label="Current Size")
+
+    def on_quality_change(quality):
+        try:
+            image = Image.open(uploaded_image.value)
+            if image.mode in ("RGBA", "P") and output_format_input.value.upper() in ['JPEG', 'JPG', 'WEBP']:
+                image = image.convert("RGB")
+            
+            compressed_image_path = compress_image(image, output_format_input.value, quality)
+            current_size = get_current_size(compressed_image_path)
+            return current_size
+        except Exception as e:
+            return str(e)
 
-    # Output components for Gradio (Updated for Gradio v3+)
-    image_output = gr.File(label="Download Compressed Image")  # Use gr.File for download
+    quality_input.change(on_quality_change, inputs=[quality_input], outputs=size_display)
 
-    # Set up the Gradio interface
     gr.Interface(
-        fn=image_compression_interface,
+        fn=update_ui,
         inputs=[image_input, output_format_input, quality_input],
-        outputs=[image_output],
+        outputs=[gr.File(label="Download Compressed Image"), size_display],
     ).launch()
 
 if __name__ == "__main__":