Update pages/Types of Data.py

pages/Types of Data.py

@@ -378,6 +378,58 @@ print("Image Shape:", image_array.shape)  # (Height, Width, Channels)
 print("Pixel Data (Top-left):", image_array[0, 0])  # Pixel value at (0, 0)
     """, language="python")
 
+
+from transformers import pipeline
+import gradio as gr
+from PIL import Image, ImageOps
+import numpy as np
+import random
+
+# Function for image augmentation
+def augment_image(image, crop_size, flip, rotation):
+    img = Image.fromarray(image)
+    
+    # Crop
+    if crop_size > 0:
+        width, height = img.size
+        left = random.randint(0, crop_size)
+        top = random.randint(0, crop_size)
+        right = width - random.randint(0, crop_size)
+        bottom = height - random.randint(0, crop_size)
+        img = img.crop((left, top, right, bottom))
+    
+    # Flip
+    if flip:
+        img = ImageOps.mirror(img)
+    
+    # Rotate
+    if rotation != 0:
+        img = img.rotate(rotation, expand=True)
+    
+    return np.array(img)
+
+# Gradio interface
+def interface(image, crop_size, flip, rotation):
+    augmented_image = augment_image(image, crop_size, flip, rotation)
+    return augmented_image
+
+app = gr.Interface(
+    fn=interface,
+    inputs=[
+        gr.Image(type="numpy"),
+        gr.Slider(0, 100, step=1, label="Crop Size"),
+        gr.Checkbox(label="Flip"),
+        gr.Slider(0, 360, step=1, label="Rotation Angle")
+    ],
+    outputs=gr.Image(type="numpy"),
+    title="Image Augmentation Tool",
+    description="Upload an image to apply cropping, flipping, and rotation."
+)
+
+if __name__ == "__main__":
+    app.launch()
+
+
     # Placeholder button for GitHub link
         if st.button("Jupyter  Notebook 🔗 (Image)"):
             st.markdown("https://colab.research.google.com/drive/1BxJuxD1mzeuDnPIjwc-06J_GIU2JgO1_?usp=sharing")