Update pages/Image_Augmentation_Handson.py

pages/Image_Augmentation_Handson.py

@@ -3,35 +3,33 @@ from PIL import Image, ImageOps
 import numpy as np
 import io
 
-# Background color for the app
+# Set Background color for the app while uploading an image
 st.markdown(
     """
     <style>
     .stApp {
-        background-color: #1E1E1E;
-        color: #FFFFFF;
+        background-color: #000000;
     }
     </style>
     """,
     unsafe_allow_html=True,
 )
 
-st.title("🔧 Image Data Augmentation Tool")
-st.write("Enhance your images with various augmentation techniques.")
+st.title("Image Data Augmentation")
 
 # Image uploader
-uploaded_image = st.file_uploader("📤 Upload an Image (jpg, png, jpeg)", type=["jpg", "png", "jpeg"])
+uploaded_image = st.file_uploader("Upload an Image", type=["jpg", "png", "jpeg"])
 
 if uploaded_image:
     try:
         image = Image.open(uploaded_image)
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+        st.image(image, caption="Uploaded Image", use_container_width=True)
 
-        st.subheader("🎨 Choose Augmentation Options")
+        st.subheader("Select Augmentation Options")
 
         # Augmentation option
-        augmentation_option = st.selectbox(
-            "Select an Augmentation Technique:",
+        augmentation_option = st.radio(
+            "Choose an augmentation:",
             [
                 "None",
                 "Cropping",
@@ -41,90 +39,92 @@ if uploaded_image:
                 "Zooming Out",
                 "Translation",
                 "Shearing",
-                "Grayscale",
-                "Invert Colors",
             ],
         )
 
-        augmented_image = image.copy()
+        augmented_image = image
 
-        # Cropping
-        if augmentation_option == "Cropping":
-            col1, col2 = st.columns(2)
-            with col1:
+        try:
+            # Cropping
+            if augmentation_option == "Cropping":
                 left = st.slider("Left Crop", 0, image.width // 2, 0)
                 top = st.slider("Top Crop", 0, image.height // 2, 0)
-            with col2:
                 right = st.slider("Right Crop", 0, image.width // 2, 0)
                 bottom = st.slider("Bottom Crop", 0, image.height // 2, 0)
-            augmented_image = image.crop((left, top, image.width - right, image.height - bottom))
-
-        # Flipping
-        elif augmentation_option == "Flipping":
-            flip_option = st.radio("Flip Type", ["Horizontal", "Vertical"])
-            augmented_image = ImageOps.mirror(image) if flip_option == "Horizontal" else ImageOps.flip(image)
-
-        # Rotation
-        elif augmentation_option == "Rotation":
-            rotation_angle = st.slider("Rotation Angle (°)", -180, 180, 0)
-            augmented_image = image.rotate(rotation_angle, expand=True)
-
-        # Zooming In
-        elif augmentation_option == "Zooming In":
-            scale_factor = st.slider("Zoom Factor (%)", 100, 200, 100) / 100.0
-            if scale_factor != 1.0:
-                new_size = (int(image.width * scale_factor), int(image.height * scale_factor))
-                zoomed_image = image.resize(new_size)
-                augmented_image = zoomed_image.crop(((new_size[0] - image.width) // 2, 
-                                                     (new_size[1] - image.height) // 2, 
-                                                     (new_size[0] + image.width) // 2, 
-                                                     (new_size[1] + image.height) // 2))
-
-        # Zooming Out
-        elif augmentation_option == "Zooming Out":
-            scale_factor = st.slider("Zoom Factor (%)", 50, 100, 100) / 100.0
-            new_size = (int(image.width * scale_factor), int(image.height * scale_factor))
-            zoomed_image = image.resize(new_size)
-            padded_image = Image.new("RGB", (image.width, image.height), (255, 255, 255))
-            padded_image.paste(zoomed_image, ((image.width - new_size[0]) // 2, (image.height - new_size[1]) // 2))
-            augmented_image = padded_image
-
-        # Translation
-        elif augmentation_option == "Translation":
-            translation_x = st.slider("Horizontal Shift (px)", -100, 100, 0)
-            translation_y = st.slider("Vertical Shift (px)", -100, 100, 0)
-            augmented_image = image.transform(
-                image.size, Image.AFFINE, (1, 0, translation_x, 0, 1, translation_y), fillcolor=(255, 255, 255)
-            )
-
-        # Shearing
-        elif augmentation_option == "Shearing":
-            shear_angle = st.slider("Shearing Angle (°)", -45, 45, 0)
-            shear_matrix = (1, np.tan(np.radians(shear_angle)), 0, 0, 1, 0)
-            augmented_image = image.transform(
-                (image.width + abs(int(image.height * np.tan(np.radians(shear_angle)))), image.height),
-                Image.AFFINE,
-                shear_matrix,
-                fillcolor=(255, 255, 255),
-            )
-
-        # Grayscale
-        elif augmentation_option == "Grayscale":
-            augmented_image = ImageOps.grayscale(image)
-
-        # Invert Colors
-        elif augmentation_option == "Invert Colors":
-            augmented_image = ImageOps.invert(image)
+                augmented_image = image.crop((left, top, image.width - right, image.height - bottom))
+
+            # Flipping
+            elif augmentation_option == "Flipping":
+                flip_option = st.selectbox("Flip Type", ["Horizontal", "Vertical"])
+                if flip_option == "Horizontal":
+                    augmented_image = ImageOps.mirror(image)
+                elif flip_option == "Vertical":
+                    augmented_image = ImageOps.flip(image)
+
+            # Rotation
+            elif augmentation_option == "Rotation":
+                rotation_angle = st.slider("Rotation (°)", 0, 360, 0)
+                if rotation_angle != 0:
+                    augmented_image = image.rotate(rotation_angle, expand=True)
+
+            # Zooming In
+            elif augmentation_option == "Zooming In":
+                scale_factor = st.slider("Zoom Factor (%)", 100, 200, 100) / 100.0
+                if scale_factor != 1.0:
+                    width, height = image.size
+                    new_width = int(width * scale_factor)
+                    new_height = int(height * scale_factor)
+                    zoomed_image = image.resize((new_width, new_height))
+                    augmented_image = zoomed_image.crop(((new_width - width) // 2, (new_height - height) // 2,
+                                                        (new_width + width) // 2, (new_height + height) // 2))
+
+            # Zooming Out
+            elif augmentation_option == "Zooming Out":
+                scale_factor = st.slider("Zoom Factor (%)", 50, 100, 100) / 100.0
+                if scale_factor != 1.0:
+                    width, height = image.size
+                    new_width = int(width * scale_factor)
+                    new_height = int(height * scale_factor)
+                    zoomed_image = image.resize((new_width, new_height))
+                    padded_image = Image.new("RGB", (width, height), (255, 255, 255))
+                    padded_image.paste(zoomed_image, ((width - new_width) // 2, (height - new_height) // 2))
+                    augmented_image = padded_image
+
+            # Translation
+            elif augmentation_option == "Translation":
+                translation_x = st.slider("Horizontal Shift (Pixels)", -100, 100, 0)
+                translation_y = st.slider("Vertical Shift (Pixels)", -100, 100, 0)
+                if translation_x != 0 or translation_y != 0:
+                    translation_matrix = (1, 0, translation_x, 0, 1, translation_y)
+                    augmented_image = image.transform(
+                        image.size, Image.AFFINE, translation_matrix, fillcolor=(255, 255, 255)
+                    )
+
+            # Shearing
+            elif augmentation_option == "Shearing":
+                shear_angle = st.slider("Shearing Angle (°)", -45, 45, 0)
+                if shear_angle != 0:
+                    shear_matrix = (1, np.tan(np.radians(shear_angle)), 0, 0, 1, 0)
+                    augmented_image = image.transform(
+                        (image.width + abs(int(image.height * np.tan(np.radians(shear_angle)))), image.height),
+                        Image.AFFINE,
+                        shear_matrix,
+                        fillcolor=(255, 255, 255),
+                    )
+        except Exception as aug_error:
+            st.error(f"Error during augmentation: {aug_error}")
 
         # Display augmented image
-        st.subheader("🖼️ Augmented Image Preview")
-        st.image(augmented_image, caption=f"{augmentation_option} Applied", use_column_width=True)
+        st.subheader("Augmented Image")
+        st.image(augmented_image, caption=f"{augmentation_option} Applied", use_container_width=True)
 
-        # Download button
+        # Download augmented image
         buffer = io.BytesIO()
         augmented_image.save(buffer, format="PNG")
         buffer.seek(0)
-        st.download_button("💾 Download Augmented Image", buffer, file_name="augmented_image.png")
+        st.download_button(
+            "Download Augmented Image🎃", buffer, file_name="augmented_image.png"
+        )
 
     except Exception as e:
-        st.error(f"❌ An error occurred: {e}")
+        st.error(f"An error occurred: {e}")