Spaces:
Sleeping
Sleeping
File size: 6,743 Bytes
aead09a 1ce9878 d82da04 7716718 2255cf1 522ecb9 e8f283e 57e4f6c a1323c8 1ce9878 175f376 1ce9878 175f376 1ce9878 d82da04 1be6131 599424e 7c28c3a 229466f 61acde4 3d09b7a 61acde4 3d09b7a 599424e 3d09b7a 1be6131 e72db24 1be6131 e72db24 1be6131 e72db24 1be6131 cb0efb1 e5deecc 599424e e5deecc d82da04 2f9a5fe e8f283e 522ecb9 e8f283e c558940 458c1fb 283940a 7716718 0b4ddd5 7716718 0b4ddd5 458c1fb 09604de 283940a 0745a74 57e4f6c 283940a 3d09b7a 0b4ddd5 aead09a 1be6131 d82da04 aead09a d5ef3b7 d82da04 d5ef3b7 d82da04 599424e d82da04 599424e 60fae74 fc34e82 175f376 1be6131 b6a1d3a 61acde4 3d09b7a 61acde4 3d09b7a d82da04 2255cf1 458c1fb 42a1ed0 3d09b7a 1be6131 a9c2669 fc34e82 06d0b82 37ab46f 1be6131 599424e aead09a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 |
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
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=True):
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)
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() |