Update app.py

app.py

@@ -20,11 +20,13 @@ st.set_option('deprecation.showPyplotGlobalUse', False)
 
 st.set_page_config(page_title="Cavity Detection Tool", layout="wide")
 
-st.title("Cavity Detection Tool")
+# st.title("Cavity Detection Tool")
 
 _, col, _ = st.columns([1, 3, 1])
 
 with col:
+    st.markdown("# Cavity Detection Tool")
+    
     st.markdown("Cavity Detection Tool (CADET) is a machine learning pipeline trained to detect X-ray cavities from noisy Chandra images of early-type galaxies. To use this tool: upload your image, select the scale of interest, and make a prediction! If you use this tool for your research, please cite [Plšek et al. 2023](https://arxiv.org/abs/2304.05457)")
     
     st.markdown("Input images should be centered 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)).")
@@ -106,23 +108,23 @@ if uploaded_file is not None:
         data = hdul[0].data
         wcs = WCS(hdul[0].header)
     
-    # # Make four columns for buttons
-    # col1, col2, col3, col4 = st.columns(4)
-    # col1.subheader("Input image")
-    # col3.subheader("Prediction")
+    # Make four columns for buttons
+    _, col1, col2, col3, col4, _ = st.columns([1,0.75,0.75,0.75,0.75,,1])
+    col1.subheader("Input image")
+    col3.subheader("Prediction")
     
-    # with col1:
-    #     st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
-    #     max_scale = int(data.shape[0] // 128)
-    #     # scale = int(st.selectbox('Scale:',[i+1 for i in range(max_scale)], label_visibility="hidden"))
-    #     scale = int(st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden"))
-    #     scale = scale.split("x")[0] // 128
+    with col1:
+        st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
+        max_scale = int(data.shape[0] // 128)
+        # scale = int(st.selectbox('Scale:',[i+1 for i in range(max_scale)], label_visibility="hidden"))
+        scale = int(st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden"))
+        scale = scale.split("x")[0] // 128
     
-    # with col2:
-    #     detect = st.button('Detect cavities')
+    with col2:
+        detect = st.button('Detect cavities')
 
-    # with col3:
-    #     decompose = st.button('Docompose cavities')
+    with col3:
+        decompose = st.button('Docompose cavities')
         
     # # Make two columns for plots
     # colA, colB = st.columns(2)