Update app.py

app.py

@@ -4,7 +4,7 @@ import matplotlib.colors as mcolors
 import cv2
 import gradio as gr
 
-from channel_functions import individual_channel_image, channel_distribution_plotter, which_channel_dominates
+from channel_functions import individual_channel_image_final, individual_channel_image, channel_distribution_plotter, which_channel_dominates
 description1 = '''This tab shows which channel dominates which part of an image\n
 Upload an image or drag an online image to the upload section. Bring it to the rectangular area. Release it when the border becomes red'''
 
@@ -12,6 +12,9 @@ original_image_plot = gr.Radio(['yes','no'], label = "Do you want the original i
 original_image_opacity = gr.Slider(0,1, value = 0.3, label = "Opacity of the original image")
 channel_opacity = gr.Slider(0,1, value = 0.7, label = "Opacity of the output channel-domination image")
 
+
+individual_channel_choice = gr.Radio(['Red','Green','Blue', 'All'], label = 'Which channel do you want to see the image in?')
+
 iface1 = gr.Interface(
                       fn = which_channel_dominates, 
                       inputs = [gr.Image(), original_image_plot, original_image_opacity, channel_opacity],
@@ -28,7 +31,13 @@ iface2 = gr.Interface(
                      description = 'Something random'
                      )
 
+iface3 = gr.Interface(
+                    fn = individual_channel_image_final,
+                    inputs = [gr.Image(), individual_channel_choice],
+                    outputs = "image"    ,
+                    title = "Images in individual channels",
+                    description = "Something random"
+                    )
 
-combinedInterface = gr.TabbedInterface([iface1, iface2],['Channel domination', 'Channel distribution'])
+combinedInterface = gr.TabbedInterface([iface1, iface2, iface3],['Channel domination', 'Channel distribution', "Individual channel"])
 combinedInterface.launch(share=False)
-