Update app.py

app.py

@@ -141,7 +141,7 @@ def decompose_cavity(pred, th2=0.7, amin=6):
     
     return image_decomposed
 
-@st.cache
+# @st.cache
 def load_file(fname):
     with fits.open(fname) as hdul:
         data = hdul[0].data
@@ -190,9 +190,6 @@ with col:
 
 if uploaded_file is not None:
     data, wcs = load_file(uploaded_file)
-    # with fits.open(uploaded_file) as hdul:
-    #     data = hdul[0].data
-    #     wcs = WCS(hdul[0].header)
     
     # Make six columns for buttons
     _, col1, col2, col3, col4, col5, col6, _ = st.columns([bordersize,0.5,0.5,0.5,0.5,0.5,0.5,bordersize])
@@ -207,17 +204,17 @@ if uploaded_file is not None:
         scale = st.selectbox('Scale:',[f"{(i+1)*128}x{(i+1)*128}" for i in range(max_scale)], label_visibility="hidden") #, on_change=change_scale)
         scale = int(scale.split("x")[0]) // 128
     
-    # # Detect button
-    # with col3: detect = st.button('Detect', key="detect")
+    # Detect button
+    with col3: detect = st.button('Detect', key="detect")
     
-    # # Threshold slider
-    # with col4:
-    #     st.markdown("")
-    #     # st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
-    #     threshold = st.slider("Threshold", 0.0, 1.0, 0.0, 0.05, key="threshold") #, label_visibility="hidden")
+    # Threshold slider
+    with col4:
+        st.markdown("")
+        # st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
+        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")
+    # Decompose button
+    with col5: decompose = st.button('Decompose', key="decompose")
         
     # Make two columns for plots
     _, colA, colB, colC, _ = st.columns([bordersize,1,1,1,bordersize])
@@ -225,31 +222,31 @@ if uploaded_file is not None:
     image = np.log10(data+1)
     plot_image(image, scale)
     
-    # if detect or threshold:
-    # # if st.session_state.get("detect", True):
-    #     y_pred, wcs = cut_n_predict(data, wcs, scale)
+    if detect or threshold:
+    # if st.session_state.get("detect", True):
+        y_pred, wcs = cut_n_predict(data, wcs, scale)
         
-    #     y_pred_th = np.where(y_pred > threshold, y_pred, 0)
+        y_pred_th = np.where(y_pred > threshold, y_pred, 0)
                 
-    #     plot_prediction(y_pred_th)
+        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")