Rename pages/Image_Augmentation_Handson.py to pages/Image_Augmentation.py

pages/{Image_Augmentation_Handson.py → Image_Augmentation.py}

@@ -1,5 +1,5 @@
 import streamlit as st
-from PIL import Image, ImageOps
+from PIL import Image, ImageOps, ImageEnhance
 import numpy as np
 import io
 
@@ -29,104 +29,85 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
-st.title("🌟 Enhanced Image Augmentation Tool", anchor=False)
+st.title("🌟 Image Augmentation Tool Hands-On Experience", anchor=False)
 
-uploaded_image = st.file_uploader("📤 Upload an Image", type=["jpg", "png", "jpeg"])
+uploaded_image = st.sidebar.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.sidebar.image(image, caption="📸 Uploaded Image", use_column_width=True)
 
-        st.subheader("✨ Select Augmentation Options")
+        st.sidebar.subheader("✨ Augmentation Options")
         
-        augmentation_option = st.radio(
-            "Choose an augmentation:",
-            [
-                "None",
-                "Cropping",
-                "Flipping",
-                "Rotation",
-                "Zoom In",
-                "Zoom Out",
-                "Translation",
-                "Shearing",
-            ],
+        augmentation_options = st.sidebar.multiselect(
+            "Choose augmentations:",
+            ["Cropping", "Flipping", "Rotation", "Zoom In", "Zoom Out", "Translation", "Shearing"]
         )
+        
+        augmented_image = image.copy()
+        
+        # Live preview toggle
+        live_preview = st.sidebar.checkbox("🔄 Live Preview", value=True)
 
-        augmented_image = image
-
-        try:
-            # Cropping
+        for augmentation_option in augmentation_options:
             if augmentation_option == "Cropping":
                 left = st.slider("Left Crop", 0, image.width // 2, 0)
                 top = st.slider("Top Crop", 0, image.height // 2, 0)
                 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))
+                augmented_image = augmented_image.crop((left, top, image.width - right, image.height - bottom))
 
-            # Flipping
             elif augmentation_option == "Flipping":
-                flip_option = st.selectbox("🔄 Flip Type", ["Horizontal", "Vertical"])
+                flip_option = st.selectbox("🔄 Flip Type", ["Horizontal", "Vertical"], key=f"flip_{augmentation_option}")
                 if flip_option == "Horizontal":
-                    augmented_image = ImageOps.mirror(image)
+                    augmented_image = ImageOps.mirror(augmented_image)
                 elif flip_option == "Vertical":
-                    augmented_image = ImageOps.flip(image)
+                    augmented_image = ImageOps.flip(augmented_image)
 
-            # Rotation
             elif augmentation_option == "Rotation":
                 rotation_angle = st.slider("⤾ Rotation (°)", 0, 360, 0)
-                augmented_image = image.rotate(rotation_angle, expand=True)
+                augmented_image = augmented_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
+                width, height = augmented_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,
-                ))
+                zoomed_image = augmented_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
+                width, height = augmented_image.size
                 new_width = int(width * scale_factor)
                 new_height = int(height * scale_factor)
-                zoomed_image = image.resize((new_width, new_height))
+                zoomed_image = augmented_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)
                 translation_matrix = (1, 0, translation_x, 0, 1, translation_y)
-                augmented_image = image.transform(
-                    image.size, Image.AFFINE, translation_matrix, fillcolor=(0, 0, 0)
+                augmented_image = augmented_image.transform(
+                    augmented_image.size, Image.AFFINE, translation_matrix, fillcolor=(0, 0, 0)
                 )
 
-            # 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),
+                augmented_image = augmented_image.transform(
+                    (augmented_image.width + abs(int(augmented_image.height * np.tan(np.radians(shear_angle)))), augmented_image.height),
                     Image.AFFINE,
                     shear_matrix,
                     fillcolor=(0, 0, 0),
                 )
-        except Exception as aug_error:
-            st.error(f"⚠️ Augmentation Error: {aug_error}")
-
+        
         st.subheader("🖼️ Augmented Image")
-        st.image(augmented_image, caption=f"{augmentation_option} Applied", use_container_width=True)
+        st.image(augmented_image, caption="Augmentation Applied", use_column_width=True)
 
         buffer = io.BytesIO()
         augmented_image.save(buffer, format="PNG")