Update app.py

app.py

@@ -1,21 +1,65 @@
 import gradio as gr
 import cv2
-import numpy as np
-invideo = cv2.VideoCapture('https://www.youtube.com/watch?v=5WbClF6WXRE')
-cv2.namedWindow('Live cam',cv.WINDOW_NORMAL)
-if not invideo.isOpened():
-    print("Cannot open camera")
-    exit()
-while True:
-    # Capture frame-by-frame
-    read_correct, frame = invideo.read()
-    if not read_correct:
-        print("cannot read correctly")
-        break
-    cv2.imshow('Video Capture from Camera', frame)
-    if cv.waitKey(25) == ord('q'):
-        break
-invideo.release()
-cv.destroyAllWindows()
 
+def color_to_gray(img):
+    new_img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
+    return new_img
+
+
+def predict(video_in, image_in_video, image_in_img):
+    if video_in == None and image_in_video == None and image_in_img == None:
+        raise gr.Error("Please upload a video or image.")
+    if image_in_video or image_in_img:
+        print("image", image_in_video, image_in_img)
+        image = image_in_video or image_in_img
+        return image
+
+    return video_in
+
+
+def toggle(choice):
+    if choice == "webcam":
+        return gr.update(visible=True, value=None), gr.update(visible=False, value=None)
+    else:
+        return gr.update(visible=False, value=None), gr.update(visible=True, value=None)
+
+
+with gr.Blocks() as blocks:
+    gr.Markdown("### Video or Image? WebCam or Upload?""")
+    with gr.Tab("Video") as tab:
+        with gr.Row():
+            with gr.Column():
+                video_or_file_opt = gr.Radio(["webcam", "upload"], value="webcam",
+                                             label="How would you like to upload your video?")
+                video_in = gr.Video(source="webcam", include_audio=False)
+                video_or_file_opt.change(fn=lambda s: gr.update(source=s, value=None), inputs=video_or_file_opt,
+                                         outputs=video_in, queue=False, show_progress=False)
+            with gr.Column():
+                video_out = gr.Video()
+        run_btn = gr.Button("Run")
+        run_btn.click(fn=predict, inputs=[video_in], outputs=[video_out])
+        gr.Examples(fn=predict, examples=[], inputs=[
+                    video_in], outputs=[video_out])
+
+    with gr.Tab("Image"):
+        with gr.Row():
+            with gr.Column():
+                image_or_file_opt = gr.Radio(["webcam", "file"], value="webcam",
+                                             label="How would you like to upload your image?")
+                image_in_video = gr.Image(source="webcam", type="filepath")
+                image_in_img = gr.Image(
+                    source="upload", visible=False, type="filepath")
+
+                image_or_file_opt.change(fn=toggle, inputs=[image_or_file_opt],
+                                         outputs=[image_in_video, image_in_img], queue=False, show_progress=False)
+            with gr.Column():
+                image_out = gr.Image()
+        run_btn = gr.Button("Run")
+        run_btn.click(fn=predict, inputs=[
+                      image_in_img, image_in_video], outputs=[image_out])
+        gr.Examples(fn=predict, examples=[],  inputs=[
+                    image_in_img, image_in_video], outputs=[image_out])
+
+blocks.queue()
+blocks.launch()