Update app.py

app.py

@@ -126,7 +126,27 @@ def infer(url_in,interpolation,fps_output):
 
     files = final_vid
 
-    return final_vid, files
+    depth_map = cv2.VideoCapture(final_vid)
+
+    ret, frame1 = depth_map.read()
+    prvs = cv2.cvtColor(frame1, cv2.COLOR_RGB2GRAY)
+    res = np.zeros_like(prvs)
+    while(1):
+      ret, frame2 = depth_map.read()
+      if not ret:
+      print('No frames grabbed!')
+      break
+      next = cv2.cvtColor(frame2, cv2.COLOR_RGB2GRAY)
+      flow = cv2.calcOpticalFlowFarneback(prvs, next, None, 0.5, 3, 15, 3, 5, 1.2, 0)
+      alpha = 0.5
+      beta = (1.0 - alpha)
+      res = cv2.addWeighted(flow, alpha, res, beta, 0.0)
+      prvs = next
+
+    cv2.imwrite('opticalfb.png', res)
+    cv2.destroyAllWindows()
+
+    return final_vid, files, 'opticalfb.png'
 
 
 def logscale(linear):
@@ -175,15 +195,16 @@ with gr.Blocks() as demo:
             with gr.Column():
                 video_output = gr.Video()
                 file_output = gr.File()
+                depth_output = gr.Image(image_mode="L")
     
     gr.Examples(
         examples=[["./examples/streetview.mp4", 1, 1]],
         fn=infer,
         inputs=[url_input,interpolation_slider,fps_output_slider],
-        outputs=[video_output,file_output],
+        outputs=[video_output,file_output,depth_output],
         cache_examples=True
     )
     
-    submit_btn.click(fn=infer, inputs=[url_input,interpolation_slider,fps_output_slider], outputs=[video_output, file_output])
+    submit_btn.click(fn=infer, inputs=[url_input,interpolation_slider,fps_output_slider], outputs=[video_output, file_output, depth_output])
 
 demo.launch()