update

app.py

@@ -62,22 +62,16 @@ def generate_gifs(gif_transcript, words, words_timestamp):
     print(f"type of words is :{type(words)}")
     print(f"length of words is :{len(words)}")
     print(f"giflist is :{giflist}")
-    #print(f"haystack and needle function returns value as : {list(get_gif_word_indexes(words, giflist))}")
-    #indx_tmp = [num for num in get_gif_word_indexes(words, giflist)]
-    #print(f"index temp is : {indx_tmp}")
+
     giflist_indxs = list(list(get_gif_word_indexes(words, giflist))[0])
     print(f"giflist_indxs is : {giflist_indxs}")
     #getting start and end timestamps for a gif video
     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, vid_cap, gif_video_out = gen_moviepy_gif(start_seconds, end_seconds)
+    gif_out = 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_video_out #vid_cap #gif_vid
+    return gif_out 
 
     
 #calling the hosted model
@@ -124,14 +118,11 @@ def get_word_timestamps(timestamps):
 #getting index of gif words in main transcript
 def get_gif_word_indexes(total_words_list, gif_words_list):  
     if not gif_words_list:
-        print("THIS IS 1")
         return
     # just optimization
     COUNT=0
     lengthgif_words_list = len(gif_words_list)
-    print("THIS IS 2")
     firstgif_words_list = gif_words_list[0]
-    print("THIS IS 3")
     
     print(f"total_words_list is :{total_words_list}")
     print(f"length of total_words_list is :{len(total_words_list)}")
@@ -140,12 +131,8 @@ def get_gif_word_indexes(total_words_list, gif_words_list):
     
     for idx, item in enumerate(total_words_list):
         COUNT+=1
-        #print("COUNT IS :", COUNT)
         if item == firstgif_words_list:
-            print("THIS IS 5")
             if total_words_list[idx:idx+lengthgif_words_list] == gif_words_list:
-                print("THIS IS 6")
-                print(f"value 1 is: {range(idx, idx+lengthgif_words_list)}")
                 print(f"value of tuple is : {tuple(range(idx, idx+lengthgif_words_list))}")
                 yield tuple(range(idx, idx+lengthgif_words_list))
 
@@ -153,13 +140,11 @@ def get_gif_word_indexes(total_words_list, gif_words_list):
 #getting start and end timestamps for gif transcript
 def get_gif_timestamps(giflist_indxs, words_timestamp):
   print(f"******** Inside get_gif_timestamps() **********")
-  #giflist_indxs = list(list(get_gif_word_indexes(words, giflist))[0])
   min_idx = min(giflist_indxs)
   max_idx = max(giflist_indxs)
   print(f"min_idx is :{min_idx}")
   print(f"max_idx is :{max_idx}")
   
-  print(f"type of words_timestamp is :{type(words_timestamp)}")
   gif_words_timestamp = words_timestamp[min_idx : max_idx+1]
   print(f"words_timestamp is :{words_timestamp}")
   print(f"gif_words_timestamp is :{gif_words_timestamp}")
@@ -174,26 +159,22 @@ def get_gif_timestamps(giflist_indxs, words_timestamp):
 def gen_moviepy_gif(start_seconds, end_seconds):
   print("******** inside moviepy_gif () ***************")
   video_path = "./ShiaLaBeouf.mp4"
-  video = mp.VideoFileClip(video_path) #.resize(0.3)
+  video = mp.VideoFileClip(video_path) 
   final_clip = video.subclip(start_seconds, end_seconds)
-  #final_clip.write_videofile("gifimage.mp4")
-  print("I am here now")
 
-  #gifclip = VideoFileClip("gifimage.mp4")
-  final_clip.write_gif("./gifimage.gif") #, program='ffmpeg', tempfiles=True, fps=15, fuzz=3)
+  final_clip.write_gif("gifimage.gif") #, program='ffmpeg', tempfiles=True, fps=15, fuzz=3)
   final_clip.close()
   
-  print("pretty good")
-  gif_img = mp.VideoFileClip("gifimage.gif")
-  print(gif_img)
-  gif_img.write_videofile("gifimage.mp4")
-  gif_vid = mp.VideoFileClip("gifimage.mp4")
+  #gif_img = mp.VideoFileClip("gifimage.gif")
+  #print(gif_img)
+  #gif_img.write_videofile("gifimage.mp4")
+  #gif_vid = mp.VideoFileClip("gifimage.mp4")
   
-  im = Image.open("gifimage.gif")
-  vid_cap = cv2.VideoCapture('gifimage.mp4')
+  #im = Image.open("gifimage.gif")
+  #vid_cap = cv2.VideoCapture('gifimage.mp4')
 
   print("At the very end")
-  return im, gif_img, gif_vid, vid_cap, "gifimage.gif" #"gifimage.mp4"
+  return "gifimage.gif" #im, gif_img, gif_vid, vid_cap,  #"gifimage.mp4"
 
 
 # showing gif
@@ -229,16 +210,8 @@ with demo:
         button_transcript = gr.Button("Generate transcripts")
         button_gifs = gr.Button("Create Gif")
         
-        #def load_gif():
-        #    print("****** inside load_gif() ******")
-        #    #created embedding  width='560' height='315' 
-        #    html_out = "<img src='./gifimage.gif' />"
-        #    print(f"html output is : {html_out}")
-        #    return 
     with gr.Row():
-        #out_gif = gr.HTML(label="Generated GIF from transcript selected", show_label=True)
-        #gr.Markdown(""" [] """)
-        out_gif = gr.Image() #gr.Video()
+        out_gif = gr.Image() 
         
     button_transcript.click(generate_transcripts, input_video, [text_transcript, text_words, text_wordstimestamps ])
     button_gifs.click(generate_gifs, [text_gif_transcript, text_words, text_wordstimestamps], out_gif )