update

app.py

@@ -14,7 +14,7 @@ import cv2
 API_URL = "https://api-inference.huggingface.co/models/facebook/wav2vec2-base-960h"
 HF_TOKEN = os.environ["HF_TOKEN"]
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-
+video_list = []
 
 def generate_transcripts(in_video): #generate_gifs(in_video, gif_transcript):
     print("********* Inside generate_transcripts() **********")
@@ -73,9 +73,10 @@ def generate_gifs(in_video, gif_transcript, words, words_timestamp, vid_speed):
     #generated .gif image
     #gif_out, vid_out = gen_moviepy_gif(in_video, start_seconds, end_seconds)
     print(f"vid_speed from SLider is : {vid_speed}")
-    slomo_vid = gen_moviepy_gif(in_video, start_seconds, end_seconds, float(vid_speed))
     
-    return slomo_vid 
+    speededit_vids_list, concat_vid = gen_moviepy_gif(in_video, start_seconds, end_seconds, float(vid_speed), video_list)
+    
+    return concat_vid #speededit_vids_list
 
     
 #calling the hosted model
@@ -160,30 +161,43 @@ def get_gif_timestamps(giflist_indxs, words_timestamp):
 
 
 #extracting the  video and building and serving a .gif image
-def gen_moviepy_gif(in_video, start_seconds, end_seconds, vid_speed):
+def gen_moviepy_gif(in_video, start_seconds, end_seconds, vid_speed, vid_list):
   print("******** inside moviepy_gif () ***************")
   #sample
-  video_path = "./ShiaLaBeouf.mp4"
+  #video_path = "./ShiaLaBeouf.mp4"
   video = mp.VideoFileClip(in_video) 
   #video = mp.VideoFileClip(video_path) 
   
+  leftover_clip_start = video.subclip(0, int(start_seconds) + float("{:.2f}".format(1-start_seconds%1))).without_audio() #float("{:.2f}".format(1-a%1))
   final_clip = video.subclip(start_seconds, end_seconds)
-  
+  leftover_clip_end = video.subclip(int(end_seconds) + float("{:.2f}".format(1-end_seconds%1)) ).without_audio() #end=None
+
   #slowmo
   print(f"vid_speed from calling function is : {vid_speed}")
-  slomo_clip = final_clip.fx(mp.vfx.speedx, vid_speed)
-  slomo_clip.write_videofile("slomo.mp4")
+  speededit_clip = final_clip.fx(mp.vfx.speedx, vid_speed)
+  speededit_clip = speededit_clip.without_audio()
+
+  #concat
+  concatenated_clip = mp.concatenate_videoclips([leftover_clip_start, speededit_clip, leftover_clip_end])
+  concatenated_clip.write_videofile("concat.mp4")
+  
+  filename = f"speededit{len(vid_list)}"
+  speededit_clip.write_videofile(filename) #("speededit0.mp4")
+  vid_list.append(filename) #["speededit0.mp4"]
   
-  #writing to RAM
-  final_clip.write_gif("gifimage.gif") #, program='ffmpeg', tempfiles=True, fps=15, fuzz=3)
-  final_clip.write_videofile("gifimage.mp4")
+  if len(vid_list) == 1:
+      speededit_clip.write_videofile("slomo.mp4")
+    
+  #writing to RAM - gif and smaller clip
+  #final_clip.write_gif("gifimage.gif") #, program='ffmpeg', tempfiles=True, fps=15, fuzz=3)
+  #final_clip.write_videofile("gifimage.mp4")
   final_clip.close()
   #reading in a variable
-  gif_img = mp.VideoFileClip("gifimage.gif")
+  #gif_img = mp.VideoFileClip("gifimage.gif")
   #gif_vid = mp.VideoFileClip("gifimage.mp4")
   #im = Image.open("gifimage.gif")
   #vid_cap = cv2.VideoCapture('gifimage.mp4')
-  return "slomo.mp4" #"gifimage.gif", "gifimage.mp4" #im, gif_img, gif_vid, vid_cap,  #"gifimage.mp4"
+  return vid_list, "concat.mp4" #"slomo.mp4", "timelapse.mp4", #"gifimage.gif", "gifimage.mp4" #im, gif_img, gif_vid, vid_cap,  #"gifimage.mp4"
 
 
 sample_video = ['./ShiaLaBeouf.mp4']
@@ -218,27 +232,53 @@ with demo:
         #to generate and display transcriptions for input video
         text_transcript = gr.Textbox(label="Transcripts", lines = 10, interactive = True )
         
-        #Just to move dgata between function hence keeping visible false
+        #Just to move data between function hence keeping visible false
         text_words = gr.Textbox(visible=False)
         text_wordstimestamps = gr.Textbox(visible=False)
+
+    with gr.Row():
+        button_transcript = gr.Button("Generate transcripts")
+
+    #For SlowMo
+    with gr.Row():
+        #to copy paste required gif transcript / or to populate by itself on pressing the button
+        text_slomo_transcript = gr.Textbox(label="Transcripts", placeholder="Copy paste transcripts here to create SlowMo Video" , lines = 5, interactive = True ) 
         
-        #to copy paste required gif transcript / or to populate by itslef on pressing the button
-        text_gif_transcript = gr.Textbox(label="Transcripts", placeholder="Copy paste transcripts here to create GIF image" , lines = 3, interactive = True ) 
-        
-        def load_gif_text(text):
-            print("****** inside load_gif_text() ******")
-            print("text for gif is : ", text)
+        def load_slomo_text(text):
+            print("****** inside load_slomo_text() ******")
+            print("text for slomo video is : ", text)
             return text
              
-        text_transcript.change(load_gif_text, text_transcript, text_gif_transcript )
+        text_transcript.change(load_slomo_text, text_transcript, text_slomo_transcript )
         
         #out_gif = gr.Image(label="Generated GIF image") 
-        out_slomo_vid = gr.Video(label="Generated GIF image")         
+        out_slomo_vid = gr.Video(label="Generated SlowMo Video")         
         
     with gr.Row():
-        button_transcript = gr.Button("Generate transcripts")
-        button_gifs = gr.Button("Create SloMo")
+        #button_transcript = gr.Button("Generate transcripts")
+        vid_speed_slomo = gr.Slider(0.1,0.9, step=0.1)
+        button_slomo = gr.Button("Create SloMo")
+
+    #For TimeLapse
+    with gr.Row():
+        #to copy paste required gif transcript / or to populate by itself on pressing the button
+        text_timelapse_transcript = gr.Textbox(label="Transcripts", placeholder="Copy paste transcripts here to create GIF image" , lines = 5, interactive = True ) 
+        
+        def load_timelapse_text(text):
+            print("****** inside load_timelapse_text() ******")
+            print("text for timelapse video is : ", text)
+            return text
+             
+        text_transcript.change(load_timelapse_text, text_transcript, text_timelapse_transcript )
+        
+        #out_gif = gr.Image(label="Generated GIF image") 
+        out_timelapse_vid = gr.Video(label="Generated TimeLapse Video")         
         
+    with gr.Row():
+        #button_transcript = gr.Button("Generate transcripts")
+        vid_speed_timelapse = gr.Slider(0.1,0.9, step=0.1)
+        button_timelapse = gr.Button("Create TimeLapse")
+                
     with gr.Row():
         #to render video example on mouse hover/click        
         examples.render()
@@ -250,11 +290,11 @@ with demo:
         
         examples.click(load_examples, examples, input_video) 
         
-        vid_speed = gr.Slider(0.1,0.9, step=0.1)
+        #vid_speed = gr.Slider(0.1,0.9, step=0.1)
         
         
     button_transcript.click(generate_transcripts, input_video, [text_transcript, text_words, text_wordstimestamps ])
-    button_gifs.click(generate_gifs, [input_video, text_gif_transcript, text_words, text_wordstimestamps, vid_speed], out_slomo_vid )
-    
+    button_slomo.click(generate_gifs, [input_video, text_slomo_transcript, text_words, text_wordstimestamps, vid_speed_slomo], out_slomo_vid )
+    button_timelapse.click(generate_gifs, [out_slomo_vid, text_timelapse_transcript, text_words, text_wordstimestamps, vid_speed_timelapse], out_timelapse_vid )
    
 demo.launch(debug=True)