Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests 
-from bs4 import BeautifulSoup
+#from bs4 import BeautifulSoup
+import cv2
 from google_img_source_search import ReverseImageSearcher
 from PIL import Image
 
@@ -15,7 +16,16 @@ size_js="""
         alert("Original width=" + realWidth + ", " + "Original height=" + realHeight);
     }"""
 
-def process_files(file):
+
+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)
+
+def process_im(file):
     read_file = Image.open(file)
     read_file.save(f"{uid}-tmp.png")
     action_input = f"{uid}-tmp.png"
@@ -48,43 +58,37 @@ def rev_im(image_url):
         </div>"""
     return (gr.HTML(f'<h1>Total Found: {count}</h1><br>{html_out}'))
 
-def find_it(inp):
-    out=""
-    try:
-        #url = f'https://lens.google.com/uploadbyurl?url={inp.strip("")}'
-        url = f'https://tineye.com/search/?url={inp.strip("")}'
-        response = requests.get(url)
-        out=response.text
-        if response.status_code == 200:
-            page_content = response.text
-            
-            soup = BeautifulSoup(page_content, "html.parser")
-            print(soup.prettify())
-            articles = soup.find_all("article") 
-            out = soup.prettify
-            print (articles)
-            
-    except Exception as e:
-        out = e
-        print (e)
-    return (out)
-#https://lens.google.com/uploadbyurl?url=
-#https://tineye.com/search/?url=
-#https://yandex.com/images/search?cbir_id=4330355%2FBhPd4CHqib3nxk9xOdS9pQ7899&rpt=imageview&url=
+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)
+
+
 
 with gr.Blocks() as app:
     with gr.Row():
+        gr.Column()
         with gr.Column():
-            inp_url=gr.Textbox(label="Image URL")
+            
+            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")
+            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:
+                inp_vid=gr.Video(label="Search Video",type='filepath')
             go_btn=gr.Button()
-        
-        with gr.Column():
-            inp_im=gr.Image(type='filepath')
+        gr.Column()            
             #paste_clip = gr.Button("Paste from Clipboard")
     with gr.Row():
         
         html_out = gr.HTML("""""")
-    #paste_clip.click(p_clip,None,inp_im)
+    source_tog.change(shuf,[source_tog],[url_box,im_box,inp_box])
     inp_im.change(process_files,inp_im,[inp_url])
+    inp_vid.change(process_files,inp_vid,[inp_url])
+
     go_btn.click(rev_im,inp_url,[html_out])
 app.launch()