Update app.py

app.py

@@ -18,12 +18,41 @@ size_js="""
 
 
 def process_vid(file):
-    read_file = Image.open(file)
-    read_file.save(f"{uid}-tmp.png")
-    action_input = f"{uid}-tmp.png"
-    out = os.path.abspath(action_input)
-    out_url = f'https://omnibus-reverse-image.hf.space/file={out}'
-    return (out_url)
+    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()
+    try:
+        for i in frame_count-1:
+            capture.set(cv2.CAP_PROP_POS_FRAMES, i)
+            ret, frame_f = capture.read(i)
+            frame_f.write(f"{uid}-vid_tmp.png")
+            out = os.path.abspath(f"{uid}-vid_tmp.png")
+
+            res = rev_img_searcher.search(out)
+            if len(res) > 0:
+                count = 0
+                for search_item in res:
+                    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
+    except Exception as e:
+        return (gr.HTML(f'{e}'))
+    return (gr.HTML('No frame matches found.'))
 
 def process_im(file):
     read_file = Image.open(file)
@@ -76,11 +105,13 @@ with gr.Blocks() as app:
             source_tog=gr.Radio(choices=["URL","Image","Video"],value="URL")
             with gr.Box(visible=True) as url_box:
                 inp_url=gr.Textbox(label="Image URL")
+                go_btn_url=gr.Button()
             with gr.Box(visible=False) as im_box:
                 inp_im=gr.Image(label="Search Image",type='filepath')
-            with gr.Box(visible=False) as vid_box:
+                go_btn_im=gr.Button()
+        with gr.Box(visible=False) as vid_box:
                 inp_vid=gr.Video(label="Search Video",type='filepath')
-            go_btn=gr.Button()
+                go_btn_vid=gr.Button()
         gr.Column()            
             #paste_clip = gr.Button("Paste from Clipboard")
     with gr.Row():
@@ -88,7 +119,7 @@ with gr.Blocks() as app:
         html_out = gr.HTML("""""")
     source_tog.change(shuf,[source_tog],[url_box,im_box,vid_box])
     inp_im.change(process_im,inp_im,[inp_url])
-    inp_vid.change(process_vid,inp_vid,[inp_url])
-
-    go_btn.click(rev_im,inp_url,[html_out])
+    go_btn_vid.click(process_vid,inp_vid,[url_box,im_box,vid_box])
+    go_btn_im.click(rev_im,inp_url,[html_out])
+    go_btn_url.click(rev_im,inp_url,[html_out])
 app.launch()