Create app.py

app.py

@@ -0,0 +1,56 @@
+import gradio as gr
+import cv2
+import numpy as np
+from PIL import Image, ImageOps
+
+def extract_components(image, component):
+    image = np.array(image)
+    
+    if component == "Red Channel":
+        red_channel = image.copy()
+        red_channel[:, :, 1] = 0  # Zero out the green channel
+        red_channel[:, :, 2] = 0  # Zero out the blue channel
+        return Image.fromarray(red_channel)
+    
+    elif component == "Green Channel":
+        green_channel = image.copy()
+        green_channel[:, :, 0] = 0  # Zero out the red channel
+        green_channel[:, :, 2] = 0  # Zero out the blue channel
+        return Image.fromarray(green_channel)
+    
+    elif component == "Blue Channel":
+        blue_channel = image.copy()
+        blue_channel[:, :, 0] = 0  # Zero out the red channel
+        blue_channel[:, :, 1] = 0  # Zero out the green channel
+        return Image.fromarray(blue_channel)
+    
+    elif component == "Grayscale":
+        grayscale = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
+        return Image.fromarray(grayscale)
+    
+    elif component == "Edges":
+        grayscale = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
+        edges = cv2.Canny(grayscale, 100, 200)
+        return Image.fromarray(edges)
+    
+    elif component == "Original":
+        return Image.fromarray(image)
+    
+    else:
+        return Image.fromarray(image)
+
+# Define the Gradio interface
+iface = gr.Interface(
+    fn=extract_components,
+    inputs=[
+        gr.Image(label="Input Image"),
+        gr.Radio(["Red Channel", "Green Channel", "Blue Channel", "Grayscale", "Edges", "Original"], label="Select Component")
+    ],
+    outputs=gr.Image(label="Output Image"),
+    title="Image to Components",
+    description="Extract different components from an image (e.g., RGB channels, grayscale, edges)",
+    theme=Nymbo/Nymbo_theme_5
+)
+
+# Launch the app
+iface.launch()