Upload app.py with huggingface_hub

app.py

@@ -12,6 +12,11 @@ def resize_image(image, target_size_kb=None, quality=None):
 
     img_byte_arr = io.BytesIO()
 
+    original_size_kb = len(image.getvalue()) / 1024  # Calculate the original size in KB
+    if target_size_kb and target_size_kb > original_size_kb:
+        st.error("Target size must be less than or equal to the original image size.")
+        return None
+
     if target_size_kb is not None:
         # Resize based on target size
         current_quality = 95  # Start with a high quality
@@ -30,38 +35,44 @@ def resize_image(image, target_size_kb=None, quality=None):
         img.save(img_byte_arr, format='JPEG', quality=quality)
     else:
         st.warning("No resize option selected. Displaying original image.")
-        img.save(img_byte_arr, format='JPEG') #save the original image in byte array
-    
-    img = Image.open(img_byte_arr) #reopen the image from the byte array
+        img.save(img_byte_arr, format='JPEG')  # Save the original image in byte array
+
+    resized_img = Image.open(img_byte_arr)  # Reopen the image from the byte array
 
-    return img
+    return resized_img
 
 st.title("Image Resizer")
-st.write("Resize and compress your images!")
+st.write("Resize and compress your images with ease!")
 
 uploaded_file = st.file_uploader("Choose an Image:", type=["jpg", "jpeg", "png"])
 
 if uploaded_file is not None:
+    original_size_kb = len(uploaded_file.getvalue()) / 1024  # Get original size in KB
+    st.info(f"Original Image Size: {original_size_kb:.2f} KB")
+
     resize_option = st.selectbox("Resize Option:", ("Select an option", "Percentage Quality", "Desired Output Size"))
 
+    resized_image = None  # Initialize resized_image as None
+
     if resize_option == "Percentage Quality":
         quality = st.slider("Quality (%)", min_value=1, max_value=100, value=85)
         resized_image = resize_image(uploaded_file, quality=quality)
     elif resize_option == "Desired Output Size":
         size_unit = st.selectbox("Size Unit", ("KB", "MB"))
-        target_size = st.number_input("Target Size", min_value=1)
+        target_size = st.number_input("Target Size", min_value=1, value=int(original_size_kb))
+
         if size_unit == "MB":
             target_size_kb = target_size * 1024
         else:
             target_size_kb = target_size
-        resized_image = resize_image(uploaded_file, target_size_kb=target_size_kb)
+
+        if target_size_kb > original_size_kb:
+            st.error("Target size must be less than or equal to the original image size.")
+        else:
+            resized_image = resize_image(uploaded_file, target_size_kb=target_size_kb)
     elif resize_option == "Select an option":
         st.info("Please select a resize option to proceed.")
-        resized_image = Image.open(uploaded_file)
-    else:
-        st.error("Invalid resize option selected.")
-        resized_image = None
-    
+
     if resized_image:
         st.image(resized_image, caption="Resized Image")
         with io.BytesIO() as buffer:
@@ -71,4 +82,4 @@ if uploaded_file is not None:
                 data=buffer.getvalue(),
                 file_name="resized_image.jpg",
                 mime="image/jpeg",
-            )
+            )