Moved BG to API site

app.py

@@ -13,7 +13,6 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-
 def get_mask(img_in):
 
     # print(f"{img_in=}")
@@ -33,34 +32,32 @@ def get_mask(img_in):
         if 'error' in response.json():
             print(f"ERROR: {response.json()['error']}")
             raise gr.Error(response.json()['error'])
-            return (None, None)
 
         result = response.json()
         # print('Result:', result)
 
-        all_bgs = list(Path("examples").glob("*.jpg"))
+        # all_bgs = list(Path("examples").glob("*.jpg"))
 
-        bg_img = Image.open(choice(all_bgs)).convert(
-            "RGBA").resize(img_in.size)
+        # bg_img = Image.open(choice(all_bgs)).convert(
+        #     "RGBA")
 
-        bg_img = ImageOps.fit(
-            bg_img,
-            img_in.size,
-            Image.ANTIALIAS
-        )
+        # bg_img = ImageOps.fit(
+        #     bg_img,
+        #     img_in.size,
+        #     Image.ANTIALIAS
+        # )
 
         mask = Image.open(io.BytesIO(base64.b64decode(
             result['mask']))).resize(img_in.size)
 
-        img_in = ImageOps.autocontrast(img_in, cutoff=0.1).convert("RGBA")
-        img_in.putalpha(mask)
+        # img_in = ImageOps.autocontrast(img_in, cutoff=0.1).convert("RGBA")
+        # img_in.putalpha(mask)
 
-        img_in = Image.alpha_composite(bg_img, img_in)
+        # img_in = Image.alpha_composite(bg_img, img_in)
 
-        return (img_in, result['emotion'])
+        return (mask, result['emotion'])
     else:
         raise gr.Error(response.text)
-        return (None, None)
 
 
 footer = r"""
@@ -71,7 +68,7 @@ DEMO FOR BG REMOVAL
 </center>
 """
 
-with gr.Blocks(title="Face Shine") as app:
+with gr.Blocks(title="ARKA + TikTok Demo") as app:
     gr.HTML("<center><h1>ARKA Remove Background</h1></center>")
     with gr.Row():
         with gr.Column():