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://omnibus-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://omnibus-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}
Title: {search_item.page_title}
Site: {search_item.page_url}
Img: {search_item.image_url}

""" return (gr.HTML(f'

Total Found: {out_cnt}


{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://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): #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.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}
Title: {search_item.page_title}
Site: {search_item.page_url}
Img: {search_item.image_url}

""" return (gr.HTML(f'

Total Found: {count}


{html_out}')) with gr.Blocks() as app: with gr.Row(): gr.Column() with gr.Column(): 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()