osanchik commited on
Commit
8df0b72
·
1 Parent(s): f7b6c7e

Added images and search function placeholder

Browse files
app.py CHANGED
@@ -1,4 +1,24 @@
1
  import streamlit as st
 
2
 
3
- x = st.slider('Select a value')
4
- st.write(x, 'squared is', x * x)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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) ]