Added images and search function placeholder
Browse files- app.py +22 -2
- img/Cat In Hat.jpg +0 -0
- img/Cat In The Hat.jpg +0 -0
- img/Fox In Socks.jpg +0 -0
- img/Girl.jpg +0 -0
- img/Green Eggs And Ham.jpg +0 -0
- img/Green Eggs with Ham.jpg +0 -0
- img/Grinch.jpg +0 -0
- img/Horton.jpg +0 -0
- Ham.jpg +0 -0
- img/Lorax.jpg +0 -0
- img/One Fish.jpg +0 -0
- Thing2.jpg +0 -0
- img/Turtle.jpg +0 -0
- img/Two Fish.jpg +0 -0
- search.py +26 -0
app.py
CHANGED
|
@@ -1,4 +1,24 @@
|
|
| 1 |
import streamlit as st
|
|
|
|
| 2 |
|
| 3 |
-
|
| 4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
+
from search import *
|
| 3 |
|
| 4 |
+
def display(search_request, search_result) :
|
| 5 |
+
|
| 6 |
+
st.header("\"" +search_request+ "\" reminds me of :")
|
| 7 |
+
i = 0
|
| 8 |
+
for row in range(0, 2):
|
| 9 |
+
for col in st.columns(2) :
|
| 10 |
+
image_name, comment = search_result[i]
|
| 11 |
+
col.image("img/"+image_name)
|
| 12 |
+
col.header(comment)
|
| 13 |
+
i = i + 1
|
| 14 |
+
return
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
|
| 18 |
+
st.title('Find my pic!')
|
| 19 |
+
|
| 20 |
+
search_request = st.text_input('', 'Search ...')
|
| 21 |
+
|
| 22 |
+
if st.button('Find!'):
|
| 23 |
+
search_result = search(search_request)
|
| 24 |
+
display(search_request, search_result)
|
img/Cat In Hat.jpg
ADDED
|
img/Cat In The Hat.jpg
ADDED
|
img/Fox In Socks.jpg
ADDED
|
img/Girl.jpg
ADDED
|
img/Green Eggs And Ham.jpg
ADDED
|
img/Green Eggs with Ham.jpg
ADDED
|
img/Grinch.jpg
ADDED
|
img/Horton.jpg
ADDED
|
Ham.jpg
RENAMED
|
File without changes
|
img/Lorax.jpg
ADDED
|
img/One Fish.jpg
ADDED
|
Thing2.jpg
RENAMED
|
File without changes
|
img/Turtle.jpg
ADDED
|
img/Two Fish.jpg
ADDED
|
search.py
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
import random
|
| 3 |
+
|
| 4 |
+
images = ["Girl.jpg",
|
| 5 |
+
"Cat In Hat.jpg",
|
| 6 |
+
"Cat In The Hat.jpg",
|
| 7 |
+
"Fox In Socks.jpg",
|
| 8 |
+
"Green Eggs And Ham.jpg",
|
| 9 |
+
"Green Eggs with Ham.jpg",
|
| 10 |
+
"Looking at Green Eggs And Ham.jpg",
|
| 11 |
+
"Grinch.jpg",
|
| 12 |
+
"Horton.jpg",
|
| 13 |
+
"Lorax.jpg",
|
| 14 |
+
"Thing1 and Thing2.jpg",
|
| 15 |
+
"Turtle.jpg",
|
| 16 |
+
"One Fish.jpg",
|
| 17 |
+
"Two Fish.jpg"]
|
| 18 |
+
|
| 19 |
+
|
| 20 |
+
|
| 21 |
+
def search(search_prompt : str):
|
| 22 |
+
"""
|
| 23 |
+
Given a search_prompt, return an array of pictures to display
|
| 24 |
+
"""
|
| 25 |
+
|
| 26 |
+
return [ (images[i], images[i].split('.')[0]) for i in random.sample(range(len(images)), 4) ]
|