Update run.py

run.py

@@ -8,6 +8,8 @@ dataset_options = [
     "OpenVideo/pexels-raw",
     "OpenVideo/Sample-2k",
 ]
+fetch_files = []
+fetch_files_select_idx = 0
 
 def login(token):
     u = HFUser.from_token(token)
@@ -19,9 +21,7 @@ def show_time(u, name):
 def list_dataset(u, repo):
     print(f"repo is: {repo}")
     files = u.list_dataset(repo)
-    if repo == "OpenVideo/Sample-2k":
-        print(f"files is: {files}")
-        return gr.Dropdown(choices=files), gr.Column(visible=True), gr.Column(visible=False)
+    fetch_files = files
     return gr.Dropdown(value=files[0], choices=files), gr.Column(visible=True), gr.Column(visible=False)
 
 def fetch_parquet(u, fname):
@@ -54,6 +54,9 @@ def next_chunks(video_chunks, epoch_idx):
     length = len(video_chunks)
     return (epoch_idx+1)%length, gr.Slider(value=0)
 
+def next_mp4_chunks(video_chunks, epoch_idx):
+    return gr.Dropdown(value=fetch_files[fetch_files_select_idx + 1], choices=fetch_files)
+
 with gr.Blocks(**GR_CONF) as Core:
     user = gr.State()
     epoch_idx = gr.State(0)
@@ -93,7 +96,8 @@ with gr.Blocks(**GR_CONF) as Core:
             submit = gr.Button("Submit", variant="primary", size="sm", scale=1)
             with gr.Row():
                 rst = gr.Button("Reset", variant="stop", size="sm", scale=1)
-                nxt = gr.Button("Next Batch/MP4", variant="secondary", size="sm", scale=1)
+                nxt = gr.Button("Next Batch", variant="secondary", size="sm", scale=1)
+                nxt_mp4 = gr.Button("Next MP4", variant="secondary", size="sm", scale=1)
 
     _auth.click(fn=login, inputs=_token, outputs=[user, name,  Auth])
     name.change(fn=show_time, inputs=[user, name], outputs=[time, Repo])
@@ -104,6 +108,7 @@ with gr.Blocks(**GR_CONF) as Core:
     _pick.change(fn=select_video, inputs=[video_chunks, epoch_idx, _pick], outputs=_video)
     _video.change(fn=show_lables, outputs=Tag)
     nxt.click(fn=next_chunks, inputs=[video_chunks, epoch_idx], outputs=[epoch_idx, _pick])
+    nxt_mp4.click(fn=next_mp4_chunks, inputs=[video_chunks, epoch_idx], outputs=[file])
 
 if __name__ == "__main__":
     Core.launch()