Update app.py

app.py

@@ -1,59 +1,55 @@
 import gradio as gr
+import cv2
+import numpy as np
 from PIL import Image
-import subprocess
-import tempfile
-import os
 
-def extract_last_frame_fast(video_path):
+def extract_last_frame(video_path):
+    # Check if video_path is None or invalid
     if not video_path:
-        return None
-
-    # Create a temporary file for the output frame
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp:
-        output_frame = tmp.name
-
-    # This ffmpeg command is magic: seeks near the end instantly without decoding everything
-    cmd = [
-        "ffmpeg",
-        "-i", video_path,
-        "-sseof", "-3",           # start 3 seconds before the end
-        "-frames:v", "1",         # extract exactly 1 frame
-        "-update", "1",
-        "-q:v", "3",              # good quality
-        output_frame,
-        "-y"
-    ]
-
-    # Run silently and fast
-    result = subprocess.run(
-        cmd,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.PIPE,
-    )
-
-    if result.returncode != 0 or not os.path.exists(output_frame):
-        # Fallback: try one more time at -1 second
-        cmd[4] = "-1"
-        subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-
-    if os.path.exists(output_frame) and os.path.getsize(output_frame) > 0:
-        img = Image.open(output_frame)
-        os.unlink(output_frame)  # cleanup
-        return img
-    else:
-        os.unlink(output_frame)
-        return None  # Gradio will show broken image icon
-
-# Ultra-fast interface
-demo = gr.Interface(
-    fn=extract_last_frame_fast,
-    inputs=gr.Video(label="Upload Video (max 6 MB)"),
-    ),
-    outputs=gr.Image(label="Last Frame"),
-    title="Instant Last Frame Extractor",
-    description="Works instantly even on 6 MB videos!",
-    allow_flagging="never",
-    cache_examples=False
+        return "Error: No video file uploaded."
+
+    try:
+        # Open the video file
+        cap = cv2.VideoCapture(video_path)
+        
+        # Check if video opened successfully
+        if not cap.isOpened():
+            return "Error: Could not open video file."
+
+        # Get the total number of frames
+        total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+        
+        # Set the frame position to the last frame
+        cap.set(cv2.CAP_PROP_POS_FRAMES, total_frames - 1)
+        
+        # Read the last frame
+        ret, frame = cap.read()
+        
+        # Release the video capture object
+        cap.release()
+        
+        if not ret:
+            return "Error: Could not read the last frame."
+        
+        # Convert the frame from BGR (OpenCV format) to RGB
+        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        
+        # Convert to PIL Image for Gradio compatibility
+        pil_image = Image.fromarray(frame_rgb)
+        
+        return pil_image
+    
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=extract_last_frame,  # Function to call
+    inputs=gr.Video(label="Upload Video"),  # Input component for video upload
+    outputs=gr.Image(label="Last Frame"),   # Output component for displaying the image
+    title="Extract Last Frame from Video",
+    description="Upload a video file to extract and display its last frame."
 )
 
-demo.launch(max_file_size="6MB")  # your limit stays exactly 6 MB
+# Launch the Gradio app
+interface.launch(max_file_size="6MB")