image murge fixed

app.py

@@ -1,40 +1,29 @@
 import gradio as gr
 from PIL import Image
+import cv2
 
 def merge_images(image1, image2, method="concatenate"):
-  """
-  Merges two images using the specified method.
+    """
+    Merges two images using the specified method.
 
-  Args:
-      image1 (PIL.Image): The first image.
-      image2 (PIL.Image): The second image.
-      method (str, optional): The method for merging. Defaults to "concatenate".
-          Supported methods:
-              - "concatenate": Concatenates the images horizontally.
-              - "average": Creates an average blend of the two images.
-              - "custom": Allows for custom logic using (image1, image2) as input.
+    Args:
+        image1 (PIL.Image): The first image.
+        image2 (PIL.Image): The second image.
+        method (str, optional): The method for merging. Defaults to "concatenate".
+            Supported methods:
+                - "concatenate": Concatenates the images horizontally.
+                - "average": Creates an average blend of the two images.
+                - "custom": Allows for custom logic using (image1, image2) as input.
 
-  Returns:
-      PIL.Image: The merged image.
-  """
-
-  if method == "concatenate":
-    # Concatenate images horizontally
-    width, height = image1.size
-    merged_image = Image.new(image1.mode, (width * 2, height))
-    merged_image.paste(image1, (0, 0))
-    merged_image.paste(image2, (width, 0))
-  elif method == "average":
-    # Create an average blend
-    merged_image = Image.blend(image1, image2, 0.5)
-  elif method == "custom":
-    # Implement your custom logic here, using image1 and image2
-    # This allows for more advanced merging techniques
-    pass
-  else:
-    raise ValueError(f"Unsupported merge method: {method}")
-
-  return merged_image
+    Returns:
+        PIL.Image: The merged image.
+    """
+    img1 = cv2.resize(image1, (512,512))
+    img2 = cv2.resize(image2, (512,512))
+    img = img1*0.5 + img2*0.5
+    img = img.astype('uint8')
+  
+    return img
 
 # Define Gradio interface
 interface = gr.Interface(