Upload 4 files

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,12 @@
+---
+title: Reverse Image
+emoji: 📈
+colorFrom: blue
+colorTo: blue
+sdk: gradio
+sdk_version: 3.47.1
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,181 @@
+import gradio as gr
+import requests 
+import yt_dlp
+import cv2
+from google_img_source_search import ReverseImageSearcher
+from PIL import Image
+import os 
+import uuid
+uid=uuid.uuid4()
+size_js="""
+    function imgSize(){
+        var myImg = document.getElementsByClassName("my_im");
+        var realWidth = myImg.naturalWidth;
+        var realHeight = myImg.naturalHeight;
+        alert("Original width=" + realWidth + ", " + "Original height=" + realHeight);
+    }"""
+
+def dl(inp):
+    out = None
+    out_file=[]
+    try:
+        inp_out=inp.replace("https://","")
+        inp_out=inp_out.replace("/","_").replace(".","_").replace("=","_").replace("?","_")
+        if "twitter" in inp:
+            os.system(f'yt-dlp "{inp}" --extractor-arg "twitter:api=syndication" --trim-filenames 160 -o "{uid}/{inp_out}.mp4" -S res,mp4 --recode mp4')  
+        else:
+            os.system(f'yt-dlp "{inp}" --trim-filenames 160 -o "{uid}/{inp_out}.mp4" -S res,mp4 --recode mp4')          
+        
+        #os.system(f'yt-dlp "{inp}" --trim-filenames 160 -o "{inp_out}.mp4" -S res,mp4 --recode mp4')  
+        out = f"{uid}/{inp_out}.mp4"
+        #out_ap = os.path.abspath(out_f)
+        #out = f'https://Jonny001-reverse-image.hf.space/file={out_ap}'
+        print (out)
+    except Exception as e:
+        print (e)
+        #out = f'{e}'
+    return out,gr.HTML(""),"",""
+
+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(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://Jonny001-reverse-image.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.'),"","")
+
+def process_im(file,url):
+    if not url.startswith("https://Jonny001"):
+        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://Jonny001-reverse-image.hf.space/file={out}'
+        return (out_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://Jonny001-reverse-image.hf.space/file={out}'    
+    rev_img_searcher = ReverseImageSearcher()
+    res = rev_img_searcher.search(out_url)
+    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}'))
+
+
+
+
+
+with gr.Blocks() as app:
+    with gr.Row():
+        gr.Column()
+        with gr.Column():
+            gr.HTML("""<center><h1>Reverse Image and Video Search</h1></center>""")
+            source_tog=gr.Radio(choices=["Image","Video"],value="Image")
+            with gr.Box(visible=True) as im_box:
+                inp_url=gr.Textbox(label="Image URL")
+                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:
+                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(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("""""")
+    with gr.Row(visible=False):
+        hid_box=gr.Textbox()
+    def shuf(tog):
+        if tog == "Image":
+            return gr.update(visible=True),gr.update(visible=False)
+        if tog == "Video":
+            return gr.update(visible=False),gr.update(visible=True)    
+    def load_image(url):
+        return url
+    im_load = 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_load = vid_url_btn.click(dl,vid_url,[inp_vid,html_out,stat_box,hid_box])
+    #inp_im.change(process_im,[inp_im,inp_url],[inp_url])
+    vid_proc = go_btn_vid.click(process_vid,[inp_vid,hid_box,every_n],[html_out,stat_box,hid_box])
+    im_proc = go_btn_im.click(rev_im,inp_im,[html_out])
+    source_tog.change(shuf,[source_tog],[im_box,vid_box],cancels=[vid_proc,im_proc,im_load,vid_load])
+
+    #go_btn_url.click(rev_im,inp_url,[html_out])
+app.queue(concurrency_count=20).launch()

requirements.txt

@@ -0,0 +1,3 @@
+yt-dlp
+opencv-python
+google-image-source-search