Spaces:
Running
Running
| 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("?","_") | |
| os.system(f'yt-dlp "{inp}" --trim-filenames 160 -o "{inp_out}.mp4" -S res,mp4 --recode mp4') | |
| out = f"{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-dev.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): | |
| 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): | |
| #image_url = 'https://i.pinimg.com/originals/c4/50/35/c450352ac6ea8645ead206721673e8fb.png' | |
| out_list = [] | |
| out_im = [] | |
| html_out = """""" | |
| rev_img_searcher = ReverseImageSearcher() | |
| res = rev_img_searcher.search(image_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(): | |
| 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') | |
| 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 == "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) | |
| 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]) | |
| 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.launch() |