Update app.py

app.py

@@ -22,7 +22,8 @@ st.set_page_config(page_title="Cavity Detection Tool", layout="wide")
 
 # st.title("Cavity Detection Tool")
 
-_, col, _ = st.columns([1, 3, 1])
+bordersize = 0.8
+_, col, _ = st.columns([bordersize, 3, bordersize])
 
 with col:
     st.markdown("# Cavity Detection Tool")
@@ -109,32 +110,33 @@ if uploaded_file is not None:
         wcs = WCS(hdul[0].header)
     
     # Make four columns for buttons
-    _, col1, col2, col3, col4, _ = st.columns([1,0.75,0.75,0.75,0.75,1])
+    _, col1, col2, col3, col4, _ = st.columns([bordersize,0.75,0.75,0.75,0.75,bordersize])
     col1.subheader("Input image")
     col3.subheader("Prediction")
     
     with col1:
-        st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
+        st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</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 = st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden")
         scale = int(scale.split("x")[0]) // 128
     
     with col2:
+        st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
         detect = st.button('Detect cavities')
 
     with col3:
         decompose = st.button('Docompose cavities')
         
     # Make two columns for plots
-    _, colA, colB, colC, _ = st.columns([1,1,1,1,1])
+    _, colA, colB, colC, _ = st.columns([bordersize,1,1,1,bordersize])
 
     image = np.log10(data+1)
     plot_image(image, scale)
     
-    # if detect:
-    #     data, wcs = cut(data, wcs, scale=scale)
-    #     image = np.log10(data+1)
+    if detect:
+        data, wcs = cut(data, wcs, scale=scale)
+        image = np.log10(data+1)
         
     #     y_pred = 0
     #     for j in [0,1,2,3]: