Update app.py

app.py

@@ -216,14 +216,20 @@ if uploaded_file is not None:
     fname = uploaded_file.name.strip(".fits")
     os.system(f'mkdir -p {fname}')
     data, wcs = load_file(uploaded_file)
+    
+    MIN = np.min(np.where(data == 0, 1, data))
+    if MIN < 1: data = data / MIN
+    image = np.log10(data+1)
 
 if st.session_state.get("example", True):
     fname = "NGC4649_example"
     os.system(f'mkdir -p {fname}')
     data, wcs = load_file(f"{fname}.fits")
+    image = np.log10(data+1)
 
 if "data" not in locals():
     data = np.zeros((128,128))
+    image = data
         
 # 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])
@@ -256,10 +262,6 @@ _, colA, colB, colC, _ = st.columns([bordersize,1,1,1,bordersize])
 
 if "data" in locals():
     # NORMALIZE IMAGE
-    MIN = np.min(np.where(data == 0, 1, data))
-    if MIN < 1: data = data / MIN
-    
-    image = np.log10(data+1)
     plot_image(image, scale)
     
     if detect or threshold or st.session_state.get("decompose", False):