add transform effect

app.py

@@ -88,18 +88,67 @@ def process_frame(frame):
                 
     return frame
 
+# Transform function
+def transform_cv2(frame, transform):
+    if transform == "cartoon":
+        # prepare color
+        img_color = cv2.pyrDown(cv2.pyrDown(frame))
+        for _ in range(6):
+            img_color = cv2.bilateralFilter(img_color, 9, 9, 7)
+        img_color = cv2.pyrUp(cv2.pyrUp(img_color))
+
+        # prepare edges
+        img_edges = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY)
+        img_edges = cv2.adaptiveThreshold(
+            cv2.medianBlur(img_edges, 7),
+            255,
+            cv2.ADAPTIVE_THRESH_MEAN_C,
+            cv2.THRESH_BINARY,
+            9,
+            2,
+        )
+        img_edges = cv2.cvtColor(img_edges, cv2.COLOR_GRAY2RGB)
+        # combine color and edges
+        img = cv2.bitwise_and(img_color, img_edges)
+        return img
+    elif transform == "edges":
+        # perform edge detection
+        img = cv2.cvtColor(cv2.Canny(frame, 100, 200), cv2.COLOR_GRAY2BGR)
+        return img
+    else:
+        return frame
+    
+
+
 # Gradio webcam input
-def webcam_input(frame):
+def webcam_input(frame, transform):
     frame = process_frame(frame)
+    frame = transform_cv2(frame, transform)
     return frame
 
 # Gradio Interface
 with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            input_img = gr.Image(label="Input", sources="webcam", streaming=True)
+    with gr.Column(elem_classes=["my-column"]):
+        with gr.Group(elem_classes=["my-group"]):
+            transform = gr.Dropdown(choices=["cartoon", "edges", "none"],
+                                    value="none", label="Transformation")
+            input_img = gr.Image(sources=["webcam"], type="numpy", streaming=True)
             next_button = gr.Button("Next Glasses")
-    input_img.stream(webcam_input, [input_img], [input_img], stream_every=0.1, concurrency_limit=30)
+    input_img.stream(webcam_input, [input_img, transform], [input_img], time_limit=30, stream_every=0.1)
+
     next_button.click(change_glasses, [], [])
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)
+
+
+
+# # Gradio Interface
+# with gr.Blocks() as demo:
+#     with gr.Row():
+#         with gr.Column():
+#             input_img = gr.Image(label="Input", sources="webcam", streaming=True)
+#             next_button = gr.Button("Next Glasses")
+#     input_img.stream(webcam_input, [input_img], [input_img], stream_every=0.1, concurrency_limit=30)
+#     next_button.click(change_glasses, [], [])
+# if __name__ == "__main__":
+#     demo.launch(share=True)