Update app.py

app.py

@@ -121,19 +121,18 @@ SIDEBAR_OPTION_COLOUR_IMAGE = "Choose a colour"
 
 SIDEBAR_OPTIONS = [SIDEBAR_OPTION_DEMO_IMAGE, SIDEBAR_OPTION_UPLOAD_IMAGE,SIDEBAR_OPTION_COLOUR_IMAGE]
 
-clr = [(185, 39, 40),(40, 96, 219),(249, 223, 2)]
-colour = ["Red","Blue","Yellow"]
-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]
 
 app_mode = st.sidebar.selectbox("Please select from the following", SIDEBAR_OPTIONS)
 photos = ["fight.jpg","shaolin-kung-fu.jpg"]
-
+clr = [(185, 39, 40),(40, 96, 219),(249, 223, 2)]
+colour = ["Red","Blue","Yellow"]
 if app_mode == SIDEBAR_OPTION_DEMO_IMAGE and app_mode == SIDEBAR_OPTION_COLOUR_IMAGE:
   st.sidebar.write(" ------ ")
   option = st.sidebar.selectbox('Please select a sample image, then click Magic Time button', photos)
+  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]
   pressed = st.sidebar.button('PoP')
   if pressed:
      st.empty()
@@ -141,9 +140,10 @@ if app_mode == SIDEBAR_OPTION_DEMO_IMAGE and app_mode == SIDEBAR_OPTION_COLOUR_I
      predict(option,option2)
 
 elif app_mode == SIDEBAR_OPTION_UPLOAD_IMAGE and app_mode == SIDEBAR_OPTION_COLOUR_IMAGE:
+  uploaded_file = st.file_uploader("Choose an image...")
+  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]
-  uploaded_file = st.file_uploader("Choose an image...")
   if uploaded_file is not None:
     predict(uploaded_file,option2)