PicFinder / app.py
osanchik's picture
added search for release 2
63d858e
raw
history blame
927 Bytes
import streamlit as st
import pandas as pd
from main import *
from setup import *
from PIL import Image
def display(search_request, search_result) :
# lorax = Image.open('img/Lorax.jpg')
# print(lorax.width, lorax.height)
# st.image(lorax, width = 250)
st.header("\"" +search_request+ "\" reminds me of :")
i = 0
for row in range(0, 2):
for col in st.columns(2) :
image_name, comment = search_result[i]
col.image("img/"+image_name, width = 300)
col.header(comment)
i = i + 1
return
downlad_images()
st.title('Find my pic!')
search_request = st.text_input('', 'Search ...')
# if st.button('Find Relsease 1!'):
# search_result = search1(search_request)
# display(search_request, search_result)
if st.button('Find Relsease 2!'):
search_result = search2(search_request)
for item in search_result :
st.write(item)