fix gt column height issue

app.py

@@ -110,14 +110,17 @@ xstack_input_parts = []
 layout_parts = []
 video_paths = []
 row_heights = [""]*num_rows
+gt_viz_indices = []
 for row_idx in range(num_rows):
     for col_idx in range(num_cols):
+        vid_idx = len(video_paths)
         cur_model = selected_models[col_idx]
         if cur_model['name'] == "gt":
             if row_idx < len(gt_viz):
                 video_path = get_path_for_viz(data_base_path, selected_dataset, selected_sequence, cur_model, gt_viz[row_idx])
                 # if not os.path.isfile(video_path):
                 #     raise ValueError("Video path does not exist: " + video_path)
+                gt_viz_indices.append(vid_idx)
             else:
                 continue
         else:
@@ -128,7 +131,6 @@ for row_idx in range(num_rows):
             else:
                 continue
 
-        vid_idx = len(video_paths)
         if row_heights[row_idx] == "":
             row_heights[row_idx] = "h{}".format(vid_idx)
         input_filter_part = "[{}:v]scale={}:-1,{}[v{}]".format(vid_idx, w, pad_str, vid_idx)
@@ -137,7 +139,10 @@ for row_idx in range(num_rows):
         xstack_input_parts.append(xstack_input_part)
         video_paths.append(video_path)
         layout_w = "+".join(["w{}".format(i) for i in range(col_idx)]) if col_idx > 0 else "0"
-        layout_h = "+".join(row_heights[:row_idx]) if row_idx > 0 else "0"
+        if cur_model['name'] == "gt":
+            layout_h = "+".join(["h{}".format(i) for i in gt_viz_indices[:-1]]) if row_idx > 0 else "0"
+        else:
+            layout_h = "+".join(row_heights[:row_idx]) if row_idx > 0 else "0"
         layout_part = layout_w + "_" + layout_h
         layout_parts.append(layout_part)