x

app.py

@@ -1,20 +1,27 @@
-from turtle import color
+from turtle import color, onclick
 import streamlit as st
 from PIL import Image, ImageOps
 import glob
 import json
 import requests
+import random
 
+def sample():
+    img_set = random.sample(list(set2ids.keys()),1)[0]
 
 prefix = 'https://raw.githubusercontent.com/BennoKrojer/imagecode-val-set/main/image-sets-val/'
 set2ids = json.load(open('set2ids.json', 'r'))
+img_set = random.sample(list(set2ids.keys()),1)[0]
 
 st.set_page_config(layout="wide")
 col1, col2 = st.columns(2)
 
-images = [prefix+'/MSR-VTT-videoTestVideo_video7016-shot1_2/'+i for i in set2ids['MSR-VTT-videoTestVideo_video7016-shot1_2']]
+images = [prefix+'/'+img_set+'/'+i for i in set2ids['MSR-VTT-videoTestVideo_video7016-shot1_2']]
 
-index= int(col1.number_input('Image Index from 0 to 9', value=0, min_value=0, max_value=len(images)-1), color='red')
+if st.button('Sample set of images with contextual description'):
+    img_set = random.sample(list(set2ids.keys()),1)[0]    
+
+index= int(col1.number_input('Image Index from 0 to 9', value=0, min_value=0, max_value=len(images)-1, color='red'))
 
 img = images[index]
 images[index] = ImageOps.expand(Image.open(requests.get(images[index], stream=True).raw),border=10,fill='blue')