Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from PIL import Image, ImageFilter
 import numpy as np
 import pandas as pd
+from streamlit_cropper import st_cropper
 
 # Simple app: convert user input into ASCII codes and binary labels
 
@@ -14,9 +15,6 @@ def string_to_binary_labels(s: str) -> list[int]:
     return bits
 
 def clean_image(img: Image.Image, min_size: int = 256) -> Image.Image:
-    """
-    Preprocess image to reduce noise: resize if needed and apply light blur.
-    """
     img = img.convert("RGB")
     if img.width < min_size or img.height < min_size:
         img = img.resize((min_size, min_size))
@@ -121,9 +119,14 @@ with tab2:
         img = Image.open(uploaded_file)
         st.image(img, caption="Uploaded Image", use_column_width=True)
 
+        st.subheader("Crop the image with drag and select")
+        cropped_img = st_cropper(img, realtime_update=True, box_color='blue', aspect_ratio=None)
+
+        st.image(cropped_img, caption="Cropped Image", use_column_width=True)
+
         max_pixels = st.slider("Max number of pixels to encode", min_value=32, max_value=1024, value=256, step=32)
 
-        binary_labels = image_to_binary_labels_rgb(img, max_pixels=max_pixels)
+        binary_labels = image_to_binary_labels_rgb(cropped_img, max_pixels=max_pixels)
 
         st.subheader("Binary Labels from Image")
         num_groups = (len(binary_labels) + 31) // 32