Update app.py

app.py

@@ -31,14 +31,19 @@ def dl(inp):
         #out = f'{e}'
     return out
 
-def process_vid(file):
+def process_vid(file,cur_frame):
     new_video_in = str(file)
     capture = cv2.VideoCapture(new_video_in)    
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
     rev_img_searcher = ReverseImageSearcher()
     html_out=""
+    if cur_frame == ""
+        start_frame = 0
+    elif cur_frame != "":
+        start_frame = int(cur_frame)
     try:
-        for i in range(frame_count-1):
+        for i in range(start_frame, frame_count-1):
+            
             print(i)
             capture.set(cv2.CAP_PROP_POS_FRAMES, i)
             ret, frame_f = capture.read(i)
@@ -66,12 +71,13 @@ def process_vid(file):
                     Img: <a href='{search_item.image_url}' target='_blank' rel='noopener noreferrer'>{search_item.image_url}</a><br>
                     <img class='my_im' src='{search_item.image_url}'><br>
                     </div>"""
-                return (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'))
+                return (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'), "Complete", i)
             else:
+                yield (None,f"Searching Frame: {i}", "")
                 pass
     except Exception as e:
         return (gr.HTML(f'{e}'))
-    return (gr.HTML('No frame matches found.'))
+    return (gr.HTML('No frame matches found.'),"","")
 
 def process_im(file):
     read_file = Image.open(file)
@@ -104,6 +110,7 @@ def rev_im(image_url):
         Img: <a href='{search_item.image_url}' target='_blank' rel='noopener noreferrer'>{search_item.image_url}</a><br>
         <img class='my_im' src='{search_item.image_url}'><br>
         </div>"""
+        
     return (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'))
 
 def shuf(tog):
@@ -132,15 +139,22 @@ with gr.Blocks() as app:
                 vid_url=gr.Textbox(label="Video URL")
                 vid_url_btn=gr.Button("Load URL")
                 inp_vid=gr.Video(label="Search Video")
-                go_btn_vid=gr.Button()
+                stat_box=gr.Textbox("Status")
+                with gr.Row():
+                    go_btn_vid=gr.Button()
+                    next_btn=gr.Button()
+                
         gr.Column()            
             #paste_clip = gr.Button("Paste from Clipboard")
     with gr.Row():
         html_out = gr.HTML("""""")
+    with gr.Row(visible=True):
+        hid_box=gr.Textbox()
+    next_btn.click(process_vid,[inp_vid,hid_box],[html_out,hid_box])
     vid_url_btn.click(dl,vid_url,inp_vid)
     source_tog.change(shuf,[source_tog],[url_box,im_box,vid_box])
     inp_im.change(process_im,inp_im,[inp_url])
-    go_btn_vid.click(process_vid,inp_vid,[html_out])
+    go_btn_vid.click(process_vid,inp_vid,[html_out,stat_box,hid_box])
     go_btn_im.click(rev_im,inp_url,[html_out])
     go_btn_url.click(rev_im,inp_url,[html_out])
 app.launch()