Upload app.py with huggingface_hub

app.py

@@ -2,74 +2,73 @@ import streamlit as st
 from PIL import Image
 import io
 
-def resize_image(image, target_size_kb=None, quality=85):
-  """Resizes and compresses an image."""
-  img = Image.open(image)
-  img_byte_arr = io.BytesIO()
-  img.save(img_byte_arr, format='JPEG', quality=quality)
-  current_size_kb = len(img_byte_arr.getvalue()) / 1024
+def resize_image(image, target_size_kb=None, quality=None):
+    """Resizes and compresses an image based on either target size or quality."""
+    try:
+        img = Image.open(image)
+    except Exception as e:
+        st.error(f"Error opening image: {e}")
+        return None
 
-  # Resize based on target size (if provided)
-  if target_size_kb:
-    while current_size_kb > target_size_kb and quality > 10:
-      quality -= 5
-      img_byte_arr = io.BytesIO()
-      img.save(img_byte_arr, format='JPEG', quality=quality)
-      current_size_kb = len(img_byte_arr.getvalue()) / 1024
-  else:
-    # Resize based on quality
-    pass  # No resizing for quality-based option
+    img_byte_arr = io.BytesIO()
 
-  img = Image.open(img_byte_arr)
-  return img
+    if target_size_kb is not None:
+        # Resize based on target size
+        current_quality = 95  # Start with a high quality
+        while True:
+            img_byte_arr = io.BytesIO()
+            img.save(img_byte_arr, format='JPEG', quality=current_quality)
+            current_size_kb = len(img_byte_arr.getvalue()) / 1024
+            if current_size_kb <= target_size_kb or current_quality <= 10:
+                break
+            current_quality -= 5  # Adjust quality in smaller steps
+        if current_quality <= 10 and current_size_kb > target_size_kb:
+            st.warning("Could not achieve the exact target size. Returning best possible result.")
+
+    elif quality is not None:
+        # Resize based on quality percentage
+        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
+
+    return img
 
 st.title("Image Resizer")
 st.write("Resize and compress your images!")
 
-# Upload image
 uploaded_file = st.file_uploader("Choose an Image:", type=["jpg", "jpeg", "png"])
 
 if uploaded_file is not None:
-  # Option selection for resizing
-  resize_option = st.selectbox("Resize Option:", ("Percentage Quality", "Desired Output Size"))
-
-  if resize_option == "Percentage Quality":
-    # User input for quality
-    quality = st.slider("Quality (%)", min_value=10, max_value=100, value=85)
-
-  elif resize_option == "Desired Output Size":
-    # Options for size units
-    size_unit_options = ("KB", "MB")
-    size_unit = st.selectbox("Size Unit", size_unit_options)
+    resize_option = st.selectbox("Resize Option:", ("Select an option", "Percentage Quality", "Desired Output Size"))
 
-    # User input for size
-    target_size = st.number_input("Target Size", min_value=1)
-
-    # Convert target size to KB based on unit
-    if size_unit == "MB":
-      target_size_kb = target_size * 1024
+    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)
+        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)
+    elif resize_option == "Select an option":
+        st.info("Please select a resize option to proceed.")
+        resized_image = Image.open(uploaded_file)
     else:
-      target_size_kb = target_size
-
-  # Process and display image
-  if uploaded_file and (quality or target_size_kb):
-    resized_image = resize_image(uploaded_file, target_size_kb, quality)
-    st.image(resized_image, caption="Resized Image")
-
-    # Download link
-    with io.BytesIO() as buffer:
-      resized_image.save(buffer, format="JPEG")
-      download_link = st.download_button(
-          label="Download Resized Image",
-          data=buffer.getvalue(),
-          file_name="resized_image.jpg",
-          mime="image/jpeg",
-      )
-
-# Requirements (req.txt)
-
-# Since the app uses only built-in libraries (streamlit and Pillow), there's no need for an external requirements file. 
-# However, if you plan to use additional libraries, you can create a req.txt file listing them. 
-# For example:
-# streamlit
-# Pillow
+        st.error("Invalid resize option selected.")
+        resized_image = None
+    
+    if resized_image:
+        st.image(resized_image, caption="Resized Image")
+        with io.BytesIO() as buffer:
+            resized_image.save(buffer, format="JPEG")
+            st.download_button(
+                label="Download Resized Image",
+                data=buffer.getvalue(),
+                file_name="resized_image.jpg",
+                mime="image/jpeg",
+            )