update

app.py

@@ -3,18 +3,21 @@ import os
 import random
 import re
 import sys
+import tempfile
 import time
 from datetime import datetime
 from glob import glob
 from pathlib import Path
 from typing import List, Optional
 from uuid import uuid4
+from zipfile import ZipFile
 
 import gradio as gr
 import numpy as np
 import pandas as pd
 import requests
 from datasets import load_dataset
+from decord import VideoReader, cpu
 from huggingface_hub import (
     CommitScheduler,
     HfApi,
@@ -29,15 +32,6 @@ cached_top_posts = None
 last_fetched = None
 last_fetched_top = None
 
-import os
-import tempfile
-from zipfile import ZipFile
-
-import numpy as np
-from PIL import Image
-from decord import VideoReader
-from decord import cpu
-
 
 def get_reddit_id(url):
     # Regular expression pattern for r/GamePhysics URLs and IDs
@@ -184,23 +178,21 @@ def get_latest_posts():
     return examples
 
 
-def row_selected(evt: gr.SelectData):
-    global cached_latest_posts_df
+def row_selected_top(evt: gr.SelectData):
     global cached_top_posts
-
-    # find which dataframe was selected
     string_value = evt.value
     row = evt.index[0]
-    target_df = None
 
-    if cached_latest_posts_df.isin([string_value]).any().any():
-        target_df = cached_latest_posts_df
-    elif cached_top_posts.isin([string_value]).any().any():
-        target_df = cached_top_posts
-    else:
-        raise gr.Error("Could not find selected post in any dataframe")
+    post_id = cached_top_posts.iloc[row]["post_id"]
+    return post_id
 
-    post_id = target_df.iloc[row]["post_id"]
+
+def row_selected_latest(evt: gr.SelectData):
+    global cached_latest_posts_df
+    string_value = evt.value
+    row = evt.index[0]
+
+    post_id = cached_latest_posts_df.iloc[row]["post_id"]
     return post_id
 
 
@@ -230,28 +222,30 @@ with gr.Blocks() as demo:
             load_btn = gr.Button("Load")
             video_player = gr.Video(interactive=False)
 
-        with gr.Column():
-            gr.Markdown("## Latest Posts")
-            latest_post_dataframe = gr.Dataframe()
-            latest_posts_btn = gr.Button("Refresh Latest Posts")
-            top_posts_btn = gr.Button("Refresh Top Posts")
+            with gr.Column():
+                gr.Markdown("## Sampled Frames from Video")
+                num_frames = gr.Slider(minimum=1, maximum=60, step=1, value=10)
+                sample_decord_btn = gr.Button("Sample decord")
 
-    with gr.Column():
-        gr.Markdown("## Sampled Frames from Video")
-        with gr.Row():
-            num_frames = gr.Slider(minimum=1, maximum=60, step=1, value=10)
-            sample_decord_btn = gr.Button("Sample decord")
+                sampled_frames = gr.Gallery()
 
-        sampled_frames = gr.Gallery()
+                download_samples_btn = gr.Button("Download Samples")
+                output_files = gr.File()
 
-        download_samples_btn = gr.Button("Download Samples")
-        output_files = gr.File()
+                download_samples_btn.click(
+                    download_samples,
+                    inputs=[reddit_id, video_player, num_frames],
+                    outputs=[output_files],
+                )
 
-        download_samples_btn.click(
-            download_samples,
-            inputs=[reddit_id, video_player, num_frames],
-            outputs=[output_files],
-        )
+        with gr.Column():
+            gr.Markdown("## Latest Posts")
+            with gr.Accordion("Latest Posts"):
+                latest_post_dataframe = gr.Dataframe()
+                latest_posts_btn = gr.Button("Refresh Latest Posts")
+            with gr.Accordion("Top Posts"):
+                top_posts_dataframe = gr.Dataframe()
+                top_posts_btn = gr.Button("Refresh Top Posts")
 
     sample_decord_btn.click(
         extract_frames_decord,
@@ -262,11 +256,15 @@ with gr.Blocks() as demo:
     load_btn.click(load_video, inputs=[reddit_id], outputs=[video_player])
 
     latest_posts_btn.click(get_latest_posts, outputs=[latest_post_dataframe])
-    top_posts_btn.click(get_top_posts, outputs=[latest_post_dataframe])
+    top_posts_btn.click(get_top_posts, outputs=[top_posts_dataframe])
 
     demo.load(get_latest_posts, outputs=[latest_post_dataframe])
+    demo.load(get_top_posts, outputs=[top_posts_dataframe])
 
-    latest_post_dataframe.select(fn=row_selected, outputs=[reddit_id]).then(
+    latest_post_dataframe.select(fn=row_selected_latest, outputs=[reddit_id]).then(
+        load_video, inputs=[reddit_id], outputs=[video_player]
+    )
+    top_posts_dataframe.select(fn=row_selected_top, outputs=[reddit_id]).then(
         load_video, inputs=[reddit_id], outputs=[video_player]
     )