Update app.py

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 import yt_dlp
 import os 
 import json
-import backgroundremover
+import rembg
+import cv2
+
 load_js = """
 function(text_input, url_params) {
     console.log(text_input, url_params);
@@ -11,8 +13,26 @@ function(text_input, url_params) {
     return [text_input, url_params]
 }
 """
+def rem_cv(inp):
+    cv2cap = 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)
+        fgmask = cv2.morphologyEx(fgmask, cv2.MORPH_OPEN, kernel)
+        yield (fgmask)
 
+        
 def rem_bg(inp):
+    out = rem_bg()
+
+
+    
+def rem_bg_og(inp):
     #transparent background .mov
     #os.system(f'backgroundremover -i "{inp}" -tv -o "{inp}.mov"')
     #video video to be overlayed output
@@ -68,7 +88,7 @@ with gr.Blocks() as app:
         text_input=gr.Textbox()
         text_output=gr.Textbox()
         url_params=gr.JSON()
-    rem_btn.click(rem_bg,in_vid,rem_vid)
+    rem_btn.click(rem_cv,in_vid,rem_vid)
     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()