Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 from io import BytesIO
 from tqdm import tqdm
 import time
+import os
 
 repo = "artificialguybr/TshirtDesignRedmond-V2"
 
@@ -117,14 +118,18 @@ document.addEventListener('DOMContentLoaded', function () {
 </script>
 """
 
-# Gallery Images with captions
-gallery_images = [
-    ("https://gaur3009-phone-c.hf.space/gradio_api/file=/tmp/gradio/387f5407e32dd2e2d9e7017977f5ee98e4f40dcb8b1d2a9ef23612255a675163/image.webp", "Red Color, iPhone 14, Naruto Anime Theme"),
-    ("https://gaur3009-phone-c.hf.space/gradio_api/file=/tmp/gradio/3037d9cae7a86b29be5969cf3a361ae847915b54d97d6d5dffb6e019a2b9ddc9/image.webp", "Red Color, iPhone 14, Darth Vader Theme"),
-    ("https://gaur3009-phone-c.hf.space/gradio_api/file=/tmp/gradio/1958c8d60219357edbb2e1f69331152f1cd1109407b274c4511a810650dd574b/image.webp", "Black, iPhone 15, Thomas Shelby (Peaky Blinders) Theme"),
-    ("https://gaur3009-phone-c.hf.space/gradio_api/file=/tmp/gradio/183392add00a7fa3d53653f856bc88d4113bd74c0d37c43de7a9af414e9c1d84/image.webp", "Green color, iPhone, Green Goblin"),
-    ("https://gaur3009-phone-c.hf.space/gradio_api/file=/tmp/gradio/b7b0b2415f15f810f71658616a885b97c0466d09f1d852684a868ab9b5d18e6c/image.webp", "White and Cream Color, iPhone 15, Iron-Man"),
-]
+# Function to load images dynamically from "New folder (4)"
+def load_gallery_images():
+    folder_path = "New folder (4)"
+    images_with_captions = []
+    for filename in os.listdir(folder_path):
+        if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.webp')):
+            img_path = os.path.join(folder_path, filename)
+            caption = filename.rsplit('.', 1)[0]  # Use filename (without extension) as caption
+            images_with_captions.append((img_path, caption))
+    return images_with_captions
+
+gallery_images = load_gallery_images()
 
 # Gradio interface
 with gr.Blocks(css=custom_css) as interface:
@@ -138,8 +143,8 @@ with gr.Blocks(css=custom_css) as interface:
             gr.Markdown("Welcome to the **AI Phone Cover Designer**! Use the 'Design' tab to start creating custom designs.")
             gr.Markdown("### Gallery")
             gallery_html = "<div class='carousel'>"
-            for img_url, caption in gallery_images:
-                gallery_html += f"<div class='image-container'><img src='{img_url}' alt='Design'><div class='caption'>{caption}</div></div>"
+            for img_path, caption in gallery_images:
+                gallery_html += f"<div class='image-container'><img src='{img_path}' alt='Design'><div class='caption'>{caption}</div></div>"
             gallery_html += "</div>"
             gr.HTML(gallery_html)
 
@@ -183,4 +188,4 @@ with gr.Blocks(css=custom_css) as interface:
             """)
 
 # Launch the app
-interface.launch(debug=True)
+interface.launch(debug=True)