return earlier local file path calculation

app.py

@@ -5,7 +5,6 @@ import glob
 import streamlit as st
 
 from utils import get_configs, get_display_names, get_path_for_viz, get_video_height, get_text_str
-# from gdrive import download_file
 
 # st.header("EVREAL  - Event-based Video Reconstruction Evaluation and Analysis Library")
 #
@@ -23,6 +22,7 @@ st.title("Result Analysis Tool")
 
 font_path = "font/Ubuntu-B.ttf"
 
+base_data_dir = "data"
 dataset_cfg_path = os.path.join("cfg", "dataset")
 model_cfg_path = os.path.join("cfg", "model")
 metric_cfg_path = os.path.join("cfg", "metric")
@@ -135,9 +135,7 @@ for row_idx in range(num_rows):
         cur_model = selected_models[col_idx]
         if cur_model['name'] == "gt":
             if row_idx < len(gt_viz):
-                video_path = get_path_for_viz(selected_dataset, selected_sequence, cur_model, gt_viz[row_idx])
-                video_path = os.path.join("data", video_path)
-                # download_file(video_path, local_video_path)
+                video_path = get_path_for_viz(base_data_dir, selected_dataset, selected_sequence, cur_model, gt_viz[row_idx])
                 if not os.path.isfile(video_path):
                     raise ValueError("Could not find video: " + video_path)
                 gt_viz_indices.append(vid_idx)
@@ -145,9 +143,7 @@ for row_idx in range(num_rows):
                 continue
         else:
             if row_idx < len(model_viz):
-                video_path = get_path_for_viz(selected_dataset, selected_sequence, cur_model, model_viz[row_idx])
-                video_path = os.path.join("data", video_path)
-                # download_file(video_path, local_video_path)
+                video_path = get_path_for_viz(base_data_dir, selected_dataset, selected_sequence, cur_model, model_viz[row_idx])
                 if not os.path.isfile(video_path):
                     raise ValueError("Could not find video: " + video_path)
             else:

utils.py

@@ -31,9 +31,9 @@ def get_display_names(configs):
     return display_names
 
 
-def get_path_for_viz(dataset, sequence, model, viz):
+def get_path_for_viz(basedir, dataset, sequence, model, viz):
     dat_seq = "_".join([dataset["name"], sequence])
-    video_path = os.path.join(model['model_id'], dat_seq, "videos", viz["name"] + ".mp4")
+    video_path = os.path.join(basedir, model['model_id'], dat_seq, "videos", viz["name"] + ".mp4")
     return video_path