x

app.py

@@ -7,29 +7,22 @@ import requests
 import random
 import io
 
-index = 0
+if 'index' not in st.session_state:
+    st.session_state.index = 0
+
 def set_back():
-    global index
-    index = 0
+    st.session_state.index = 0
 
 st.set_page_config(layout="wide")
 st.markdown("**This is a demo of the *ImageCoDe* dataset. Sample an example description with the left +/- button on the right and compare all the images with the +/- button on the right. If you want to know the groundtruth solution, scroll down to the end of the page!**")
 
 col1, col2 = st.columns(2)
 
-m = st.markdown("""
-<style>
-div.stButton > button:first-child {
-    background-color: rgb(204, 49, 49);
-}
-</style>""", unsafe_allow_html=True)
-
 prefix = 'https://raw.githubusercontent.com/BennoKrojer/imagecode-val-set/main/image-sets-val/'
 set2ids = json.load(open('set2ids.json', 'r'))
 descriptions = json.load(open('valid_list.json', 'r'))
 
-
-example = int()
+index = st.session_state.index
 index = int(col2.number_input('Image Index from 0 to 9', value=0, min_value=0, max_value=9))
 example_idx = int(col1.number_input('Sample an example from the validation set', value=0, min_value=0, max_value=len(descriptions)-1, on_change=set_back))
 img_set, idx, descr = descriptions[example_idx]