Update app.py

app.py

@@ -69,6 +69,8 @@ DEFAULT_PRESET = "Robosuite Square Correction"
 REPO_TYPE = "dataset"
 DEFAULT_CHUNK_LEN = 16
 DEFAULT_DISPLAY_SCALE = 1
+VIDEO_STATUS_FIGSIZE = (6.0, 1.8)
+VIDEO_STATUS_DPI = 120
 PREFERRED_IMAGE_KEYS = [
     "image1",
     "image2",
@@ -720,7 +722,7 @@ def get_video_status_plot_base(repo_id, filename, traj_id, valid_window_len):
 
     no_teacher, valid_indices = _compute_valid_start_indices(traj, int(valid_window_len))
 
-    fig, ax = plt.subplots(figsize=(10.5, 2.8), dpi=170)
+    fig, ax = plt.subplots(figsize=VIDEO_STATUS_FIGSIZE, dpi=VIDEO_STATUS_DPI)
     ax.step(
         np.arange(total_steps),
         no_teacher,
@@ -741,14 +743,14 @@ def get_video_status_plot_base(repo_id, filename, traj_id, valid_window_len):
 
     ax.set_xlim(0, max(total_steps - 1, 1))
     ax.set_ylim(-0.38, 1.1)
-    ax.set_ylabel("Flag", fontsize=10)
-    ax.set_xlabel("Timestep index", fontsize=10)
+    ax.set_ylabel("Flag", fontsize=8)
+    ax.set_xlabel("Timestep index", fontsize=8)
     ax.set_yticks([0, 1])
     ax.set_yticklabels(["False", "True"])
     ax.grid(True, axis="x", alpha=0.2)
-    ax.set_title("no_teacher_action and valid starts", fontsize=11)
-    ax.tick_params(axis="both", labelsize=9)
-    ax.legend(loc="upper right", fontsize=9)
+    ax.set_title("no_teacher_action and valid starts", fontsize=9)
+    ax.tick_params(axis="both", labelsize=7)
+    ax.legend(loc="upper right", fontsize=7)
     fig.tight_layout()
     fig.canvas.draw()
 
@@ -849,7 +851,7 @@ def _get_fast_video_writer(out_path, fps):
     )
 
 
-def build_current_trajectory_video(preset_name, custom_repo_id, custom_filename, traj_id, image_keys, display_scale, reverse_channels, fps, valid_window_len, video_stride=1):
+def build_current_trajectory_video(preset_name, custom_repo_id, custom_filename, traj_id, image_keys, display_scale, reverse_channels, fps, valid_window_len, video_stride=4):
     if imageio is None:
         return None, "Video export requires imageio and imageio-ffmpeg in requirements.txt."
 
@@ -1121,7 +1123,7 @@ def build_app():
             preload_btn = gr.Button("Preload current trajectory")
             video_btn = gr.Button("Build trajectory video")
             video_fps = gr.Slider(minimum=1, maximum=30, value=10, step=1, label="Video FPS")
-            video_stride = gr.Slider(minimum=1, maximum=10, value=1, step=1, label="Video frame stride")
+            video_stride = gr.Slider(minimum=1, maximum=10, value=4, step=1, label="Video frame stride")
 
         preload_status = gr.Textbox(label="Preload / video status", lines=4, value="Not preloaded yet.", interactive=False)