Update app.py

app.py

@@ -26,6 +26,8 @@ from huggingface_hub import (
     snapshot_download,
 )
 from PIL import Image
+import concurrent.futures
+
 
 cached_latest_posts_df = None
 cached_top_posts = None
@@ -33,6 +35,13 @@ last_fetched = None
 last_fetched_top = None
 
 
+def resize_image(image):
+    width, height = image.size
+    new_width = width * 0.35
+    new_height = height * 0.35
+    return image.resize((int(new_width), int(new_height)), Image.BILINEAR)
+
+
 def get_reddit_id(url):
     # Regular expression pattern for r/GamePhysics URLs and IDs
     pattern = r"https://www\.reddit\.com/r/GamePhysics/comments/([0-9a-zA-Z]+).*|([0-9a-zA-Z]+)"
@@ -106,6 +115,47 @@ def extract_frames_decord(video_path, num_frames=10):
         raise Exception(f"Error extracting frames from video: {e}")
 
 
+def extract_frames_decord_preview(video_path, num_frames=10):
+    try:
+        start_time = time.time()
+
+        print(f"Extracting {num_frames} frames from {video_path}")
+
+        # Load the video
+        vr = VideoReader(video_path, ctx=cpu(0))
+
+        # Calculate the indices for the frames to be extracted
+        total_frames = len(vr)
+        frame_indices = np.linspace(
+            0, total_frames - 1, num_frames, dtype=int, endpoint=False
+        )
+
+        # Extract frames
+        batch_frames = vr.get_batch(frame_indices).asnumpy()
+
+        # Convert frames to PIL Images
+        frame_images = [
+            Image.fromarray(batch_frames[i]) for i in range(batch_frames.shape[0])
+        ]
+
+        end_time = time.time()
+        print(f"Decord extraction took {end_time - start_time} seconds")
+
+        # # resize images  to save bandwidth, keep aspect ratio
+        # for i, image in enumerate(frame_images):
+        #     width, height = image.size
+        #     new_width = int(width * 0.35)
+        #     new_height = int(height * 0.35)
+        #     frame_images[i] = image.resize((new_width, new_height), Image.ANTIALIAS)
+
+        with concurrent.futures.ThreadPoolExecutor() as executor:
+            frame_images = list(executor.map(resize_image, frame_images))
+
+        return frame_images
+    except Exception as e:
+        raise Exception(f"Error extracting frames from video: {e}")
+
+
 def get_top_posts():
     global cached_top_posts
     global last_fetched_top
@@ -212,7 +262,7 @@ def load_video(url):
         and r1.status_code != 302
     ):
         raise gr.Error(
-            f"Video is not in the repo, please try another post. - {r.status_code  }"
+            f"Video is not in the repo, please try another post. - {r1.status_code  }"
         )
 
     if r1.status_code == 200 or r1.status_code == 302:
@@ -234,11 +284,11 @@ with gr.Blocks() as demo:
             video_player = gr.Video(interactive=False)
 
             with gr.Column():
-                gr.Markdown("## Sampled Frames from Video")
+                gr.Markdown("## Sample frames")
                 num_frames = gr.Slider(minimum=1, maximum=60, step=1, value=10)
-                sample_decord_btn = gr.Button("Sample decord")
+                sample_decord_btn = gr.Button("Sample frames")
 
-                sampled_frames = gr.Gallery()
+                sampled_frames = gr.Gallery(label="Sampled frames preview")
 
                 download_samples_btn = gr.Button("Download Samples")
                 output_files = gr.File()
@@ -250,16 +300,16 @@ with gr.Blocks() as demo:
                 )
 
         with gr.Column():
-            gr.Markdown("## Latest Posts")
-            with gr.Accordion("Latest Posts"):
+            gr.Markdown("## Reddits Posts")
+            with gr.Tab("Latest Posts"):
                 latest_post_dataframe = gr.Dataframe()
                 latest_posts_btn = gr.Button("Refresh Latest Posts")
-            with gr.Accordion("Top Posts"):
+            with gr.Tab("Top Monthly Posts"):
                 top_posts_dataframe = gr.Dataframe()
                 top_posts_btn = gr.Button("Refresh Top Posts")
 
     sample_decord_btn.click(
-        extract_frames_decord,
+        extract_frames_decord_preview,
         inputs=[video_player, num_frames],
         outputs=[sampled_frames],
     )