Spaces:
Runtime error
Runtime error
File size: 3,479 Bytes
85e80c1 246152c 85e80c1 246152c 85e80c1 246152c 85e80c1 0eeef10 246152c 0eeef10 85e80c1 |
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 |
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])
|