Spaces:
Running
Running
import gradio as gr | |
import requests | |
from bs4 import BeautifulSoup | |
from google_img_source_search import ReverseImageSearcher | |
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 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}')) | |
def find_it(inp): | |
out="" | |
try: | |
#url = f'https://lens.google.com/uploadbyurl?url={inp.strip("")}' | |
url = f'https://tineye.com/search/?url={inp.strip("")}' | |
response = requests.get(url) | |
out=response.text | |
if response.status_code == 200: | |
page_content = response.text | |
soup = BeautifulSoup(page_content, "html.parser") | |
print(soup.prettify()) | |
#articles = soup.find_all("div", class_="SoaBEf") | |
#articles = soup.find_all("a") | |
#articles = soup.find_all("article", class_="MQsxIb xTewfe R7GTQ keNKEd j7vNaf Cc0Z5d VkAdve GU7x0c JMJvke q4atFc") | |
articles = soup.find_all("article") | |
out = soup.prettify | |
print (articles) | |
except Exception as e: | |
out = e | |
print (e) | |
return (out) | |
#https://lens.google.com/uploadbyurl?url= | |
#https://tineye.com/search/?url= | |
#https://yandex.com/images/search?cbir_id=4330355%2FBhPd4CHqib3nxk9xOdS9pQ7899&rpt=imageview&url= | |
with gr.Blocks() as app: | |
with gr.Row(): | |
with gr.Column(): | |
inp_url=gr.Textbox(label="Image URL") | |
go_btn=gr.Button() | |
inp_im=gr.Image() | |
with gr.Row(): | |
html_out = gr.HTML("""""") | |
go_btn.click(rev_im,inp_url,[html_out]) | |
app.launch() |