Update app.py

app.py

@@ -42,20 +42,25 @@ input_prompt = textwrap.dedent("""
     """)
 
 # Display sample input images
-sample_images_folder = pathlib.Path(__file__).parent / "examples"
+sample_images_folder = pathlib.Path(__file__).parent / "sample_images"
 sample_images = [sample_image.name for sample_image in sample_images_folder.glob("*.jpg") if sample_image.is_file()]
 
-sample_image_uploaded = st.sidebar.file_uploader("Choose a sample image", type=["jpg", "jpeg"], options=sample_images)
+sample_image_selected = st.sidebar.selectbox("Choose a sample image", sample_images)
 
-if sample_image_uploaded:
-    uploaded_file = sample_image_uploaded
-else:
-    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+uploaded_file = st.sidebar.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
-image = ""
 if uploaded_file is not None:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image.", use_column_width=True)
+    if sample_image_selected:
+        sample_image_path = sample_images_folder / sample_image_selected
+        uploaded_file.name = sample_image_selected
+        uploaded_file.seek(0)
+        with open(sample_image_path, "wb") as f:
+            f.write(uploaded_file.read())
+        image = Image.open(sample_image_path)
+        st.image(image, caption="Uploaded Image.", use_column_width=True)
+    else:
+        image = Image.open(uploaded_file)
+        st.image(image, caption="Uploaded Image.", use_column_width=True)
 
 submit = st.button("Submit")