App: Add video thumbnails

[skip ci]

Author: khemarato

scripts/gdrive_app.py

@@ -24,9 +24,10 @@
 
 from collections import OrderedDict
 from functools import lru_cache
-import pdfutils as pdfutils
-import gdrive_base as gdrive_base
-from strutils import thumbnail_path_for_file
+import pdfutils
+import videoutils
+import gdrive_base
+from strutils import thumbnail_path_for_file, THUMBNAIL_SIZES
 
 
 class ThumbnailWorker(QRunnable):
@@ -35,6 +36,7 @@ def __init__(self, item, cancel_flag, emit_callback):
         self.item = item
         self.cancel_flag = cancel_flag
         self.emit_callback = emit_callback
+        self.size = 'normal'
 
     def is_cancelled(self):
         return self.cancel_flag[0]
@@ -55,19 +57,24 @@ def _process_item(self, item):
             cache_path = gcache.get_cache_path_for_file(item)
             if not cache_path:
                 return
-            thumb_path = thumbnail_path_for_file(cache_path, shared=True, size='normal')
+            size = THUMBNAIL_SIZES[self.size]
+            thumb_path = thumbnail_path_for_file(cache_path, shared=True, size=self.size)
             if thumb_path.exists():
                 img = QImage(str(thumb_path))
             elif cache_path.exists():
                 if self.is_cancelled(): return
                 if mime == 'application/pdf':
-                    thumbnail_bytes = pdfutils.get_cached_pdf_thumbnail(cache_path, size='normal')
+                    thumbnail_bytes = pdfutils.get_cached_pdf_thumbnail(cache_path, size=self.size)
+                    img = QImage()
+                    img.loadFromData(thumbnail_bytes)
+                elif mime.startswith('video/'):
+                    thumbnail_bytes = videoutils.get_cached_video_thumbnail(cache_path, size=self.size)
                     img = QImage()
                     img.loadFromData(thumbnail_bytes)
             else:
-                if mime == 'application/pdf':
+                if mime == 'application/pdf' or mime.startswith('video/'):
                     if self.is_cancelled(): return
-                    thumbnail_bytes = gdrive_base.fetch_preview_image(file_id, size=128)
+                    thumbnail_bytes = gdrive_base.fetch_preview_image(file_id, size=size)
                     if thumbnail_bytes:
                         img = QImage()
                         img.loadFromData(thumbnail_bytes)
@@ -374,7 +381,7 @@ def populate_files(self, items: List[Dict[str, Any]]):
                 list_item.setIcon(QIcon(cached_pixmap))
             else:
                 list_item.setIcon(get_mime_icon(mime))
-                if mime == 'application/pdf':
+                if mime == 'application/pdf' or mime.startswith('video/'):
                     items_needing_thumbnails.append(item)
 
             list_item.setData(Qt.UserRole, item)
@@ -395,7 +402,7 @@ def update_visible_thumbnails(self):
             mime = file_data.get('mimeType', '')
             file_id = file_data.get('id', '')
 
-            if mime == 'application/pdf' and file_id not in self.queued_thumbnails:
+            if (mime == 'application/pdf' or mime.startswith('video/')) and file_id not in self.queued_thumbnails:
                 if not self.thumbnail_cache.get(file_id):
                     item_rect = self.file_view.visualItemRect(item)
                     if expanded_rect.intersects(item_rect):

scripts/videoutils.py

@@ -0,0 +1,56 @@
+#!/bin/python3
+import subprocess
+from pathlib import Path
+from strutils import thumbnail_path_for_file, THUMBNAIL_SIZES
+
+def render_video_thumbnail(video_path: Path, size=128) -> bytes:
+    """Uses ffmpeg to extract a frame from a video file."""
+    try:
+        # We try to get a frame at 1 second in.
+        # -ss 1: seek to 1 second
+        # -i: input file
+        # -vframes 1: extract 1 frame
+        # -f image2pipe: output as image pipe
+        # -vcodec png: encode as png
+        # -vf scale: resize maintaining aspect ratio
+        cmd = [
+            'ffmpeg',
+            '-ss', '1',
+            '-i', str(video_path),
+            '-vframes', '1',
+            '-f', 'image2pipe',
+            '-vcodec', 'png',
+            '-vf', f'scale={size}:-1',
+            '-'
+        ]
+        # We use a short timeout and capture output
+        result = subprocess.run(cmd, capture_output=True, timeout=10)
+        if result.returncode == 0:
+            return result.stdout
+        else:
+            print(f"ffmpeg error for {video_path}: {result.stderr.decode(errors='replace')}")
+            return b''
+    except subprocess.TimeoutExpired:
+        print(f"ffmpeg timed out for {video_path}")
+        return b''
+    except Exception as e:
+        print(f"Error rendering video thumbnail for {video_path}: {e}")
+        return b''
+
+def get_cached_video_thumbnail(video_path: Path, size='normal') -> bytes:
+    """Returns the cached thumbnail bytes for a video file, or generates it if missing."""
+    thumbnail_path = thumbnail_path_for_file(video_path, shared=True, size=size)
+    if thumbnail_path.is_file():
+        return thumbnail_path.read_bytes()
+    
+    # Also check non-shared cache
+    alt_path = thumbnail_path_for_file(video_path, shared=False, size=size)
+    if alt_path.is_file():
+        return alt_path.read_bytes()
+
+    tsize = THUMBNAIL_SIZES[size]
+    thebytes = render_video_thumbnail(video_path, size=tsize)
+    if thebytes:
+        thumbnail_path.parent.mkdir(exist_ok=True, parents=True)
+        thumbnail_path.write_bytes(thebytes)
+    return thebytes