Update app.py

app.py

@@ -31,50 +31,53 @@ def dl(inp):
         #out = f'{e}'
     return out
 
-def process_vid(file,cur_frame):
+def process_vid(file,cur_frame,every_n):
     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=""
+    count = 1
     if cur_frame == "" or cur_frame==None:
         start_frame = 0
     elif cur_frame != "" and cur_frame!=None:
         start_frame = int(cur_frame)
     try:
         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)
-            cv2.imwrite(f"{uid}-vid_tmp{i}.png", frame_f)
-            out = os.path.abspath(f"{uid}-vid_tmp{i}.png")
-            out_url = f'https://omnibus-reverse-image-dev.hf.space/file={out}'
-            print(out)
-            res = rev_img_searcher.search(out_url)
-            #print (res)
-            if len(res) > 0:
-                count = 0
-                for search_item in res:
-                    print (f'counting {count}')
-                    count+=1
-                    out_dict={
-                        'Title': f'{search_item.page_title}',
-                        'Site': f'{search_item.page_url}',
-                        'Img': f'{search_item.image_url}',
-                    }
-                    print (dir(search_item))
-                    html_out = f"""{html_out}
-                    <div>
-                    Title: {search_item.page_title}<br>
-                    Site: <a href='{search_item.page_url}' target='_blank' rel='noopener noreferrer'>{search_item.page_url}</a><br>
-                    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}'), f"Found frame: {i}", i)
-            else:
-                pass
-        yield (None,f"Searching Frame: {i}", "")
+            if count == int(every_n):
+                count = 1
+                print(i)
+                capture.set(cv2.CAP_PROP_POS_FRAMES, i)
+                ret, frame_f = capture.read(i)
+                cv2.imwrite(f"{uid}-vid_tmp{i}.png", frame_f)
+                out = os.path.abspath(f"{uid}-vid_tmp{i}.png")
+                out_url = f'https://omnibus-reverse-image-dev.hf.space/file={out}'
+                print(out)
+                res = rev_img_searcher.search(out_url)
+                #print (res)
+                if len(res) > 0:
+                    #count = 0
+                    for search_item in res:
+                        print (f'counting {count}')
+                        count+=1
+                        out_dict={
+                            'Title': f'{search_item.page_title}',
+                            'Site': f'{search_item.page_url}',
+                            'Img': f'{search_item.image_url}',
+                        }
+                        print (dir(search_item))
+                        html_out = f"""{html_out}
+                        <div>
+                        Title: {search_item.page_title}<br>
+                        Site: <a href='{search_item.page_url}' target='_blank' rel='noopener noreferrer'>{search_item.page_url}</a><br>
+                        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}'), f"Found frame: {i}", i)
+                else:
+                    pass
+            count +=1
+            yield (None,f"Searching Frame: {i}", "")
     except Exception as e:
         return (gr.HTML(f'{e}'),"","")
     return (gr.HTML('No frame matches found.'),"","")
@@ -113,13 +116,7 @@ def rev_im(image_url):
         
     return (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'))
 
-def shuf(tog):
-    if tog == "URL":
-        return gr.update(visible=True),gr.update(visible=False),gr.update(visible=False)
-    if tog == "Image":
-        return gr.update(visible=False),gr.update(visible=True),gr.update(visible=False)
-    if tog == "Video":
-        return gr.update(visible=False),gr.update(visible=False),gr.update(visible=True)
+
 
 
 
@@ -139,6 +136,8 @@ 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")
+                with gr.Row():
+                    every_n=gr.Number(value = 10)
                 stat_box=gr.Textbox(label="Status")
                 with gr.Row():
                     go_btn_vid=gr.Button("Start")
@@ -150,11 +149,18 @@ with gr.Blocks() as app:
         html_out = gr.HTML("""""")
     with gr.Row(visible=True):
         hid_box=gr.Textbox()
+    def shuf(tog):
+        if tog == "URL":
+            return gr.update(visible=True),gr.update(visible=False),gr.update(visible=False)
+        if tog == "Image":
+            return gr.update(visible=False),gr.update(visible=True),gr.update(visible=False)
+        if tog == "Video":
+            return gr.update(visible=False),gr.update(visible=False),gr.update(visible=True)        
     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,stat_box,hid_box])
+    go_btn_vid.click(process_vid,[inp_vid,hid_box,every_n],[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()