Update image_display.py

image_display.py

@@ -1,4 +1,4 @@
-def add_caption(image, caption, font_size=20, font_color=(255, 0, 0), font_path=None):
+def add_text(image, text, font_size=20, font_color=(255, 0, 0), font_path=None):
     # Create a drawing context
     draw = ImageDraw.Draw(image)
 
@@ -9,13 +9,13 @@ def add_caption(image, caption, font_size=20, font_color=(255, 0, 0), font_path=
         font = ImageFont.load_default()
 
     # Calculate text size and position
-    text_width, text_height = draw.textsize(caption, font=font)
+    text_width, text_height = draw.textsize(text, 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)
+    draw.text((text_x, text_y), text, font=font, fill=font_color)
 
 def display_side_by_side_with_captions(image1_path, image2_path, output_path):
     # Open images using Pillow