Update app.py

app.py

@@ -128,10 +128,13 @@ def dl(inp):
         inp_out=inp_out.replace("/","_").replace(".","_")
         os.system(f'yt-dlp "{inp}" --trim-filenames 100 -o "{uid}/{inp_out}.mp4"')  
         out = f"{uid}/{inp_out}.mp4"
+        capture = cv2.VideoCapture(out)
+        fps = capture.get(cv2.CAP_PROP_FPS)
+        capture.release()
     except Exception as e:
         print (e)
         out = None
-    return out,out,out,out,out
+    return out,out,out,out,out,fps
 
 def trim_vid(vid,start_time,end_time):
     print (vid)
@@ -177,10 +180,11 @@ def make_gif():
     out = f"{uid}/clip_gif.gif"
     
     return out,out
-def update_speed(inp,clip_speed):
+def update_speed(inp,clip_speed,fps):
+    mod_fps=float(fps)*float(clip_speed)
     clip = VideoFileClip(inp)
-    final = clip.fx( vfx.speedx, clip_speed) 
-    final.write_videofile(f'{inp}') 
+    final = clip.fx(vfx.speedx, clip_speed) 
+    final.write_videofile(f'{inp}', fps=mod_fps) 
     return f'{inp}'
 with gr.Blocks() as app:
     with gr.Tab("Load"):
@@ -203,6 +207,7 @@ with gr.Blocks() as app:
                 trim_btn=gr.Button("Trim")
                 out_trim=gr.Video(format="mp4")
                 hid_box = gr.Textbox(visible=True)
+                hid_fps = gr.Textbox(visible=True)
 
     with gr.Tab("Frames"):
         with gr.Row():
@@ -244,12 +249,12 @@ with gr.Blocks() as app:
     def echo_fn(inp):
         return inp 
     #outp_vid.change(echo_fn,outp_vid,[out_trim])
-    speed_btn.click(update_speed,[hid_box,clip_speed],outp_vid)
+    speed_btn.click(update_speed,[hid_box,clip_speed,hid_fps],outp_vid)
     gif_btn.click(make_gif,None,[gif_show,gif_file])
     trim_btn.click(trim_vid,[hid_box,start_f,end_f],[out_trim,in_vid,trim_count])
     outp_vid.change(load_video,outp_vid,[frame_count,fps,end_f]).then(trim_vid,[hid_box,start_f,end_f],[out_trim,in_vid,trim_count])
     frame_btn.click(capture_function,[out_trim],[frame_gal,frame_file,frame_stat])
     rem_btn.click(rem_bg,[out_trim],[rem_vid,frame_num,rem_bg_vid])
-    go_btn.click(dl,inp_url,[outp_vid,outp_file,out_trim,hid_box])
+    go_btn.click(dl,inp_url,[outp_vid,outp_file,out_trim,hid_box,hid_fps])
     app.load(fn=predict, inputs=[text_input,url_params], outputs=[text_output,text_input],_js=load_js)
 app.launch()