updated thumbnail for better performance

Author: stephengpope

services/v1/video/thumbnail.py

@@ -18,49 +18,44 @@
 
 import os
 import ffmpeg
-from services.file_management import download_file
 from config import LOCAL_STORAGE_PATH
 
 def extract_thumbnail(video_url, job_id, second=0):
     """
     Extract a thumbnail from a video at the specified timestamp.
-    
+    Uses ffmpeg direct streaming to avoid downloading the entire video file.
+
     Args:
         video_url (str): URL of the video to extract thumbnail from
         job_id (str): Unique identifier for the job
         second (float): Timestamp in seconds to extract the thumbnail from (default: 0)
-        
+
     Returns:
         str: Path to the extracted thumbnail image
     """
-    # Download the video from the provided URL
-    video_path = download_file(video_url, os.path.join(LOCAL_STORAGE_PATH, f"{job_id}_input"))
-    
     # Set output path for the thumbnail
     thumbnail_path = os.path.join(LOCAL_STORAGE_PATH, f"{job_id}_thumbnail.jpg")
-    
+
     try:
-        # Extract thumbnail using ffmpeg at the specified timestamp
+        # Extract thumbnail directly from URL using ffmpeg streaming
+        # analyzeduration and probesize are set low to reduce initial buffering
         (
             ffmpeg
-            .input(video_path, ss=second)  # 'ss' is the seek parameter for the timestamp
-            .output(thumbnail_path, vframes=1)  # vframes=1 extracts a single frame
+            .input(video_url, ss=second, analyzeduration='100K', probesize='100K')
+            .output(thumbnail_path, vframes=1, update=1)
             .overwrite_output()
             .run(capture_stdout=True, capture_stderr=True)
         )
-        
-        # Clean up the downloaded video file
-        os.remove(video_path)
-        
+
         # Ensure the thumbnail file exists
         if not os.path.exists(thumbnail_path):
             raise FileNotFoundError(f"Thumbnail file {thumbnail_path} was not created")
-            
+
         return thumbnail_path
-        
+
     except Exception as e:
         print(f"Thumbnail extraction failed: {str(e)}")
-        # Clean up any downloaded files on error
-        if os.path.exists(video_path):
-            os.remove(video_path)
+        # Clean up partial thumbnail file on error
+        if os.path.exists(thumbnail_path):
+            os.remove(thumbnail_path)
         raise