Update generate_dataset.py

generate_dataset.py

@@ -4,18 +4,22 @@ from PIL import Image
 
 
 
-def merge_images_horizontally(img1, img2):
+def merge_images_horizontally(img1, img2, logo_path):
     
+    logo = Image.open(logo_path)
+    logo = logo.resize((100, 20))
+    img2.paste(logo, (0, 0))
+
     img1_width, img1_height = img1.size
     img2_width, img2_height = img2.size
 
     merged_width = img1_width + img2_width
     merged_height = max(img1_height, img2_height)
-    merged_image = Image.new('RGBA', (merged_width + 20, merged_height))
+    merged_image = Image.new('RGBA', (merged_width + 60, merged_height))
 
    
-    merged_image.paste(img1, (0, 0))
-    merged_image.paste(img2, (img1_width + 20, 0))
+    merged_image.paste(img1, (15, 0))
+    merged_image.paste(img2, (img1_width + 30, 0))
   
     return merged_image