Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 import gradio as gr
 import cv2
 import tempfile
+import os
 
 # Load YOLOv8 model
 model = YOLO("best.pt")  # Ensure best.pt is in the same directory
@@ -17,7 +18,7 @@ def predict_image(image):
 def predict_video(video):
     try:
         # Save video to a temporary file
-        temp_video_path = tempfile.NamedTemporaryFile(suffix=".mp4").name
+        temp_video_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
         with open(temp_video_path, "wb") as f:
             f.write(video.read())
 
@@ -32,8 +33,8 @@ def predict_video(video):
         height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
 
         # Define codec and create a video writer
-        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-        output_path = tempfile.NamedTemporaryFile(suffix=".mp4").name
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Use 'mp4v' for compatibility
+        output_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
         out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
 
         while cap.isOpened():
@@ -52,7 +53,11 @@ def predict_video(video):
         cap.release()
         out.release()
 
-        return output_path
+        # Ensure the output file exists and is playable
+        if os.path.exists(output_path):
+            return output_path
+        else:
+            return "Error: Annotated video could not be created."
 
     except Exception as e:
         return f"An error occurred while processing the video: {str(e)}"