Update main app file

app.py

@@ -100,12 +100,15 @@ def main():
         html_content = f.read()
 
     st.markdown(html_content, unsafe_allow_html=True)
+    st.markdown('### Model Architecture')
+    st.image('figures/medsam.png', caption="Segment Anything - MedSAM", width=600)
+    st.markdown('### Demo')
 
     with st.container():
         col1, col2, col3 = st.columns(3)
 
         with col1:
-            option = st.selectbox('Segmentation mode', ('Click'))
+            option = st.selectbox('Segmentation mode', ('Click', 'Box'))
 
         with col2:
             st.write("Show or Hide Mask")
@@ -118,9 +121,9 @@ def main():
         st.write("Upload Image")
         st.file_uploader(label='Upload image',type=['png','jpg','tif'], key='uploaded_image', on_change=image_preprocess_callback, args=(SAM_MODEL,), label_visibility="hidden")
 
-    result_image = None
     canvas_input, canvas_output = st.columns(2)
     if 'image' in st.session_state:
+        result_image = None
         with canvas_input:
             st.write("Select Interest Area/Objects")
             if st.session_state.image is not None:
@@ -132,8 +135,8 @@ def main():
                 st.write("Result")
                 st.image(result_image)
 
-    else:
-        print(f'embedding is empty - {option} - {show_mask} - {radius_width}')
+    # else:
+        # print(f'embedding is empty - {option} - {show_mask} - {radius_width}')
         # if 'image' in st.session_state:
             # if st.session_state.image is None:
                 # st.session_state.clear()