Update app.py

app.py

@@ -4,12 +4,16 @@ import numpy as np
 from PIL import Image
 import io
 
-st.title(" Image Augmentation Tool (Multi-image Support)")
+st.title("🖼️ Image Augmentation Tool (Multi-image Support)")
 
-uploaded_files = st.file_uploader("Upload one or more images", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
+uploaded_files = st.file_uploader(
+    "Upload one or more images", 
+    type=["jpg", "jpeg", "png"], 
+    accept_multiple_files=True
+)
 
 if uploaded_files:
-    st.sidebar.title("Transformations")
+    st.sidebar.title("🔧 Transformations")
 
     option = st.sidebar.selectbox("Choose a transformation", (
         "None",
@@ -29,8 +33,6 @@ if uploaded_files:
         image = Image.open(uploaded_file)
         img_array = np.array(image)
 
-        st.image(image, caption="Original Image", use_column_width=True)
-
         transformed_img = img_array.copy()
 
         try:
@@ -51,26 +53,36 @@ if uploaded_files:
                 transformed_img = cv2.warpAffine(transformed_img, M, (transformed_img.shape[1], transformed_img.shape[0]))
             elif option == "Crop (center 50%)":
                 h, w = transformed_img.shape[:2]
-                start_x = w//4
-                start_y = h//4
-                transformed_img = transformed_img[start_y:3*start_y, start_x:3*start_x]
+                start_x = w // 4
+                start_y = h // 4
+                transformed_img = transformed_img[start_y:3 * start_y, start_x:3 * start_x]
+
+            # Display images side-by-side
+            col1, col2 = st.columns(2)
+
+            with col1:
+                st.image(image, caption="Original Image", use_column_width=True)
+
+            with col2:
+                if option != "None":
+                    if len(transformed_img.shape) == 2:
+                        st.image(transformed_img, caption=f"Transformed: {option}", use_column_width=True, channels="GRAY")
+                    else:
+                        st.image(transformed_img, caption=f"Transformed: {option}", use_column_width=True)
 
-            if option != "None":
-                if len(transformed_img.shape) == 2:
-                    st.image(transformed_img, caption=f"Transformed: {option}", use_column_width=True, channels="GRAY")
+                    # Download button
+                    result = Image.fromarray(transformed_img)
+                    buf = io.BytesIO()
+                    result.save(buf, format="PNG")
+                    byte_im = buf.getvalue()
+                    st.download_button(
+                        label=f"⬇️ Download {uploaded_file.name}",
+                        data=byte_im,
+                        file_name=f"{uploaded_file.name.split('.')[0]}_transformed.png",
+                        mime="image/png"
+                    )
                 else:
-                    st.image(transformed_img, caption=f"Transformed: {option}", use_column_width=True)
+                    st.info("No transformation selected.")
 
-                # Download
-                result = Image.fromarray(transformed_img)
-                buf = io.BytesIO()
-                result.save(buf, format="PNG")
-                byte_im = buf.getvalue()
-                st.download_button(
-                    label=f"Download {uploaded_file.name}",
-                    data=byte_im,
-                    file_name=f"{uploaded_file.name.split('.')[0]}_transformed.png",
-                    mime="image/png"
-                )
         except Exception as e:
             st.error(f"❌ Error with {uploaded_file.name}: {e}")