GH-59: Fix the last frame generation issue (GH-60)

Author: ArtyomVancyan

src/thumbnails/__init__.py

@@ -22,7 +22,7 @@
 from .thumbnail import ThumbnailVTT
 from .thumbnail import register_thumbnail
 
-__version__ = "0.1.6"
+__version__ = "0.1.7"
 __all__ = (
     "Generator",
     "Thumbnail",

src/thumbnails/video.py

@@ -88,17 +88,20 @@ def _extract_frame(self, start_time):
         output = os.path.join(self.tempdir.name, filename)
         os.close(os.open(output, os.O_CREAT, mode=0o664))
 
-        cmd = (
-            ffmpeg_bin,
-            "-ss", offset,
+        ffmpeg_options = (
             "-i", self.filepath,
             "-loglevel", "error",
             "-vframes", "1",
             output,
             "-y",
         )
 
-        subprocess.Popen(cmd).wait()
+        subprocess.Popen((ffmpeg_bin, "-ss", offset, *ffmpeg_options)).wait()
+
+        if os.stat(output).st_size == 0:
+            # Check if the frame is empty. If so, try to extract it again with a smaller offset.
+            # This handles the case when ffmpeg cannot extract the last frame of the video.
+            subprocess.Popen((ffmpeg_bin, "-sseof", "-0.1", *ffmpeg_options)).wait()
 
     def extract_frames(self):
         """Extracts the frames from the video by given intervals."""