Adding a default list of test

src/app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 from utils import load_model, segment_person, resize_image, split_stereo_image
-
+from testing import get_image_names
 # Load model and processor once
 processor, model = load_model()
 
@@ -80,6 +80,7 @@ demo = gr.Interface(
         gr.Image(label="Stereo_pair"),
         gr.Image(label="3D Anaglyph Image")
     ],
+    examples= get_image_names(),
     title="3D Person Segmentation Viewer",
     description="Upload a person photo and optionally a background image. Outputs anaglyph and stereo views."
 )

src/testing.py

@@ -1,4 +1,33 @@
-from app import create_anaglyph
-
 # Provide paths to your test images
-create_anaglyph("person.png", "bg.png", "test_anaglyph.png")
+import os
+import random
+
+def get_image_names():
+    # Get background images
+    background_dir = "public/images/backgrounds"
+    background_images = [f for f in os.listdir(background_dir) if f.endswith(('.jpg', '.jpeg', '.png'))]
+    
+    # Get people images
+    people_dir = "public/images/people" 
+    people_images = [f for f in os.listdir(people_dir) if f.endswith(('.jpg', '.jpeg', '.png'))]
+    
+
+    generate_testing_list = []
+
+    random.shuffle(background_images)
+    random.shuffle(people_images)
+
+    min_length = min(len(background_images), len(people_images))
+
+    # Get 10 random pairs
+    for i in range(min_length):
+        background = background_images[i]
+        person = people_images[i]
+        generate_testing_list.append(list(( os.path.join(people_dir, person), os.path.join(background_dir, background), 10, 100)))
+
+    
+    return generate_testing_list
+
+
+
+