import streamlit as st

import sys

sys.path.insert(0, "image_search_engine")


from image_search_engine import utils
from image_search_engine.product_image_search import JumiaProductSearch
from PIL import Image, ImageOps

import requests
from streamlit_image_select import image_select

jumia = JumiaProductSearch()


def get_search_results(query):
    res = jumia.search(query, 9)

    images, names, urls = [], [], []

    for i, record in enumerate(res["matches"]):
        metadata = record["metadata"]
        images.append(metadata["product_image_url"])
        names.append(metadata["product_name"])
        urls.append(metadata["product_url"])

    return images, names, urls


banner_img = Image.open(utils.PACKAGE_DIR.parent / "jumia_lens.png")
st.image(banner_img)

st.markdown("#### Currently limited to the following product categories; wrist watches, backpacks, usb drives/hubs, mouse, ergonomic/office chairs, earpiece/headsets")

input_options = st.radio("Select Input Option", ("image upload", "use example images"))

img = None

if input_options == "image upload":
    img = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])


else:
    with st.expander(label="Choose sample image", expanded=False):
        img = image_select(
            label="Use example image",
            images=[
                "https://ng.jumia.is/unsafe/fit-in/500x500/filters:fill(white)/product/88/3305912/1.jpg?5807",
                "https://ng.jumia.is/unsafe/fit-in/500x500/filters:fill(white)/product/74/0464341/1.jpg?7325",
                "https://watchlocker.ng/wp-content/uploads/2021/04/JY8085-14H.jpg",
                "https://www-konga-com-res.cloudinary.com/w_auto,f_auto,fl_lossy,dpr_auto,q_auto/media/catalog/product/M/L/196920_1641394875.jpg",
                "https://www-konga-com-res.cloudinary.com/w_auto,f_auto,fl_lossy,dpr_auto,q_auto/media/catalog/product/I/K/154983_1595624114.jpg",
                "https://ng.jumia.is/unsafe/fit-in/500x500/filters:fill(white)/product/73/3254702/1.jpg?5592",
                "https://store.storeimages.cdn-apple.com/4668/as-images.apple.com/is/MKUQ3_VW_34FR+watch-44-alum-midnight-cell-se_VW_34FR_WF_CO_GEO_AE?wid=1400&hei=1400&trim=1%2C0&fmt=p-jpg&qlt=95&.v=1683237043713",
                "https://ng.jumia.is/unsafe/fit-in/500x500/filters:fill(white)/product/71/6579011/1.jpg?5730",
            ],
        )


if img:
    if isinstance(img, str):
        image = Image.open(requests.get(img, stream=True).raw)
    else:
        image = Image.open(img)

    image = ImageOps.exif_transpose(image)

    with st.columns(3)[1]:
        st.markdown("###   Query Image.")
        st.image(image)
    st.markdown(" ")
    st.markdown("### Search results.")
    st.markdown(" ")
    n = 3
    product_images, product_names, product_urls = get_search_results(image)

    for i, col in enumerate(st.columns(n)):
        positions = (i, i + 3, i + 6)
        names = [product_names[i] for i in positions]
        images = [product_images[i] for i in positions]
        urls = [product_urls[i] for i in positions]

        with col:
            st.write(names[0])
            st.image(images[0])
            st.write(urls[0])

            st.write(names[1])
            st.image(images[1])
            st.write(urls[1])

            st.write(names[2])
            st.image(images[2])
            st.write(urls[2])

            # st.write(names[3])
            # st.image(images[3])
            # st.write(urls[3])