Update app.py

app.py

@@ -29,49 +29,60 @@ def dl(inp):
     except Exception as e:
         print (e)
         #out = f'{e}'
-    return out
+    return out,gr.HTML(""),"",""
 
-def process_vid(file):
+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 = int(every_n)
+    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(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.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}'))
-            else:
-                pass
+        for i in range(start_frame, frame_count-1):
+            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)
+                out_cnt =0
+                if len(res) > 0:
+                    #count = 0
+                    for search_item in res:
+                        print (f'counting {count}')
+                        out_cnt+=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: {out_cnt}</h1><br>{html_out}'), f"Found frame: {i}", i+int(every_n))
+                else:
+                    pass
+            count +=1
+            print (i+1)
+            #return (None,f"Searching Frame: {i}", "")
     except Exception as e:
-        return (gr.HTML(f'{e}'))
-    return (gr.HTML('No frame matches found.'))
+        return (gr.HTML(f'{e}'),"","")
+    return (gr.HTML('No frame matches found.'),"","")
 
 def process_im(file):
     read_file = Image.open(file)
@@ -104,15 +115,10 @@ 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):
-    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)
+
 
 
 
@@ -132,15 +138,31 @@ 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()
+                with gr.Row():
+                    every_n=gr.Number(label = "Every /nth frame", value = 10)
+                stat_box=gr.Textbox(label="Status")
+                with gr.Row():
+                    go_btn_vid=gr.Button("Start")
+                    next_btn=gr.Button("Next")
+                
         gr.Column()            
             #paste_clip = gr.Button("Paste from Clipboard")
     with gr.Row():
         html_out = gr.HTML("""""")
-    vid_url_btn.click(dl,vid_url,inp_vid)
+    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,every_n],[html_out,stat_box,hid_box])
+    vid_url_btn.click(dl,vid_url,[inp_vid,html_out,stat_box,hid_box])
     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,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.queue(concurrency_count=10).launch()
+app.launch()