Update app.py

app.py

@@ -7,16 +7,11 @@ import cv2
 import uuid
 from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 
-
 uid=uuid.uuid4()
 if not os.path.exists(f'{uid}-frames'): os.makedirs(f'{uid}-frames')
 if not os.path.exists(f'{uid}-rembg'): os.makedirs(f'{uid}-rembg')
 
-
-
-
 load_js = """
-
 function(text_input, url_params) {
     console.log(text_input, url_params);
     const params = new URLSearchParams(window.location.search);
@@ -26,10 +21,8 @@ function(text_input, url_params) {
 """
 def rem_cv(inp):
     cap = cv2.VideoCapture(f'{inp}')
-    
     kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(3,3))
     fgbg = cv2.bgsegm.createBackgroundSubtractorGMG()
-    
     while True:
         ret, frame = cap.read()
         fgmask = fgbg.apply(frame)
@@ -49,7 +42,6 @@ def capture_function(vid=None):
     new_video_in = str(vid)
     capture = cv2.VideoCapture(new_video_in)
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
-    
     fbox=[]
     cnt=0
     frame_count1= int(frame_count)
@@ -59,16 +51,13 @@ def capture_function(vid=None):
         frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
         cv2.imwrite(f'{uid}/{i}.png',frame_ff)
         fbox.append(f'{uid}/{i}.png')
-        
     return fbox,fbox
 
-    
-
 def im_2_vid(images,fps):
     #width, height = Image.open(images[0]).size
     this_im=cv2.imread(images[0])
     height=this_im.shape[0]
-    width= this im.shape[1]
+    width= this_im.shape[1]
     print (width,height)
     size = (width, height)
     codec = cv2.VideoWriter_fourcc(*'mp4v') #DIVX, DIVD
@@ -81,7 +70,6 @@ def rem_bg(vid):
     capture = cv2.VideoCapture(new_video_in)    
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
     fps = capture.get(cv2.CAP_PROP_FPS)
-    
     fbox2=[]
     cnt=0
     frame_count1= int(frame_count)
@@ -98,7 +86,6 @@ def rem_bg(vid):
 
     out_vid = im_2_vid(fbox2, fps)
     return (fbox2,frame_num,out_vid)
-
     
 def rem_bg_og(inp):
     #transparent background .mov
@@ -111,9 +98,8 @@ def rem_bg_og(inp):
     #os.system(f'backgroundremover -i "/path/to/video.mp4" -tg -o "output.gif"')
     #output to matte background
     os.system(f'backgroundremover -i "{inp}" -mk -o "{inp}.matte.mp4"')
-
-    
     return f'{inp}.matte.mp4'
+
 def predict(text, url_params):
     mod_url=""
     mod=gr.HTML("")
@@ -157,13 +143,8 @@ def other():
     fps = capture.get(cv2.CAP_PROP_FPS)
     fbox3=[]
     frame_count1= int(frame_count)
-
     start = (int(fps*int(start_f)))
-
     capture.set(cv2.CAP_PROP_POS_FRAMES, i-1)
-
-
-    
     for i in range(int(frame_count1)):
         capture.set(cv2.CAP_PROP_POS_FRAMES, i)
         ret, frame_f = capture.read(i)
@@ -175,9 +156,6 @@ def other():
         frame_num=f'Working on {i+1} of {frame_count}'
         yield fbox2,frame_num,None    
 
-
-
-    
 with gr.Blocks() as app:
     with gr.Tab("Load"):
         with gr.Row():