import gradio as gr
import requests 
from bs4 import BeautifulSoup
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 process_files(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}'))

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("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(type='filepath')
    with gr.Row():
        
        html_out = gr.HTML("""""")
    inp_im.change(process_files,inp_im,[inp_url])
    go_btn.click(rev_im,inp_url,[html_out])
app.launch()