Updated app.py

app.py

@@ -40,18 +40,31 @@ def upload():
 
 def main(options):
     col1,col2=st.columns(2)
+    image=None
+    original_image=None
     
-   
+    if 'c01' not in st.session_state:
+        st.session_state.c01=False
+    c01_state=st.session_state.c01
+    c01_state=st.sidebar.checkbox('Take a picture for prediction',value=c01_state)
+
+    if 'c02' not in st.session_state:
+        st.session_state.c02=False
+    c02_state=st.session_state.c02
+    c02_state=st.sidebar.checkbox('Detect',value=c02_state)
     
     
     with col1:
-        if st.checkbox('Take a picture for prediction'):
-            image=0
-            original_image=0
+        if c01_state:
+           
             with col1:
                 image, original_image= upload()
-                if original_image is not None and original_image is not None and st.button('Prediction'):  # Check if original_image is not None
+                if original_image is not None and original_image is not None and c02_state: 
+                    
+                    # Check if original_image is not None
                     pass
+                    st.session_state.c02=c02_state
+            st.session_state.c01=c01_state
     with col2:
         st.warning('Wait for few seconds!!')
         progress_bar = st.progress(0.0)