Update app.py

app.py

@@ -193,7 +193,7 @@ with col_1:
 
 with col_2:
     st.markdown("<br style='margin:15px 0'>", unsafe_allow_html=True)
-    example = st.button("Example")
+    example = st.button("Example", key="example")
     st.markdown('<style>div.row-widget.stButton > div{margin-left: auto;}</style>', unsafe_allow_html=True)
     
     # with col_2:
@@ -215,20 +215,14 @@ if uploaded_file is not None:
     fname = uploaded_file.name.strip(".fits")
     os.system(f'mkdir -p {fname}')
     data, wcs = load_file(uploaded_file)
-    st.session_state['loaded'] = True
-    st.session_state['example'] = False
 
 if example:
     fname = "NGC4649_example"
     os.system(f'mkdir -p {fname}')
     data, wcs = load_file(f"{fname}.fits")
-    st.session_state['loaded'] = False
-    st.session_state['example'] = True
 
 if "data" not in locals():
     data = np.zeros((128,128))
-    st.session_state['loaded'] = False
-    st.session_state['example'] = False
         
 # 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])
@@ -259,7 +253,7 @@ with col5: decompose = st.button('Decompose', key="decompose")
 # Make two columns for plots
 _, colA, colB, colC, _ = st.columns([bordersize,1,1,1,bordersize])
 
-if st.session_state['loaded'].get("decompose", True) or st.session_state['example'].get("decompose", True):
+if (uploaded_file is not None) or st.session_state.get("example", False):
     # NORMALIZE IMAGE
     MIN = np.min(np.where(data == 0, 1, data))
     if MIN < 1: data = data / MIN