Update app.py

app.py

@@ -13,6 +13,7 @@ def get_frames(video_in):
     clip = mp.VideoFileClip(video_in)
     clip_resized = clip.resize(height=512)
     clip_resized.write_videofile("video_resized.mp4")
+    print("video resized to 512 height")
     
     # Opens the Video file with CV2
     cap= cv2.VideoCapture("video_resized.mp4")
@@ -29,11 +30,11 @@ def get_frames(video_in):
     
     cap.release()
     cv2.destroyAllWindows()
-
+    print("broke the video into frames")
     return frames, fps
 
 def create_video(frames, fps):
-    
+    print("building video result")
     clip = ImageSequenceClip(frames, fps=fps)
     clip.write_videofile("movie.mp4", fps=fps)
     
@@ -41,14 +42,14 @@ def create_video(frames, fps):
 
 
 def infer(prompt,video_in, seed_in, trim_value):
-    
+    print(prompt)
     break_vid = get_frames(video_in)
     
     frames_list= break_vid[0]
     fps = break_vid[1]
     n_frame = int(trim_value*fps)
     result_frames = []
-    
+    print("set stop frames to: " + n_frame)
     for i in frames_list[0:n_frame]:
         pix2pix_img = pix2pix(prompt,5.5,1.5,i,15,"",512,512,seed_in,fn_index=0)
         images = [os.path.join(pix2pix_img[0], img) for img in os.listdir(pix2pix_img[0])]
@@ -56,7 +57,7 @@ def infer(prompt,video_in, seed_in, trim_value):
         print("frame " + i + ": done;")
 
     final_vid = create_video(result_frames, fps)
-    
+    print("finished !")
     return final_vid
 
 title = """