Update demo.py

demo.py

@@ -125,7 +125,10 @@ def generate_new_experiment() -> Experiment:
     random_folder = random.randint(1, 3)  # Adjusted to match folder numbering
 
     # Generate a two-digit sample number as a string
-    random_int = random.randint(0, NUMBER_OF_EXPERIMENTS - 1)
+    if random_folder > 1:
+        random_int = random.randint(0, NUMBER_OF_EXPERIMENTS - 1)
+    else:
+        random_int = random.randint(0, 10)
     video_name = f"sample00_rep{random_int:02d}.mp4"
 
     # Paths to the three videos