Update app.py

app.py

@@ -9,10 +9,18 @@ def add_label_to_image(image, label):
     # Create a drawing context
     draw = ImageDraw.Draw(image)
     
-    # Calculate the size and position of the text
-    text_bbox = draw.textbbox((0, 0), label)
+    # Define font size and color (adjust font path for your environment)
+    font_path = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"  # Example font path
+    font_size = 60  # Larger font size for better visibility
+    try:
+        font = ImageFont.truetype(font_path, font_size)
+    except:
+        font = ImageFont.load_default()
+    
+    # Calculate the size and position of the text (aligned to the left)
+    text_bbox = draw.textbbox((0, 0), label, font=font)
     text_width, text_height = text_bbox[2] - text_bbox[0], text_bbox[3] - text_bbox[1]
-    position = ((image.width - text_width) // 2, image.height - text_height - 10)  # Centered at the bottom
+    position = (20, image.height - text_height - 20)  # Left-aligned with margin
     
     # Add a semi-transparent rectangle behind the text for better visibility
     rect_margin = 10
@@ -23,9 +31,10 @@ def add_label_to_image(image, label):
         position[1] + text_height + rect_margin,
     ]
     draw.rectangle(rect_position, fill=(0, 0, 0, 128))  # Semi-transparent black
-    draw.text(position, label, fill="white")
+    draw.text(position, label, fill="white", font=font)
     return image
 
+
 # Function to plot, label, and save an image
 def plot_and_save_agent_image(agent_image, label, save_path=None):
     # Convert AgentImage to a raw PIL Image