1

app.py

@@ -8,6 +8,7 @@ import base64
 import requests
 import moviepy.editor as mp
 from PIL import Image, ImageSequence
+import cv2
 
 
 API_URL = "https://api-inference.huggingface.co/models/facebook/wav2vec2-base-960h"
@@ -70,13 +71,13 @@ def generate_gifs(gif_transcript, words, words_timestamp):
     start_seconds, end_seconds = get_gif_timestamps(giflist_indxs, words_timestamp)
     print(f"start_seconds, end_seconds  are : ({start_seconds}, {end_seconds})")
     #generated .gif image
-    im, gif_img, gif_vid = gen_moviepy_gif(start_seconds, end_seconds)
+    im, gif_img, gif_vid, vid_cap = gen_moviepy_gif(start_seconds, end_seconds)
     im.save('./gifimage1.gif', save_all=True)
     #gif_img = gen_moviepy_gif(start_seconds, end_seconds)
     #gif_img = f"./gifimage.gif"
     #html_out = "<img src= '" + gif_img + "' alt='create a gif from video' width='100%'/>"
     #print("html out is :", html_out)        
-    return gif_vid
+    return vid_cap #gif_vid
 
     
 #calling the hosted model
@@ -189,8 +190,10 @@ def gen_moviepy_gif(start_seconds, end_seconds):
   gif_vid = mp.VideoFileClip("gifimage.mp4")
   
   im = Image.open("gifimage.gif")
+  vid_cap = cv2.VideoCapture('gifimage.mp4')
+
   print("At the very end")
-  return im, gif_img, gif_vid
+  return im, gif_img, gif_vid, vid_cap
 
 
 # showing gif