Update app.py

app.py

@@ -1,6 +1,7 @@
 from PIL import Image, ImageDraw, ImageFont
 from dotenv import load_dotenv
 import matplotlib.pyplot as plt
+from moviepy.editor import *
 from io import BytesIO
 from glob import glob
 import gradio as gr
@@ -144,8 +145,30 @@ def testing_directcounter(input_img):
     img, text = display_detectionsandcountings_directcounter(input_img, countings, prob_th=0, cth = 0)
     return img, text
 
+def extractframes(vid_path):
+    clip = VideoFileClip(vid_path)
+    clip = clip.subclip(0, 10)
+    stride = int(round(clip.fps,0))/2
+    frames = []
+    for i,frame in enumerate(clip.iter_frames()):
+       if i % stride == 0:
+           frames.append(frame) 
+    return frames
+
+def processvideo(vid_path, detector): 
+    frames = extractframes(vid_path)
+    img_list = []
+    for frame in frames:
+        img, text = detector(frame)
+        img_list.append(np.asarray(img))
+    clip = ImageSequenceClip(img_list, fps=2)
+    outvid_path = "out.mp4"
+    clip.write_videofile(outvid_path)
+    return outvid_path
+
 def video_identity(video):
     vidupload2aws(video)
+    video = processvideo(vid_path, detector = testing_yolocounter)
     return video
 
 with gr.Blocks() as demo: