Update image_display.py

image_display.py

@@ -1,6 +1,23 @@
-from PIL import Image
-
-def display_side_by_side(image1_path, image2_path, output_path):
+def add_caption(image, caption, font_size=20, font_color=(255, 0, 0), font_path=None):
+    # Create a drawing context
+    draw = ImageDraw.Draw(image)
+
+    # Load a font
+    if font_path:
+        font = ImageFont.truetype(font_path, font_size)
+    else:
+        font = ImageFont.load_default()
+
+    # Calculate text size and position
+    text_width, text_height = draw.textsize(caption, font=font)
+    width, height = image.size
+    text_x = (width - text_width) // 2
+    text_y = 10  # Adjust this value for vertical position
+
+    # Draw text on image
+    draw.text((text_x, text_y), caption, font=font, fill=font_color)
+
+def display_side_by_side_with_captions(image1_path, image2_path, output_path):
     # Open images using Pillow
     image1 = Image.open(image1_path)
     image2 = Image.open(image2_path)
@@ -22,6 +39,10 @@ def display_side_by_side(image1_path, image2_path, output_path):
     side_by_side.paste(image1, (0, 0))
     side_by_side.paste(image2, (width, 0))
 
+    # Add captions to each image
+    add_caption(image1, "Image 1", font_color=(255, 0, 0))
+    add_caption(image2, "Image 2", font_color=(255, 0, 0))
+
     # Save the side by side image to a file
     side_by_side.save(output_path)
 
@@ -29,7 +50,7 @@ def display_side_by_side(image1_path, image2_path, output_path):
 if __name__ == "__main__":
     image1_path = 'path_to_image1.jpg'
     image2_path = 'path_to_image2.jpg'
-    output_path = 'side_by_side.jpg'
-    display_side_by_side(image1_path, image2_path, output_path)
+    output_path = 'side_by_side_with_captions.jpg'
+    display_side_by_side_with_captions(image1_path, image2_path, output_path)
 
-    print(f"Images displayed side by side and saved as {output_path}")
+    print(f"Images displayed side by side with captions and saved as {output_path}")