x

app.py

@@ -5,6 +5,9 @@ import glob
 import json
 import requests
 import random
+import io
+import streamlit.components.v1 as components
+
 
 def sample():
     img_set = random.sample(list(set2ids.keys()),1)[0]
@@ -13,6 +16,7 @@ prefix = 'https://raw.githubusercontent.com/BennoKrojer/imagecode-val-set/main/i
 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)
 
@@ -22,9 +26,21 @@ 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))
+# components.html(
+#     """
+# <script>
+# const elements = window.parent.document.querySelectorAll('.stNumberInput div[data-baseweb="input"] > div')
+# console.log(elements)
+# elements[1].style.backgroundColor = 'red'
+# </script>
+# """,
+#     height=0,
+#     width=0,
+# )
+
 
 img = images[index]
-images[index] = ImageOps.expand(Image.open(requests.get(images[index], stream=True).raw),border=10,fill='blue')
+images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True))),border=10,fill='blue')
 
 col1.image(img, use_column_width=True)
 col2.image(images, width=150, caption=list(range(10)))