Update app.py

app.py

@@ -86,14 +86,14 @@ def update(user_choice, data_folder=VIDEO_PATH):
 def replay(left, right):  
     return left, right
 
-def parse_envs(folder='./videos'):
+def parse_envs(folder=VIDEO_PATH):
     envs = []
     for f in os.listdir(folder):
         if os.path.isdir(os.path.join(folder, f)):
             envs.append(f)
     return envs
 
-def build_interface(iter=3, data_folder='./videos'):
+def build_interface(iter=3, data_folder=VIDEO_PATH):
     HF_TOKEN = os.getenv('HF_TOKEN')
     print(HF_TOKEN)
     HF_TOKEN = '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'  # my HF token
@@ -110,16 +110,16 @@ def build_interface(iter=3, data_folder='./videos'):
             # some initial videos
             if FORMAT == 'mp4':
                 left_video_path = os.path.join(os.path.dirname(__file__), 
-                                    "videos/rl-video-episode-0.mp4")
+                                    f"{VIDEO_PATH}/rl-video-episode-0.mp4")
                 right_video_path = os.path.join(os.path.dirname(__file__), 
-                                    "videos/rl-video-episode-1.mp4")
+                                    f"{VIDEO_PATH}/rl-video-episode-1.mp4")
                 left = gr.PlayableVideo(left_video_path, label="left_video")
                 right = gr.PlayableVideo(right_video_path, label="right_video")
             else:
                 left_video_path = os.path.join(os.path.dirname(__file__), 
-                                    "videos/rl-video-episode-0.gif")
+                                    f"{VIDEO_PATH}/rl-video-episode-0.gif")
                 right_video_path = os.path.join(os.path.dirname(__file__), 
-                                    "videos/rl-video-episode-1.gif")  
+                                    f"{VIDEO_PATH}/rl-video-episode-1.gif")  
                 left = gr.Image(left_video_path, shape=(1024, 768), label="left_video")
                 # right = gr.Image(right_video_path).style(height=768, width=1024)
                 right = gr.Image(right_video_path, label="right_video")