Update app.py

app.py

@@ -130,7 +130,7 @@ def decompose_cavity(pred, th2=0.7, amin=10):
         ccd.write(f"predictions/predicted_{i+1}.fits", overwrite=True)
         image_decomposed += (i+1) * np.where(cav > 0, 1, 0)
 
-    shutil.make_archive("predictions", 'zip', "predictions")
+    # shutil.make_archive("predictions", 'zip', "predictions")
     
     return image_decomposed
 
@@ -166,13 +166,14 @@ if uploaded_file is not None:
     col6.subheader("")
     
     with col1:
-        st.markdown("""<style>[data-baseweb="select"] {margin-top: -26px;}</style>""", unsafe_allow_html=True)
+        # 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)
         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 = int(scale.split("x")[0]) // 128
 
     # Detect button
-    with col3: detect = st.button('Detect')
+    with col3: detect = st.button('Detect', key="detect")
 
     # Threshold slider
     with col4:
@@ -184,16 +185,17 @@ if uploaded_file is not None:
     with col5: decompose = st.button('Decompose', key="decompose")
 
     with col6:
-        st.markdown("")
-        st.markdown("")
+        st.markdown("<br><br>")
         # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
         fname = uploaded_file.name.strip(".fits")
-        shutil.make_archive("predictions", 'zip', "predictions")
-        with open('predictions.zip', 'rb') as f:
-            res = f.read()
 
         if st.session_state.get("decompose", False):
             st.session_state.disabled = False
+
+        shutil.make_archive("predictions", 'zip', "predictions")
+        with open('predictions.zip', 'rb') as f:
+            res = f.read()
+        
         download = st.download_button(label="Download", data=res, 
                                       file_name=f'{fname}_{int(scale*128)}.zip', 
                                       disabled=st.session_state.get("disabled", True),