Update python.py

python.py

@@ -1,188 +1,174 @@
 import streamlit as st
-from PIL import Image
+from PIL import Image, ImageEnhance, ImageOps
 import random
-import os
-import zipfile
 import io
 
-# Directory to save images
-SAVE_DIR = r"C:\Users\syam0\Downloads\Image App"
-
-# Ensure the directory exists
-if not os.path.exists(SAVE_DIR):
-    os.makedirs(SAVE_DIR)
-
-# Augmentation functions with user inputs
-def scale_image(image, count, scale_factor=None):
-    if scale_factor is None:
-        scale_factor = random.uniform(0.8, 1.2)
-    return [image.resize((int(image.width * scale_factor), int(image.height * scale_factor))) for _ in range(count)]
-
-def translate_image(image, count, axis, translation_range=(-40, 40)):
-    translations = []
-    for _ in range(count):
-        if axis == "x-axis":
-            translation_x = random.randint(*translation_range)
-            translation_y = 0
-        elif axis == "y-axis":
-            translation_x = 0
-            translation_y = random.randint(*translation_range)
-        else:
-            translation_x = random.randint(*translation_range)
-            translation_y = random.randint(*translation_range)
-        translated_image = image.transform(image.size, Image.AFFINE, (1, 0, translation_x, 0, 1, translation_y))
-        translations.append(translated_image)
-    return translations
-
-def crop_image(image, count, crop_margin=None):
-    if crop_margin is None:
-        crop_margin = random.randint(5, 20)
-    return [image.crop((crop_margin, crop_margin, image.width - crop_margin, image.height - crop_margin)) for _ in range(count)]
-
-def rotate_image(image, count, angle=None):
-    if angle is None:
-        angle = random.randint(0, 360)
-    return [image.rotate(angle) for _ in range(count)]
-
-def shear_image(image, count, shear_factor=None):
-    if shear_factor is None:
-        shear_factor = random.uniform(-0.5, 0.5)
-    return [
-        image.transform(image.size, Image.AFFINE, (1, shear_factor, 0, shear_factor, 1, 0)) 
-        for _ in range(count)
-    ]
-
-# Function to save images to the specified directory
-def save_images(images, prefix):
-    saved_files = []
-    for i, img in enumerate(images):
-        filename = os.path.join(SAVE_DIR, f"{prefix}_image_{i+1}.png")
-        img.save(filename, format="PNG")
-        saved_files.append(filename)
-    return saved_files
-
-# Function to zip the images
-def create_zip(saved_files):
-    zip_buffer = io.BytesIO()
-    with zipfile.ZipFile(zip_buffer, "w") as zipf:
-        for file in saved_files:
-            zipf.write(file, os.path.basename(file))
-    zip_buffer.seek(0)
-    return zip_buffer
-
-# Custom CSS for background and styling
-st.markdown(
-    """
-    <style>
-    body {
-        background: linear-gradient(#ffaa80, #ff5500, #b33c00);  /* Gradient background from yellow to red */
-        font-family: 'Arial', sans-serif;
-        color: #333;
-    }
-    .stApp {
-        background: linear-gradient(#66d9ff, #d966ff, #66ff66);  /* Gradient background for the entire app */
-    }
-    .stFileUploader {
-        display: flex;
-        justify-content: center;
-        align-items: center;
-        padding: 30px;
-        border: 2px dashed #4CAF50;  /* Green dashed border for a fresh look */
-        background-color: rgba(76, 175, 80, 0.1);  /* Light green background on hover */
-        border-radius: 8px;
-        cursor: pointer;
-    }
-    .stFileUploader:hover {
-        background-color: rgba(76, 175, 80, 0.2);  /* Slightly darker green on hover */
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
-
-# Streamlit app
-st.title("Image Augmentation App with Gradient Background")
-
-# Image uploader with custom styling
-uploaded_image = st.file_uploader("Drag and Drop Your Image", type=["jpg", "jpeg", "png"], label_visibility="collapsed")
-
-if uploaded_image:
-    image = Image.open(uploaded_image)
-    st.image(image, caption="Uploaded Image", use_container_width=True)  # Fixed the deprecation warning by using `use_container_width=True`
-
-    # User inputs for augmentation
-    st.write("### Generate Augmented Images")
-    num_augments = st.number_input("How many augmented images would you like to generate?", min_value=1, max_value=20, value=5)
-
-    # Additional inputs for specific techniques
-    rotate_angle = st.number_input("Enter the angle for rotation (0-360):", min_value=0, max_value=360, value=random.randint(0, 360))
-    scale_factor = st.number_input("Enter the scaling factor (0.8-1.2):", min_value=0.8, max_value=1.2, value=random.uniform(0.8, 1.2))
-    crop_margin = st.number_input("Enter the crop margin (5-20):", min_value=5, max_value=20, value=random.randint(5, 20))
-    shear_factor = st.number_input("Enter the shear factor (-0.5 to 0.5):", min_value=-0.5, max_value=0.5, value=random.uniform(-0.5, 0.5))
-    translate_axis = st.selectbox("Select axis for translation:", ["x-axis", "y-axis", "both axes"])
-
-    # Generate and save all augmentations
-    if st.button("Generate All Techniques"):
-        all_augmentations = scale_image(image, num_augments, scale_factor) + \
-                            translate_image(image, num_augments, translate_axis) + \
-                            crop_image(image, num_augments, crop_margin) + \
-                            rotate_image(image, num_augments, rotate_angle) + \
-                            shear_image(image, num_augments, shear_factor)
+# --- Helper Functions for Image Transformations ---
+def translate_image(image, x_offset, y_offset):
+    """Translate the image by x and y offsets."""
+    return image.transform(
+        image.size,
+        Image.AFFINE,
+        (1, 0, x_offset, 0, 1, y_offset),
+        resample=Image.BICUBIC
+    )
+
+def rotate_image(image, angle):
+    """Rotate the image by a given angle."""
+    return image.rotate(angle, expand=True)
+
+def scale_image(image, scale_factor):
+    """Scale the image by a given factor."""
+    new_size = (int(image.width * scale_factor), int(image.height * scale_factor))
+    return image.resize(new_size)
+
+def crop_image(image, left, upper, right, lower):
+    """Crop the image."""
+    return image.crop((left, upper, right, lower))
+
+def flip_image(image, flip_type):
+    """Flip the image horizontally or vertically."""
+    if flip_type == 'Horizontal':
+        return ImageOps.mirror(image)
+    elif flip_type == 'Vertical':
+        return ImageOps.flip(image)
+    return image
+
+# --- Main App Function ---
+def main():
+    st.set_page_config(
+        page_title="Interactive Image Augmentation Tool",
+        layout="wide"
+    )
+    
+    # --- Styling ---
+    st.markdown(
+        """
+        <style>
+        body {
+            background: linear-gradient(to right, #F5F5DC, #E6D8C3, #D2B48C, #DEB887);
+            color: #5A5A5A;
+        }
+        .stTitle {
+            color: #3E2723;
+            text-align: center;
+            font-family: 'Arial', sans-serif;
+        }
+        .stButton>button {
+            background-color: #C19A6B;
+            color: white;
+            border-radius: 8px;
+        }
+        .stDownloadButton>button {
+            background-color: #8B5E3C;
+            color: white;
+            border-radius: 8px;
+        }
+        .instruction-box {
+            background: #FAF0E6;
+            border-left: 5px solid #C19A6B;
+            padding: 10px;
+            margin: 20px 0;
+            border-radius: 5px;
+        }
+        .use-case-box {
+            background: #FFF5EE;
+            border-left: 5px solid #8B5E3C;
+            padding: 10px;
+            margin: 20px 0;
+            border-radius: 5px;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
+    # --- Title and Introduction ---
+    st.title("🖼️ Interactive Image Augmentation Tool")
+    st.write("""
+    Welcome to the **Interactive Image Augmentation Tool**! This app allows you to transform and enhance your images 
+    using various augmentation techniques like **Translation**, **Rotation**, **Scaling**, **Cropping**, and **Flipping**.
+    """)
+
+    st.markdown("""
+    <div class='instruction-box'>
+    <h4>📝 Instructions for Use:</h4>
+    <ul>
+        <li>Upload an image using the uploader below.</li>
+        <li>Select a transformation technique from the dropdown menu.</li>
+        <li>Adjust the parameters using sliders and dropdowns.</li>
+        <li>Preview the transformed image.</li>
+        <li>Download the transformed image if you're satisfied.</li>
+    </ul>
+    </div>
+    """, unsafe_allow_html=True)
+
+    st.markdown("""
+    <div class='use-case-box'>
+    <h4>🌟 Who Can Use This App?</h4>
+    <ul>
+        <li><b>Data Scientists:</b> Generate augmented datasets for image classification models.</li>
+        <li><b>Graphic Designers:</b> Experiment with creative visual effects.</li>
+        <li><b>Researchers:</b> Prepare data for analysis and experiments.</li>
+        <li><b>Students:</b> Learn image processing techniques interactively.</li>
+    </ul>
+    </div>
+    """, unsafe_allow_html=True)
+
+    # --- File Uploader ---
+    uploaded_file = st.file_uploader("📤 Upload an Image", type=['jpg', 'jpeg', 'png'])
+    if uploaded_file:
+        image = Image.open(uploaded_file)
+        st.image(image, caption="Original Image", use_column_width=True)
         
-        # Save images to the specified folder
-        st.write("Generating and Saving Images...")
-        saved_files = save_images(all_augmentations, "combined")
+        st.subheader("🔄 Apply Transformations")
+        transformation = st.selectbox(
+            "Choose a Transformation",
+            ["Translate", "Rotate", "Scale", "Crop", "Flip"]
+        )
         
-        # Display generated images
-        st.write("Generated Augmented Images:")
-        for img in all_augmentations:
-            st.image(img, use_container_width=True)
-
-        # Create a zip file
-        zip_buffer = create_zip(saved_files)
+        if transformation == "Translate":
+            x_offset = st.slider("X Offset", -100, 100, 0)
+            y_offset = st.slider("Y Offset", -100, 100, 0)
+            transformed_image = translate_image(image, x_offset, y_offset)
         
-        # Provide a download button for the zip file
-        st.download_button(
-            label="Download Augmented Images (ZIP)",
-            data=zip_buffer,
-            file_name="augmented_images.zip",
-            mime="application/zip"
-        )
-
-    # Generate and save technique-specific augmentations
-    st.write("### Select Specific Technique")
-    technique = st.selectbox("Choose an augmentation technique:", 
-                              ["Scale", "Translate", "Crop", "Rotate", "Shear"])
-    technique_count = st.number_input(f"How many images do you want for {technique}?", min_value=1, max_value=20, value=5)
-
-    if st.button(f"Generate and Save {technique} Images"):
-        if technique == "Scale":
-            specific_augmentations = scale_image(image, technique_count, scale_factor)
-        elif technique == "Translate":
-            specific_augmentations = translate_image(image, technique_count, translate_axis)
-        elif technique == "Crop":
-            specific_augmentations = crop_image(image, technique_count, crop_margin)
-        elif technique == "Rotate":
-            specific_augmentations = rotate_image(image, technique_count, rotate_angle)
-        elif technique == "Shear":
-            specific_augmentations = shear_image(image, technique_count, shear_factor)
+        elif transformation == "Rotate":
+            angle = st.slider("Rotation Angle", 0, 360, 0)
+            transformed_image = rotate_image(image, angle)
+        
+        elif transformation == "Scale":
+            scale = st.slider("Scale Factor", 0.5, 2.0, 1.0)
+            transformed_image = scale_image(image, scale)
         
-        # Save images to the specified folder
-        st.write(f"Generating and Saving {technique} Images...")
-        saved_files = save_images(specific_augmentations, technique.lower())
+        elif transformation == "Crop":
+            left = st.slider("Left", 0, image.width, 0)
+            upper = st.slider("Upper", 0, image.height, 0)
+            right = st.slider("Right", left, image.width, image.width)
+            lower = st.slider("Lower", upper, image.height, image.height)
+            transformed_image = crop_image(image, left, upper, right, lower)
         
-        # Display generated images
-        st.write(f"Generated {technique} Augmented Images:")
-        for img in specific_augmentations:
-            st.image(img, use_container_width=True)
+        elif transformation == "Flip":
+            flip_type = st.selectbox("Flip Type", ["None", "Horizontal", "Vertical"])
+            transformed_image = flip_image(image, flip_type)
+        
+        else:
+            st.warning("Select a valid transformation!")
+            return
         
-        # Create a zip file
-        zip_buffer = create_zip(saved_files)
+        st.image(transformed_image, caption="Transformed Image", use_column_width=True)
         
-        # Provide a download button for the zip file
+        # --- Download Image ---
+        img_buffer = io.BytesIO()
+        transformed_image.save(img_buffer, format="PNG")
         st.download_button(
-            label=f"Download {technique} Augmented Images (ZIP)",
-            data=zip_buffer,
-            file_name=f"{technique.lower()}_augmented_images.zip",
-            mime="application/zip"
-        )
+            label="📥 Download Transformed Image",
+            data=img_buffer.getvalue(),
+            file_name="transformed_image.png",
+            mime="image/png"
+        )
+
+    st.markdown("---")
+    st.write("🎯 **Experiment with transformations and download your enhanced images today!**")
+
+if __name__ == "__main__":
+    main()