Update app.py

app.py

@@ -290,7 +290,11 @@ def process_generate(fore, back, prompt, intensity, mode, refprompt, isrmbg, mod
         image_width, image_height = resize_to_fit((768,768), (image_width, image_height))
         fore.resize(max_size)
 
-    
+
+    if back is not None:
+        fore = Image.alpha_composite(back.resize(fore.size).convert('RGBA'), fore)
+        print(f"Background Merged!")
+        mode = "refiner"
         
     
     forestr = image_to_base64(fore.convert("RGBA"))    
@@ -321,6 +325,10 @@ def process_generate(fore, back, prompt, intensity, mode, refprompt, isrmbg, mod
         "refprompt" : " transparent glass "    
     }
     '''    
+    if back is not None:
+        fore = rmbgfore.convert("RGBA")
+        print(f"Background removed!")
+        
     image = process(data, api_endpoint, api_key)
 
     return image