Update app.py

app.py

@@ -195,9 +195,8 @@ with col_1:
     uploaded_file = st.file_uploader("Choose a FITS file", type=['fits'], on_change=reset_all)
 
 with col_2:
-    st.markdown("<br style='margin:15px 0'>", unsafe_allow_html=True)
-    if st.button("Example"):
-        st.session_state['example'] = True
+    st.markdown("<br style='margin:14px 0'>", unsafe_allow_html=True)
+    if st.button("Example"): st.session_state['example'] = True
     
     # with col_2:
     #     st.markdown("### Examples")
@@ -263,34 +262,33 @@ with col5: decompose = st.button('Decompose', key="decompose")
 # Make two columns for plots
 _, colA, colB, colC, _ = st.columns([bordersize,1,1,1,bordersize])
 
-if "data" in locals():
-    # NORMALIZE IMAGE
-    plot_image(data, scale)
-    
-    if detect or threshold or st.session_state.get("decompose", False):
+# NORMALIZE IMAGE
+plot_image(data, scale)
 
-        y_pred, wcs = cut_n_predict(data, wcs, scale)
-        
-        y_pred_th = np.where(y_pred > threshold, y_pred, 0)
-                
-        plot_prediction(y_pred_th)
+if detect or threshold or st.session_state.get("decompose", False):
 
-        if decompose or st.session_state.get("download", False):            
-            image_decomposed = decompose_cavity(y_pred_th, fname)
+    y_pred, wcs = cut_n_predict(data, wcs, scale)
+    
+    y_pred_th = np.where(y_pred > threshold, y_pred, 0)
+            
+    plot_prediction(y_pred_th)
 
-            plot_decomposed(image_decomposed)
+    if decompose or st.session_state.get("download", False):            
+        image_decomposed = decompose_cavity(y_pred_th, fname)
 
-            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)
-        
-                # if st.session_state.get("download", False):
-        
-                shutil.make_archive(fname, 'zip', fname)
-                with open(f"{fname}.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")
+        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)
+    
+            # if st.session_state.get("download", False):
+    
+            shutil.make_archive(fname, 'zip', fname)
+            with open(f"{fname}.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")