update app.py

app.py

@@ -2,15 +2,25 @@ import gradio as gr
 from rembg import remove
 from PIL import ImageOps, ImageDraw
 
-def process_image(foreImg, flip, backImg, left, top, width, height, isLoc):
+def process_image(foreImg, flip, backImg, left, top, method, hscale, vscale, width, height, isLoc):
 	if foreImg is None:
 		return None
 
+	bgwidth, bgheight = backImg.size
 	imwidth, imheight = foreImg.size
-	if width > 0:
-		imwidth = width
-	if height > 0:
-		imheight = height
+	if method == 'horizontal scale factor':
+		scale = bgwidth * hscale / imwidth
+		imwidth = int(imwidth * scale)
+		imheight = int(imheight * scale)
+	elif method == 'vertical scale factor':
+		scale = bgheight * vscale / imheight
+		imwidth = int(imwidth * scale)
+		imheight = int(imheight * scale)
+	else:
+		if width > 0:
+			imwidth = width
+		if height > 0:
+			imheight = height
 
 	if isLoc:
 		if backImg is None:
@@ -39,6 +49,9 @@ app = gr.Interface(
 		gr.Image(label='background', type='pil'),
 		gr.Slider(maximum=4000, step=1, label='left'),
 		gr.Slider(maximum=4000, step=1, label='top'),
+		gr.Radio(['horizontal scale factor', 'vertical scale factor', 'size (width and height)'], label='size specification method', value='horizontal scale factor'),
+		gr.Slider(minimum=0.01, maximum=1.0, step=0.01, label='horizontal scale factor', value=0.5),
+		gr.Slider(minimum=0.01, maximum=1.0, step=0.01, label='vertical scale factor', value=0.5),
 		gr.Slider(maximum=4000, step=1, label='width'),
 		gr.Slider(maximum=4000, step=1, label='height'),
 		gr.Checkbox(label='check location only'),
@@ -46,7 +59,7 @@ app = gr.Interface(
 	outputs='image',
 	allow_flagging='never',
 	concurrency_limit=20,
-	examples=[['examples/foreground.jpg', False, 'examples/background.jpg', 720, 540, 256, 256, False]],
+	examples=[['examples/foreground.jpg', False, 'examples/background.jpg', 720, 540, 'size (width and height)', 0.5, 0.5, 256, 256, False]],
 	#cache_examples=False
 )
 app.launch()