Create app.py

app.py

@@ -0,0 +1,43 @@
+import streamlit as st
+import os
+import random
+from PIL import Image
+import kagglehub
+
+# Step 1: Download the latest version of the dataset
+path = kagglehub.dataset_download("imreallyjohn/cartoonset10k")
+
+# Display the path to the dataset files
+st.write("Path to dataset files:", path)
+
+# Step 2: Set the images folder
+images_folder = os.path.join(path, "cartoonset10k/renamed_images/renamed_images")  # Adjust as necessary
+
+# Step 3: List all image files in the dataset directory
+image_files = [f for f in os.listdir(images_folder) if f.endswith('.png')]  # Filter for .png files
+
+# Initialize a variable to hold the selected image path
+selected_image_path = None
+
+# Step 4: Create a button to generate a random avatar
+if st.button("Generate Avatar"):
+    if image_files:
+        # Select a random available image
+        random_image_filename = random.choice(image_files)  # Select a random image
+        selected_image_path = os.path.join(images_folder, random_image_filename)
+
+        # Load and display the selected image
+        selected_image = Image.open(selected_image_path)
+        st.image(selected_image, caption=random_image_filename)  # Display image with caption
+        st.success(f"Displayed image: {random_image_filename}")  # Show success message
+    else:
+        st.warning("No available images to display.")  # Show warning if no images are found
+
+# Optional: Add a download link for the displayed image
+if selected_image_path:
+    st.download_button(
+        label="Download Avatar",
+        data=open(selected_image_path, "rb").read(),
+        file_name=os.path.basename(selected_image_path),
+        mime="image/png"
+    )