Update app.py

app.py

@@ -148,9 +148,18 @@ with col:
     st.markdown("To use this tool: upload your image, select the scale of interest, and make a prediction!")
     st.markdown("Input images should be centred at the centre of the galaxy and point sources should be filled with surrounding background ([dmfilth](https://cxc.cfa.harvard.edu/ciao/ahelp/dmfilth.html)).")
     st.markdown("If you use this tool for your research, please cite [Plšek et al. 2023](https://arxiv.org/abs/2304.05457)")
-    
+
+_, col_1, col_2, _ = st.columns([bordersize, 2.5, 0.5, bordersize])
+
+with col1:
     # Create file uploader widget
     uploaded_file = st.file_uploader("Choose a FITS file", type=['fits'])
+
+with col2:
+    example = st.button("Example")
+
+def change_state(b):
+    st.session_state.value = 0.0
     
 # If file is uploaded, read in the data and plot it
 if uploaded_file is not None:
@@ -168,9 +177,9 @@ if uploaded_file is not None:
     with col1:
         # st.markdown("""<style>[data-baseweb="select"] {margin-top: -26px;}</style>""", unsafe_allow_html=True)
         # st.markdown("<style>{margin-top: -26px;}</style>", unsafe_allow_html=True)
-        st.markdown("<br style='margin:-14px 0'>", unsafe_allow_html=True)
+        st.markdown("<br style='margin:-44px 0'>", unsafe_allow_html=True)
         max_scale = int(data.shape[0] // 128)
-        scale = st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden")
+        scale = st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden", on_change=change_state)
         scale = int(scale.split("x")[0]) // 128
 
     # Detect button
@@ -180,7 +189,7 @@ if uploaded_file is not None:
     with col4:
         st.markdown("")
         # st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
-        threshold = st.slider("Threshold", 0.0, 1.0, 0.0, 0.05) #, label_visibility="hidden")
+        threshold = st.slider("Threshold", 0.0, 1.0, st.session_state.value, 0.05) #, label_visibility="hidden")
         
     # Decompose button
     with col5: decompose = st.button('Decompose', key="decompose")