Update app.py

app.py

@@ -63,20 +63,22 @@ def show_image(image, title="Image"):
 
 # Function to download sample images
 def download_sample_images():
-    sample_images_folder = "sample_images"
-    sample_images = os.listdir(sample_images_folder)
+    sample_images = ["sample_image1.jpg", "sample_image2.jpg", "sample_image3.jpg"]  # Add the names of your sample images
 
     for image_name in sample_images:
-        image_path = os.path.join(sample_images_folder, image_name)
-        with open(image_path, "rb") as f:
-            image_bytes = f.read()
-            st.download_button(
-                label=f"Download {image_name}",
-                data=image_bytes,
-                key=f"download_{image_name}",
-                file_name=image_name,
-                mime="image/jpeg",
-            )
+        image_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), image_name)
+        if os.path.exists(image_path):
+            with open(image_path, "rb") as f:
+                image_bytes = f.read()
+                st.download_button(
+                    label=f"Download {image_name}",
+                    data=image_bytes,
+                    key=f"download_{image_name}",
+                    file_name=image_name,
+                    mime="image/jpeg",
+                )
+        else:
+            st.warning(f"Sample image {image_name} not found.")
 
 # Streamlit app
 def main():