Update app.py

app.py

@@ -76,7 +76,29 @@ def rem_bg(vid):
         print (i)
         cv2.imwrite(f'{uid}-rembg/{i}.png',out)
         fbox2.append(f'{uid}-rembg/{i}.png')
-        yield fbox2
+        frame_num=f'Working on {i+1} of {frame_count}'
+        yield fbox2,frame_num
+
+
+def im_2_vid():
+    width, height = Image.open(images[0]).size
+    size = (width, height)
+    codec = cv2.VideoWriter_fourcc(*'mp4v') #DIVX, DIVD
+    if (len(sys.argv) < 2 and n > 0):
+        video = cv2.VideoWriter(vidName, codec, nfs, (W, H))
+    else:
+        video = cv2.VideoWriter(vidName, codec, nfs, size)
+    
+    #Do not open images in 'Image' as well as 'CV2', else following error occurs
+    #<built-in function imread> returned NULL without setting an error
+    for img in images:
+        #In case images are of different size, resize them to user specified values
+        if (len(sys.argv) < 2 and n > 0):
+            newImg = cv2.imread(img)
+            newImg = cv2.resize(newImg, (W, H))
+            video.write(newImg)
+        else:
+            video.write(img)
    
 
 
@@ -145,6 +167,7 @@ with gr.Blocks() as app:
                 in_vid=gr.Video()
             with gr.Column():
                 #rem_vid=gr.Video()
+                frame_num=gr.Textbox(label="Progress")
                 rem_vid=gr.Gallery(columns=6)
     with gr.Row(visible=False):
         text_input=gr.Textbox()
@@ -152,7 +175,7 @@ with gr.Blocks() as app:
         url_params=gr.JSON()
     outp_vid.change(load_video,outp_vid,[frame_count,fps])
     frame_btn.click(capture_function,[outp_vid],[frame_gal,frame_file])
-    rem_btn.click(rem_bg,[outp_vid],rem_vid)
+    rem_btn.click(rem_bg,[outp_vid],[rem_vid,frame_num])
     go_btn.click(dl,inp_url,[outp_vid,in_vid,outp_file])
     app.load(fn=predict, inputs=[text_input,url_params], outputs=[text_output,text_input],_js=load_js)
 app.launch()