Update app.py

app.py

@@ -119,10 +119,16 @@ def build_interface(iter=3, data_folder=VIDEO_PATH):
             # some initial values
             envs = parse_envs()   
             env_name = envs[random.randint(0, len(envs)-1)] # random pick an env 
-            left_video_path = os.path.join(os.path.dirname(__file__), 
-                                f"{VIDEO_PATH}/{env_name}/rl-video-episode-0.{FORMAT}")
-            right_video_path = os.path.join(os.path.dirname(__file__), 
-                                f"{VIDEO_PATH}/{env_name}/rl-video-episode-1.{FORMAT}")
+            # choose video
+            videos = os.listdir(os.path.join(data_folder, env_name))
+            video_files = []
+            for f in videos:
+                if f.endswith(f'.{FORMAT}'):
+                    video_files.append(os.path.join(data_folder, env_name, f))
+            # randomly choose two videos
+            selected_video_ids = np.random.choice(len(video_files), 2, replace=False)
+            left_video_path = video_files[selected_video_ids[0]]
+            right_video_path = video_files[selected_video_ids[1]]
             if FORMAT == 'mp4':
                 left = gr.PlayableVideo(left_video_path, label="left_video")
                 right = gr.PlayableVideo(right_video_path, label="right_video")