video forward to back

app.py

@@ -269,11 +269,24 @@ def encode_img_fn(img, noise):
     return "success", imgs[0], img_out, w, get_thumbnail(img_out)
 
 
+def gen_video2(self, w1, w2, noise, path, frame_num=10):
+        video = imageio.get_writer(path, mode='I', fps=frame_num // 2, codec='libx264', bitrate='16M')
+        lin = np.linspace(0, 1, frame_num)
+        for i in range(0, frame_num):
+            img = self.get_img(((1 - lin[i]) * w1) + (lin[i] * w2), noise)
+            video.append_data(img)
+        for i in range(frame_num, 0, -1):
+            img = self.get_img(((1 - lin[i]) * w1) + (lin[i] * w2), noise)
+            video.append_data(img)
+    video.close()
+
+
 def gen_video_fn(w1, w2, noise, frame):
     if w1 is None or w2 is None:
         return None
-    model.gen_video(w1, w2, noise, "video.mp4", int(frame))
-    return "video.mp4"
+#    model.gen_video(w1, w2, noise, "video.mp4", int(frame))
+    model.gen_video2(w1, w2, noise, "video.mp4", int(frame))
+    return "videoFB.mp4"
 
 
 if __name__ == '__main__':