Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from PIL import Image, ImageOps
 import io
+from streamlit_drawable_canvas import st_canvas
 
 # Set Streamlit page configuration
 st.set_page_config(page_title="Image Transformer", page_icon="🖼️", layout="wide")
@@ -33,7 +34,7 @@ if uploaded_file is not None:
     with col1:
         st.image(image, caption="Original Image", use_column_width=False, width=250)
 
-    # Variables to store slider values
+    # Variables to store slider values or manual inputs
     new_width = None
     new_height = None
     crop_values = None
@@ -41,22 +42,45 @@ if uploaded_file is not None:
     pixel_size = None
     quality = None
 
-    # Display sliders based on operation, but only apply when the button is pressed
+    # Display sliders/input fields based on operation, but only apply when the button is pressed
     if operation == "Resize":
-        new_width = st.slider("Resize Image - Width", 50, 500, 100)
-        new_height = st.slider("Resize Image - Height", 50, 500, 100)
+        st.write("Resize Image:")
+        new_width = st.number_input("Width (in px)", min_value=50, max_value=2000, value=image.width)
+        new_height = st.number_input("Height (in px)", min_value=50, max_value=2000, value=image.height)
 
     elif operation == "Crop":
-        crop_values = st.slider("Crop Image - (left, upper, right, lower)", 0, 100, (10, 10, 90, 90))
+        st.write("Crop Image:")
+        st.write("Draw a rectangle on the image to crop it.")
+        canvas_result = st_canvas(
+            stroke_width=1,
+            stroke_color="#FF0000",
+            background_image=image,
+            update_streamlit=True,
+            height=image.height,
+            width=image.width,
+            drawing_mode="rect",
+            key="crop_canvas"
+        )
+        if canvas_result.json_data is not None:
+            try:
+                # Get the rectangle from the canvas
+                rect = canvas_result.json_data["objects"][0]
+                left = int(rect["left"])
+                top = int(rect["top"])
+                width = int(rect["width"])
+                height = int(rect["height"])
+                crop_values = (left, top, left + width, top + height)
+            except IndexError:
+                st.write("No rectangle drawn yet.")
 
     elif operation == "Rotate":
-        angle = st.slider("Rotate Image", 0, 360, 0)
+        angle = st.number_input("Rotate Image clockwise (in degrees)", min_value=0, max_value=360, value=0)
 
     elif operation == "Pixelate":
-        pixel_size = st.slider("Pixelate Image", 1, 20, 10)
+        pixel_size = st.number_input("Pixelate Size (larger number = more pixelation)", min_value=1, max_value=50, value=10)
 
     elif operation == "Compress":
-        quality = st.slider("Compress Image (Quality)", 10, 100, 80)
+        quality = st.number_input("Image Quality (Compression)", min_value=10, max_value=100, value=80)
 
     # Apply button with dynamic label based on the operation
     apply_button_label = f"Apply {operation}"
@@ -76,7 +100,7 @@ if uploaded_file is not None:
 
             # Resize Operation
             elif operation == "Resize" and new_width and new_height:
-                resized_image = image.resize((new_width, new_height))
+                resized_image = image.resize((new_width, new_height), Image.LANCZOS)
                 st.image(resized_image, caption=" ", use_column_width=False, width=250)
 
                 # Download button for resized image
@@ -96,7 +120,7 @@ if uploaded_file is not None:
 
             # Rotate Operation
             elif operation == "Rotate" and angle is not None:
-                rotated_image = image.rotate(angle)
+                rotated_image = image.rotate(-angle)  # Negative to rotate clockwise
                 st.image(rotated_image, caption=" ", use_column_width=False, width=250)
 
                 # Download button for rotated image
@@ -127,10 +151,10 @@ if uploaded_file is not None:
 
             # Compress Operation
             elif operation == "Compress" and quality:
-                compressed_image = image.copy()  # In real-world, this should apply compression logic
-                st.image(compressed_image, caption=" ", use_column_width=False, width=250)
+                # Save the image with the specified quality
+                buffered = io.BytesIO()
+                image.save(buffered, format="JPEG", quality=quality)
+                st.image(image, caption=" ", use_column_width=False, width=250)
 
                 # Download button for compressed image
-                buffered = io.BytesIO()
-                compressed_image.save(buffered, format="PNG")
-                st.download_button("Download Image", data=buffered.getvalue(), file_name="compressed_image.png", mime="image/png")
+                st.download_button("Download Image", data=buffered.getvalue(), file_name="compressed_image.jpeg", mime="image/jpeg")