Update app.py

app.py

@@ -81,7 +81,7 @@ if uploaded_file is not None:
         data = hdul[0].data
         wcs = WCS(hdul[0].header)
     
-    # Make two columns
+    # Make four columns for buttons
     col1, col2, col3, col4 = st.columns(4)
     col1.subheader("Input image")
     col3.subheader("Prediction")
@@ -91,16 +91,14 @@ if uploaded_file is not None:
         smooth = st.button("Smooth")
     
     with col2:
-        # st.markdown("""<style>[data-baseweb="select"] {margin-top: -52px;}</style>""", unsafe_allow_html=True)
-        st.markdown("""<style>[data-baseweb="select"] {margin-top: 22px;}</style>""", unsafe_allow_html=True)
-        
+        st.markdown("""<style>[data-baseweb="select"] {margin-top: 17px;}</style>""", unsafe_allow_html=True)
         max_scale = int(data.shape[0] // 128)
-        # scale = st.slider("Scale", 1, max_scale, 1, 1)
         scale = int(st.selectbox('Scale:',[i+1 for i in range(max_scale)], label_visibility="hidden"))
     
     with col3:
         detect = st.button('Detect cavities')
 
+    # Make two columns for plots
     colA, colB = st.columns(2)
 
     image = np.log10(data+1)
@@ -126,9 +124,10 @@ if uploaded_file is not None:
         ccd = CCDData(y_pred, unit="adu", wcs=wcs)
         ccd.write("predicted.fits", overwrite=True)
         with open('predicted.fits', 'rb') as f:
-            data = f.read()
+            res = f.read()
         
         with col4:
-            st.markdown("""<style>[data-baseweb="select"] {margin-top: 32px;}</style>""", unsafe_allow_html=True)
+            pass
+            # st.markdown("""<style>[data-baseweb="select"] {margin-top: 32px;}</style>""", unsafe_allow_html=True)
         #     # download = st.button('Download')
-        #     download = st.download_button(label="Download", data=data, file_name="predicted.fits", mime="application/octet-stream")
+        #     download = st.download_button(label="Download", data=res, file_name="predicted.fits", mime="application/octet-stream")