Reverse-Image / app.py
Omnibus's picture
Update app.py
efa29a0 verified
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}
<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://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}
<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()