test fps reduction function

main_noweb.py

@@ -46,8 +46,8 @@ def check_fps(video):
     return nframes, fps
 
 
-def get_frames(video, fps=30, height=512, width=512):
-# extract the file name and extension
+def get_frames(video, fps=20, height=512, width=512):
+    
     split_tup = os.path.splitext(video)
 
     file_name = split_tup[0]
@@ -57,8 +57,12 @@ def get_frames(video, fps=30, height=512, width=512):
         print("Converting to mp4")
         video = file_name+".mp4"
 
+
     clip = moviepy.VideoFileClip(video)
     
+    if clip.duration > 10:
+            raise gr.Error("Please provide or record a video shorter than 10 seconds...")
+
     #check fps
     if clip.fps > fps:
         print("vide rate is over 15, resetting to 15")
@@ -67,10 +71,8 @@ def get_frames(video, fps=30, height=512, width=512):
     else:
         print("video rate is OK")
         #clip_resized = clip.resize(height=height)
-        clip.write_videofile(video, fps=clip.fps)
+        #clip.write_videofile(video, fps=clip.fps)
     
-    print("video resized to 512 height")
-  
 
 def check_extension(video):
     split_tup = os.path.splitext(video)
@@ -194,9 +196,9 @@ def UI():
                                 submit_hand_file = gr.Button("Make 2d hand estimation")
 
                     with gr.Row():
-                        video_output1 = gr.Video(height=512,  label = "Estimate human 2d poses", show_label=True)
-                        video_output2 = gr.Video(height=512,  label = "Estimate human 3d poses", show_label=True)
-                        video_output3 = gr.Video(height=512,  label = "Estimate human hand poses", show_label=True)
+                        video_output1 = gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True)
+                        video_output2 = gr.PlayableVideo(height=512,  label = "Estimate human 3d poses", show_label=True)
+                        video_output3 = gr.PlayableVideo(height=512,  label = "Estimate human hand poses", show_label=True)
                     
                     gr.Markdown("Download the .json file that contains the keypoint positions for each frame in the video.")
                     jsonoutput = gr.File(file_types=[".json"])