fix image geneartion

app.py

@@ -62,24 +62,28 @@ def maskHead(input):
     return output_image
 
 def random_positioning(input, output_size=(1024, 1024)):
-    if input is None:
-        raise ValueError("Impossible to load image")
+    background = cv2.imread("default.jpeg")
+    if background is None:
+        raise ValueError("Unable to load background image")
+
+    background = cv2.resize(background, output_size, interpolation=cv2.INTER_AREA)
     
+    if input_image is None:
+        raise ValueError("Unable to load input image")
+
     scale_factor = random.uniform(0.5, 1.0)
+    new_size = (int(input_image.shape[1] * scale_factor), int(input_image.shape[0] * scale_factor))
     
-    new_size = (int(input.shape[1] * scale_factor), int(input.shape[0] * scale_factor))
-    
-    resized_image = cv2.resize(input, new_size, interpolation=cv2.INTER_AREA)
-    
-    background = np.zeros((output_size[1], output_size[0], 3), dtype=np.uint8)
-    
+    resized_image = cv2.resize(input_image, new_size, interpolation=cv2.INTER_AREA)
+
+    if background.shape[2] != resized_image.shape[2]:
+        raise ValueError("Input image and background image must have the same number of channels")
+
     x_offset = random.randint(0, output_size[0] - new_size[0])
     y_offset = random.randint(0, output_size[1] - new_size[1])
-    
+    # Sobrepor a imagem redimensionada na imagem de fundo
     background[y_offset:y_offset+new_size[1], x_offset:x_offset+new_size[0]] = resized_image
-    background = np.clip(background, 0, 255) 
-    background = background.astype(np.uint8)
-
+    
     return background
 
 
@@ -112,7 +116,7 @@ def execute(image, prompt):
   img = cv2.imread(image)
   img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
 
-  imgs = [ random_positioning(img)]
+  imgs = [ random_positioning(img), random_positioning(img)]
 
   pipe.load_lora_weights("XLabs-AI/flux-RealismLora", weight_name='lora.safetensors')
   response = []
@@ -154,7 +158,6 @@ def execute(image, prompt):
 description = "This is an unofficial implementation of the ip face adapter for FLUX DEV and does not explicitly follow the ip face model, I created a wrapper with inpaint and mediapipe, I like using Fake IP Adapter"
 title = "Flux IP Face Adapter"
 iface = gr.Interface(
-    theme=gr.themes.Glass(),
     fn=execute,
     description=description,
     title=title,