Update app.py

app.py

@@ -214,7 +214,7 @@ if uploaded_file is not None:
         threshold = st.slider("Threshold", 0.0, 1.0, 0.0, 0.05, key="threshold") #, label_visibility="hidden")
         
     # Decompose button
-    with col5: decompose = st.button('Decompose', key="decompose")
+    # with col5: decompose = st.button('Decompose', key="decompose")
         
     # Make two columns for plots
     _, colA, colB, colC, _ = st.columns([bordersize,1,1,1,bordersize])
@@ -230,23 +230,23 @@ if uploaded_file is not None:
                 
         plot_prediction(y_pred_th)
     
-        if decompose or st.session_state.get("download", False):            
-            image_decomposed = decompose_cavity(y_pred_th)
+        # if decompose or st.session_state.get("download", False):            
+        #     image_decomposed = decompose_cavity(y_pred_th)
     
-            plot_decomposed(image_decomposed)
+        #     plot_decomposed(image_decomposed)
     
-            with col6:
-                st.markdown("<br style='margin:4px 0'>", unsafe_allow_html=True)
-                # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
-                fname = uploaded_file.name.strip(".fits")
+        #     with col6:
+        #         st.markdown("<br style='margin:4px 0'>", unsafe_allow_html=True)
+        #         # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
+        #         fname = uploaded_file.name.strip(".fits")
         
-                # if st.session_state.get("download", False):
+        #         # if st.session_state.get("download", False):
         
-                shutil.make_archive("predictions", 'zip', "predictions")
-                with open('predictions.zip', 'rb') as f:
-                    res = f.read()
+        #         shutil.make_archive("predictions", 'zip', "predictions")
+        #         with open('predictions.zip', 'rb') as f:
+        #             res = f.read()
                 
-                download = st.download_button(label="Download", data=res, key="download", 
-                                                file_name=f'{fname}_{int(scale*128)}.zip', 
-                                                # disabled=st.session_state.get("disabled", True), 
-                                                mime="application/octet-stream")
+        #         download = st.download_button(label="Download", data=res, key="download", 
+        #                                         file_name=f'{fname}_{int(scale*128)}.zip', 
+        #                                         # disabled=st.session_state.get("disabled", True), 
+        #                                         mime="application/octet-stream")