Update app.py

app.py

@@ -4,13 +4,8 @@ import cv2 # opencv2 package for python.
 import torch
 from pytube import YouTube
 from ultralyticsplus import YOLO, render_result
-#from imageai.Detection import ObjectDetection
 
 
-#obj_detect.setModelPath(r"C:/Datasets/yolo.h5")
-#obj_detect.loadModel()
-
-#from torch import hub # Hub contains other models like FasterRCNN
 model = YOLO('ultralyticsplus/yolov8s')
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 URL = "https://www.youtube.com/watch?v=dQw4w9WgXcQ" #URL to parse
@@ -22,10 +17,6 @@ model.overrides['agnostic_nms'] = False  # NMS class-agnostic
 model.overrides['max_det'] = 1000  # maximum number of detections per image
 model.to(device)
 
-#class Player:
-    
-
-
 
 def load(URL):
     yt = YouTube(URL)
@@ -36,19 +27,17 @@ def load(URL):
     frame_count = int(player.get(cv2.CAP_PROP_FRAME_COUNT))
     frame_fps = (player.get(cv2.CAP_PROP_FPS))
     #process.release()
-   
     return vid_cap,frame_num,frame_count,frame_fps
     
 def vid_play(cap,frame_num):
-    #player = cv2.VideoCapture(cap)
     assert player.isOpened() # Make sure that their is a stream. 
     player.set(cv2.CAP_PROP_POS_FRAMES, int(frame_num))
-
     ret, frame_bgr = player.read(int(frame_num))
     frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)        
     results = model.predict(frame)
     render = render_result(model=model, image=frame, result=results[0])
     return render
+
 def fw_fn(cur,last):
     next = cur+1
     if next > last:
@@ -65,25 +54,19 @@ def tog_off():
     return 0
         
 def pl_fn(cap,cur,last,fps,pl_tog):
-    #player = cv2.VideoCapture(cap)
-    #assert player.isOpened() # Make sure that their is a stream. 
     player.set(cv2.CAP_PROP_POS_FRAMES, cur)
     ret, frame_bgr = player.read(cur)
     frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
     results = model.predict(frame)
     render = render_result(model=model, image=frame, result=results[0])
     if pl_tog ==1:
-        #ins_cnt+=1
         cur+=1
     else:
         cur = cur        
     return render,cur
 
-
-
-
 with gr.Blocks() as app:
-    #player = gr.State()
+    gr.Markdown("""<center><h1>Mediocre Video Object Detection</h1></center>""")
     with gr.Row():
         with gr.Column():
             youtube_url = gr.Textbox(label="YouTube URL",value=f"{URL}")
@@ -91,10 +74,10 @@ with gr.Blocks() as app:
             output_win = gr.Video()
         with gr.Column():
             with gr.Row():
-                cur_frame = gr.Number()
-                fps_frames = gr.Number()
-                total_frames = gr.Number(interactive=False)
-            run_button = gr.Button()
+                cur_frame = gr.Number(label="Current Frame")
+                fps_frames = gr.Number(label="Video FPS",interactive=False)
+                total_frames = gr.Number(label="Total Frames",interactive=False)
+            #run_button = gr.Button()
             with gr.Row():
                 bk = gr.Button("<")
                 pl = gr.Button("Play")
@@ -102,17 +85,15 @@ with gr.Blocks() as app:
                 fw = gr.Button(">")
             det_win = gr.Image(source="webcam", streaming=True)
     with gr.Row():
-        pl_tog=gr.Number()
-        ins_cnt=gr.Number()   
-    pl.click(tog_on,None,pl_tog)
-    st.click(tog_off,None,pl_tog)
+        pl_tog=gr.Number(visible=False)
+        ins_cnt=gr.Number(visible=False)   
+    pl.click(tog_on,None,pl_tog,show_progress=False)
+    st.click(tog_off,None,pl_tog,show_progress=False)
     pl_tog.change(pl_fn,[output_win,cur_frame,total_frames,fps_frames,pl_tog],[det_win,cur_frame],show_progress=False)
-    #cur_frame.change(pl_fn,[output_win,cur_frame,total_frames,fps_frames,pl_tog],[det_win,cur_frame],every=0.1,show_progress=False)
-    
     cur_frame.change(pl_fn,[output_win,cur_frame,total_frames,fps_frames,pl_tog],[det_win,cur_frame],show_progress=False)
-    bk.click(bk_fn,[cur_frame],cur_frame)
-    fw.click(fw_fn,[cur_frame,total_frames],cur_frame)
+    bk.click(bk_fn,[cur_frame],cur_frame,show_progress=False)
+    fw.click(fw_fn,[cur_frame,total_frames],cur_frame,show_progress=False)
     load_button.click(load,youtube_url,[output_win,cur_frame,total_frames,fps_frames])
-    run_button.click(vid_play, [output_win,cur_frame], det_win) 
+    #run_button.click(vid_play, [output_win,cur_frame], det_win) 
 
 app.queue(concurrency_count=10).launch()