Update app.py

app.py

@@ -119,7 +119,7 @@ SIDEBAR_OPTION_DEMO_IMAGE = "Select a Demo Image"
 SIDEBAR_OPTION_UPLOAD_IMAGE = "Upload an Image"
 SIDEBAR_OPTION_COLOUR_IMAGE = "Choose a colour"
 
-SIDEBAR_OPTIONS = [SIDEBAR_OPTION_DEMO_IMAGE, SIDEBAR_OPTION_UPLOAD_IMAGE,SIDEBAR_OPTION_COLOUR_IMAGE]
+SIDEBAR_OPTIONS = [SIDEBAR_OPTION_DEMO_IMAGE, SIDEBAR_OPTION_UPLOAD_IMAGE]
 
 
 app_mode = st.sidebar.selectbox("Please select from the following", SIDEBAR_OPTIONS)
@@ -133,7 +133,7 @@ if app_mode == SIDEBAR_OPTION_DEMO_IMAGE:
   colour[0] = clr[0]
   colour[1] = clr[1]
   colour[2] = clr[2]
-if app_mode == SIDEBAR_OPTION_DEMO_IMAGE:
+  if app_mode == SIDEBAR_OPTION_DEMO_IMAGE:
   st.sidebar.write(" ------ ")
   option = st.sidebar.selectbox('Please select a sample image, then click Magic Time button', photos)
   pressed = st.sidebar.button('PoP')
@@ -141,8 +141,14 @@ if app_mode == SIDEBAR_OPTION_DEMO_IMAGE:
      st.empty()
      st.sidebar.write('Please wait for the magic to happen! This may take up to a minute.')
      predict(option,option2)
-
-elif app_mode == SIDEBAR_OPTION_UPLOAD_IMAGE:
+else:
+  if app_mode == SIDEBAR_OPTION_DEMO_IMAGE:
+  st.sidebar.write(" ------ ")
+  option2 = st.sidebar.selectbox('Please select a sample image, then click Magic Time button', colour)
+  colour[0] = clr[0]
+  colour[1] = clr[1]
+  colour[2] = clr[2]
+  if app_mode == SIDEBAR_OPTION_UPLOAD_IMAGE:
   uploaded_file = st.file_uploader("Choose an image...")
   if uploaded_file is not None:
     predict(uploaded_file,option2)