Update app.py

app.py

@@ -45,13 +45,13 @@ def process_vid(file,cur_frame,every_n):
     try:
         for i in range(start_frame, frame_count-1):
             if count == int(every_n):
-                count = 0
+                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}'
+                out_url = f'https://omnibus-reverse-image-dev.hf.space/file={out}'
                 print(out)
                 res = rev_img_searcher.search(out_url)
                 #print (res)
@@ -84,21 +84,28 @@ def process_vid(file,cur_frame,every_n):
         return (gr.HTML(f'{e}'),"","")
     return (gr.HTML('No frame matches found.'),"","")
 
-def process_im(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)
+def process_im(file,url):
+    if not url.startswith("https://omnibus"):
+        return url
+    else:
+        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)
 
-def rev_im(image_url):
+def rev_im(image):
     #image_url = 'https://i.pinimg.com/originals/c4/50/35/c450352ac6ea8645ead206721673e8fb.png'
     out_list = []
     out_im = []
     html_out = """"""
+    image=cv2.imread(image)
+    cv2.imwrite(f"{uid}-im_tmp.png", image)
+    out = os.path.abspath(f"{uid}-im_tmp.png")
+    out_url = f'https://omnibus-reverse-image-dev.hf.space/file={out}'    
     rev_img_searcher = ReverseImageSearcher()
-    res = rev_img_searcher.search(image_url)
+    res = rev_img_searcher.search(out_url)
     count = 0
     for search_item in res:
         count+=1
@@ -127,11 +134,10 @@ with gr.Blocks() as app:
         gr.Column()
         with gr.Column():
             
-            source_tog=gr.Radio(choices=["URL","Image","Video"],value="URL")
-            with gr.Box(visible=True) as url_box:
+            source_tog=gr.Radio(choices=["Image","Video"],value="Image")
+            with gr.Box(visible=True) as im_box:
                 inp_url=gr.Textbox(label="Image URL")
-                go_btn_url=gr.Button()
-            with gr.Box(visible=False) as im_box:
+                load_im_btn=gr.Button("Load Image")
                 inp_im=gr.Image(label="Search Image",type='filepath')
                 go_btn_im=gr.Button()
             with gr.Box(visible=False) as vid_box:
@@ -152,17 +158,18 @@ with gr.Blocks() as app:
     with gr.Row(visible=False):
         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)
+            return gr.update(visible=True),gr.update(visible=False)
         if tog == "Video":
-            return gr.update(visible=False),gr.update(visible=False),gr.update(visible=True)        
+            return gr.update(visible=False),gr.update(visible=True)    
+    def load_image(url):
+        return url
+    load_im_btn.click(load_image,inp_url,inp_im)
     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])
+    source_tog.change(shuf,[source_tog],[im_box,vid_box])
+    #inp_im.change(process_im,[inp_im,inp_url],[inp_url])
     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()
+    go_btn_im.click(rev_im,inp_im,[html_out])
+    #go_btn_url.click(rev_im,inp_url,[html_out])
+app.launch()