Update pages/Image_Augmentation_Handson.py

pages/Image_Augmentation_Handson.py

@@ -3,31 +3,43 @@ from PIL import Image, ImageOps
 import numpy as np
 import io
 
-# Set Background color for the app while uploading an image
+# Custom CSS for enhanced styling
 st.markdown(
     """
     <style>
     .stApp {
-        background-color: #000000;
+        background-color: #121212;
+        color: #E0E0E0;
+    }
+    .stTitle {
+        text-align: center;
+        color: #4CAF50;
+    }
+    .css-1d391kg p, .css-1v0mbdj p {
+        color: #B0BEC5;
+    }
+    .css-1aumxhk .stSlider {
+        background-color: #4CAF50;
+    }
+    .css-2trqyj, .css-16idsys p {
+        color: #BB86FC;
     }
     </style>
     """,
     unsafe_allow_html=True,
 )
 
-st.title("Image Data Augmentation")
+st.title("🌟 Enhanced Image Augmentation Tool", anchor=False)
 
-# Image uploader
-uploaded_image = st.file_uploader("Upload an Image", 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_container_width=True)
-
-        st.subheader("Select Augmentation Options")
+        st.image(image, caption="📸 Uploaded Image", use_container_width=True)
 
-        # Augmentation option
+        st.subheader("✨ Select Augmentation Options")
+        
         augmentation_option = st.radio(
             "Choose an augmentation:",
             [
@@ -35,8 +47,8 @@ if uploaded_image:
                 "Cropping",
                 "Flipping",
                 "Rotation",
-                "Zooming In",
-                "Zooming Out",
+                "Zoom In",
+                "Zoom Out",
                 "Translation",
                 "Shearing",
             ],
@@ -55,7 +67,7 @@ if uploaded_image:
 
             # Flipping
             elif augmentation_option == "Flipping":
-                flip_option = st.selectbox("Flip Type", ["Horizontal", "Vertical"])
+                flip_option = st.selectbox("🔄 Flip Type", ["Horizontal", "Vertical"])
                 if flip_option == "Horizontal":
                     augmented_image = ImageOps.mirror(image)
                 elif flip_option == "Vertical":
@@ -63,68 +75,63 @@ if uploaded_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
+                rotation_angle = st.slider("⤾ Rotation (°)", 0, 360, 0)
+                augmented_image = image.rotate(rotation_angle, expand=True)
+
+            # Zoom In
+            elif augmentation_option == "Zoom In":
+                scale_factor = st.slider("🔍 Zoom Factor (%)", 100, 200, 100) / 100.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,
+                ))
+
+            # Zoom Out
+            elif augmentation_option == "Zoom Out":
+                scale_factor = st.slider("🔎 Zoom Out Factor (%)", 50, 100, 100) / 100.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), (0, 0, 0))
+                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)
-                    )
+                translation_x = st.slider("↔ Horizontal Shift (Pixels)", -100, 100, 0)
+                translation_y = st.slider("↕ Vertical Shift (Pixels)", -100, 100, 0)
+                translation_matrix = (1, 0, translation_x, 0, 1, translation_y)
+                augmented_image = image.transform(
+                    image.size, Image.AFFINE, translation_matrix, fillcolor=(0, 0, 0)
+                )
 
             # 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),
-                    )
+                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=(0, 0, 0),
+                )
         except Exception as aug_error:
-            st.error(f"Error during augmentation: {aug_error}")
+            st.error(f"⚠️ Augmentation Error: {aug_error}")
 
-        # Display augmented image
-        st.subheader("Augmented Image")
+        st.subheader("🖼️ Augmented Image")
         st.image(augmented_image, caption=f"{augmentation_option} Applied", use_container_width=True)
 
-        # 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}")