Update app.py

app.py

@@ -29,7 +29,7 @@ if LOAD_DATA_GOOGLE_DRIVE:  # download data from google drive
     zip_files = [join(path_to_zip_file, f) for f in listdir(path_to_zip_file)]
     for f in zip_files:
         if f.endswith(".zip"):
-            directory_to_extract_to = f.replace('.zip', '')
+            directory_to_extract_to = path_to_zip_file # extracted file itself contains a folder
             print(f'extract data {f} to {directory_to_extract_to}')
             with zipfile.ZipFile(f, 'r') as zip_ref:
                 zip_ref.extractall(directory_to_extract_to)
@@ -117,7 +117,6 @@ def replay(left, right):
 
 def parse_envs(folder=VIDEO_PATH):
     envs = []
-    print('folder: ', os.listdir(folder))
     for f in os.listdir(folder):
         if os.path.isdir(os.path.join(folder, f)):
             envs.append(f)
@@ -147,7 +146,6 @@ def build_interface(iter=3, data_folder=VIDEO_PATH):
                 if f.endswith(f'.{FORMAT}'):
                     video_files.append(os.path.join(data_folder, env_name, f))
             # randomly choose two videos
-            print(video_files, envs)
             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]]