Update app.py

app.py

@@ -6,28 +6,30 @@ from PIL import Image
 
 # Step 1: Download the latest version of the dataset and get the path
 path = kagglehub.dataset_download("imreallyjohn/cartoonset10k")
-images_folder = "/home/user/.cache/kagglehub/datasets/imreallyjohn/cartoonset10k/versions/1"  # Exact path to image directory
-
+images_folder = "/home/user/.cache/kagglehub/datasets/imreallyjohn/cartoonset10k/versions/1"
 # Display the dataset path
 st.write("Path to dataset files:", images_folder)
 
-# Button to generate a random avatar
-if st.button("Generate Random Avatar"):
-    # Step 2: List all image files in the dataset directory
-    try:
-        image_files = [f for f in os.listdir(images_folder) if f.endswith('.png')]  # List .png files
+# Step 2: List all image files in the dataset directory
+try:
+    image_files = [f for f in os.listdir(images_folder) if f.endswith('.png')]  # List .png files
+    num_images = len(image_files)
+
+    # Display the number of images loaded
+    st.write(f"Total images loaded: {num_images}")
+
+    if num_images == 0:
+        st.error("No available images to display. Please check the folder path and contents.")
+except Exception as e:
+    st.error(f"Error accessing images: {e}")
 
-        # Check if there are images in the directory
-        if image_files:
-            # Select a random available image
-            random_image_filename = random.choice(image_files)
-            image_path = os.path.join(images_folder, random_image_filename)
-            
-            # Load and display the selected image
-            selected_image = Image.open(image_path)
-            st.image(selected_image, caption=f"Displayed image: {random_image_filename}")
-            st.write(f"Displayed image: {random_image_filename}")
-        else:
-            st.error("No available images to display.")
-    except Exception as e:
-        st.error(f"Error accessing images: {e}")
+# Button to generate a random avatar if images are available
+if num_images > 0 and st.button("Generate Random Avatar"):
+    # Step 3: Select a random available image
+    random_image_filename = random.choice(image_files)
+    image_path = os.path.join(images_folder, random_image_filename)
+    
+    # Load and display the selected image
+    selected_image = Image.open(image_path)
+    st.image(selected_image, caption=f"Displayed image: {random_image_filename}")
+    st.write(f"Displayed image: {random_image_filename}")