from PIL import Image, ImageDraw, ImageFont
import os

img = Image.open('/tmp/workspace/usg-labels/original.png').convert('RGBA')
overlay = Image.new('RGBA', img.size, (0, 0, 0, 0))
draw = ImageDraw.Draw(overlay)

W, H = img.size  # 722 x 322

# Try to load a decent font, fallback to default
try:
    font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 13)
    font_small = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 11)
except:
    font = ImageFont.load_default()
    font_small = font

YELLOW = (255, 230, 0, 255)
WHITE  = (255, 255, 255, 255)
CYAN   = (0, 220, 255, 255)
GREEN  = (0, 255, 120, 255)
RED    = (255, 80, 80, 255)
ORANGE = (255, 160, 40, 255)
BLACK_T= (0, 0, 0, 180)

def draw_label(draw, text, tip_xy, label_xy, color=YELLOW, font=font):
    """Draw arrow from tip_xy to label_xy, then text."""
    tx, ty = tip_xy
    lx, ly = label_xy
    # arrow line
    draw.line([tx, ty, lx, ly], fill=color, width=2)
    # arrowhead (small dot at tip)
    r = 3
    draw.ellipse([tx-r, ty-r, tx+r, ty+r], fill=color)
    # text background
    bbox = draw.textbbox((lx, ly), text, font=font)
    pad = 3
    draw.rectangle([bbox[0]-pad, bbox[1]-pad, bbox[2]+pad, bbox[3]+pad], fill=BLACK_T)
    draw.text((lx, ly), text, fill=color, font=font)

# ── STRUCTURE POSITIONS (tuned to 722x322 image) ──────────────────────────────
# The image shows a longitudinal neck USG. From analysis:
# Top ~0-15px : skin line (thin bright)
# ~15-40px    : subcutaneous tissue
# ~40-65px    : strap muscles (thin hypoechoic band)
# Left side (~x=120-230, y=65-130): cricoid cartilage (large hypoechoic dome)
# Right side (~x=300-600, y=70-130): tracheal rings (string of pearls)
# ~y=145-160 : Air-Mucosa interface (bright hyperechoic line)
# ~y=160+    : Posterior acoustic shadowing

# 1. Skin line - top bright line
draw_label(draw, "Skin", (60, 8), (70, 0), WHITE, font_small)

# 2. Subcutaneous tissue
draw_label(draw, "Subcutaneous tissue", (160, 22), (170, 14), CYAN, font_small)

# 3. Strap muscles
draw_label(draw, "Strap muscles\n(Sternohyoid/Sternothyroid)", (90, 52), (10, 38), GREEN, font_small)

# 4. Cricoid cartilage (left, large hypoechoic)
draw_label(draw, "Cricoid cartilage", (155, 95), (5, 120), YELLOW, font)

# 5. Tracheal ring 1
draw_label(draw, "Tracheal ring (T1)", (310, 95), (330, 72), ORANGE, font_small)

# 6. Tracheal ring 2
draw_label(draw, "T2", (420, 95), (425, 72), ORANGE, font_small)

# 7. Tracheal ring 3
draw_label(draw, "T3", (530, 95), (535, 72), ORANGE, font_small)

# 8. Air-Mucosa interface
draw_label(draw, "Air-Mucosa (A-M) interface\n(hyperechoic)", (400, 155), (430, 178), RED, font_small)

# 9. Posterior acoustic shadowing
draw_label(draw, "Posterior acoustic shadowing", (400, 240), (420, 252), (180, 180, 255, 255), font_small)

# 10. "String of pearls" sign label
draw.rectangle([295, 55, 590, 68], fill=BLACK_T)
draw.text((300, 56), '"String of pearls" sign', fill=ORANGE, font=font_small)

# Orientation labels
draw.rectangle([0, H-22, 90, H], fill=BLACK_T)
draw.text((4, H-20), "◄ Superior", fill=WHITE, font=font_small)
draw.rectangle([W-100, H-22, W, H], fill=BLACK_T)
draw.text((W-96, H-20), "Inferior ►", fill=WHITE, font=font_small)

# Composite
result = Image.alpha_composite(img, overlay).convert('RGB')
result.save('/tmp/workspace/usg-labels/labeled_usg.jpg', quality=95)
print("Saved:", result.size)

from PIL import Image, ImageDraw, ImageFont

img = Image.open('/tmp/workspace/usg-labels2/original.png').convert('RGBA')
overlay = Image.new('RGBA', img.size, (0, 0, 0, 0))
draw = ImageDraw.Draw(overlay)

W, H = img.size  # 823 x 531

try:
    font      = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 14)
    font_sm   = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 12)
    font_tiny = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 11)
except:
    font = font_sm = font_tiny = ImageFont.load_default()

# Color palette
YELLOW  = (255, 230, 0,   255)
WHITE   = (255, 255, 255, 255)
CYAN    = (0,   220, 255, 255)
GREEN   = (60,  255, 100, 255)
RED     = (255,  80,  80, 255)
ORANGE  = (255, 165,  40, 255)
MAGENTA = (255,  80, 220, 255)
LBLUE   = (130, 180, 255, 255)
BLACK_T = (0,     0,   0, 180)

def label(text, tip, lpos, color, fnt=font_sm, line_width=2):
    tx, ty = tip
    lx, ly = lpos
    # draw line
    draw.line([tx, ty, lx, ly], fill=color, width=line_width)
    # dot at tip
    r = 4
    draw.ellipse([tx-r, ty-r, tx+r, ty+r], fill=color)
    # text background
    lines = text.split('\n')
    # measure all lines
    max_w = max(draw.textlength(l, font=fnt) for l in lines)
    line_h = fnt.size + 2
    total_h = line_h * len(lines)
    pad = 3
    draw.rectangle([lx-pad, ly-pad, lx+max_w+pad, ly+total_h+pad], fill=BLACK_T)
    for i, line in enumerate(lines):
        draw.text((lx, ly + i*line_h), line, fill=color, font=fnt)

# ── STRUCTURE LAYOUT (823 x 531 image, transverse neck view) ──────────────────
# Based on analysis:
#   Centre (x≈370-430, y≈100-380): Trachea/cricoid - large dark hypoechoic dome
#   Left lobe thyroid (x≈130-280, y≈150-330): medium hypoechoic oval
#   Right lobe thyroid (x≈530-680, y≈150-280): small-medium hypoechoic oval
#   Right side circle (x≈620-680, y≈280-360): carotid artery (ring-like)
#   Top band: skin + subcutaneous tissue + strap muscles
#   Deep to trachea: posterior acoustic shadowing
#   Isthmus: thin band connecting thyroid lobes across midline, anterior to trachea

# 1. Skin line (top)
label("Skin", (200, 12), (210, 2), WHITE, font_tiny)

# 2. Subcutaneous tissue
label("Subcutaneous tissue", (350, 28), (355, 15), CYAN, font_tiny)

# 3. Strap muscles (thin band ~y=45-80)
label("Strap muscles\n(Sternohyoid/Sternothyroid)", (200, 62), (10, 48), GREEN, font_tiny)

# 4. Left lobe of thyroid
label("Left lobe\n(Thyroid)", (190, 230), (10, 215), YELLOW, font_sm)

# 5. Right lobe of thyroid
label("Right lobe\n(Thyroid)", (600, 210), (635, 175), YELLOW, font_sm)

# 6. Isthmus (thin strip connecting lobes ~y=150, across trachea anterior surface)
label("Isthmus of thyroid", (380, 135), (385, 10), ORANGE, font_sm)

# 7. Trachea / Cricoid cartilage (large central dark dome)
label("Trachea\n(Cricoid ring)", (385, 260), (430, 290), RED, font_sm)

# 8. Air-Mucosa interface (bright arc at bottom of tracheal lumen)
label("Air-Mucosa (A-M)\ninterface", (390, 370), (420, 385), MAGENTA, font_tiny)

# 9. Posterior acoustic shadowing
label("Posterior acoustic\nshadowing", (390, 445), (400, 455), LBLUE, font_tiny)

# 10. Carotid artery (right side - concentric ring structure)
label("Common carotid\nartery (CCA)", (648, 320), (660, 340), (255, 120, 120, 255), font_tiny)

# 11. Depth markers (yellow L markers on left side of image)
label("Depth markers\n(1 cm, 2 cm)", (20, 290), (25, 295), (200, 255, 80, 255), font_tiny)

# Orientation label
draw.rectangle([0, H-22, 120, H], fill=BLACK_T)
draw.text((4, H-20), "Transverse (axial) view", fill=WHITE, font=font_tiny)

# Composite and save
result = Image.alpha_composite(img, overlay).convert('RGB')
out = '/tmp/workspace/usg-labels2/labeled_usg_neck.jpg'
result.save(out, quality=96)
print("Saved:", out, result.size)

Successfully replaced 1 instance(s) of the string in '/tmp/workspace/usg-labels2/label.py'

Successfully replaced 1 instance(s) of the string in '/tmp/workspace/usg-labels2/label.py'

from PIL import Image, ImageDraw, ImageFont

img = Image.open('/tmp/workspace/usg-larynx/original.png').convert('RGBA')
overlay = Image.new('RGBA', img.size, (0, 0, 0, 0))
draw = ImageDraw.Draw(overlay)

W, H = img.size  # 612 x 527

try:
    font_bold = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 13)
    font_sm   = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 12)
    font_tiny = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 11)
except:
    font_bold = font_sm = font_tiny = ImageFont.load_default()

YELLOW  = (255, 230,   0, 255)
WHITE   = (255, 255, 255, 255)
CYAN    = (  0, 220, 255, 255)
GREEN   = ( 60, 255, 100, 255)
RED     = (255,  80,  80, 255)
ORANGE  = (255, 165,  40, 255)
MAGENTA = (255,  80, 220, 255)
LBLUE   = (130, 180, 255, 255)
PINK    = (255, 160, 180, 255)
BLACK_T = (  0,   0,   0, 185)

def label(text, tip, lpos, color, fnt=font_sm, lw=2):
    tx, ty = tip
    lx, ly = lpos
    draw.line([tx, ty, lx, ly], fill=color, width=lw)
    r = 4
    draw.ellipse([tx-r, ty-r, tx+r, ty+r], fill=color)
    lines = text.split('\n')
    max_w = max(draw.textlength(l, font=fnt) for l in lines)
    lh = fnt.size + 3
    pad = 3
    draw.rectangle([lx-pad, ly-pad, lx+max_w+pad, ly+lh*len(lines)+pad], fill=BLACK_T)
    for i, line in enumerate(lines):
        draw.text((lx, ly + i*lh), line, fill=color, font=fnt)

# ── Image is 612 x 527 ────────────────────────────────────────────────────
# Transverse USG of larynx at level of TRUE VOCAL CORDS
#
# Structure positions (from AI analysis, % → pixels):
#   Skin/subcut:              y≈5%  → y≈26
#   Strap muscles:            y≈15% → y≈79
#   Thyroid cart apex:        x≈42%,y≈24% → (257,126)
#   Right thyroid lamina:     x≈22%,y≈42% → (135,221)
#   Left thyroid lamina:      x≈72%,y≈42% → (440,221)
#   Right true vocal cord:    x≈32%,y≈45% → (196,237)
#   Left true vocal cord:     x≈62%,y≈48% → (379,253)
#   Right arytenoid:          x≈38%,y≈42% → (232,221)
#   Left arytenoid:           x≈62%,y≈58% → (379,306)
#   Air-mucosa/glottis:       x≈45%,y≈55% → (275,290)
#   Posterior shadow:         x≈45%,y≈80% → (275,422)

# 1. Skin
label("Skin", (200, 10), (210, 2), WHITE, font_tiny)

# 2. Subcutaneous tissue
label("Subcutaneous tissue", (340, 22), (345, 10), CYAN, font_tiny)

# 3. Strap muscles
label("Strap muscles", (130, 58), (10, 44), GREEN, font_tiny)

# 4. Thyroid cartilage - anterior commissure / apex (inverted V shape at top centre)
label("Thyroid cartilage\n(anterior commissure)", (257, 120), (265, 88), YELLOW, font_sm)

# 5. Right thyroid cartilage lamina
label("Thyroid cartilage\n(Rt. lamina)", (130, 210), (5, 195), YELLOW, font_tiny)

# 6. Left thyroid cartilage lamina
label("Thyroid cartilage\n(Lt. lamina)", (445, 218), (455, 200), YELLOW, font_tiny)

# 7. Right true vocal cord (vocalis muscle)
label("Right true\nvocal cord", (196, 237), (10, 270), ORANGE, font_sm)

# 8. Left true vocal cord
label("Left true\nvocal cord", (382, 252), (440, 270), ORANGE, font_sm)

# 9. Right arytenoid cartilage
label("Arytenoid\n(Rt.)", (232, 218), (232, 165), PINK, font_tiny)

# 10. Left arytenoid cartilage
label("Arytenoid\n(Lt.)", (379, 300), (520, 295), PINK, font_tiny)

# 11. Air-mucosa interface / Rima glottidis
label("Air-Mucosa interface\n(Rima glottidis)", (275, 288), (285, 310), RED, font_sm)

# 12. Posterior acoustic shadowing
label("Posterior acoustic\nshadowing", (275, 420), (285, 435), LBLUE, font_tiny)

# 13. Paraglottic space (lateral dark areas flanking cords)
label("Paraglottic\nspace (Rt.)", (155, 255), (5, 320), MAGENTA, font_tiny)
label("Paraglottic\nspace (Lt.)", (455, 255), (460, 320), MAGENTA, font_tiny)

# Orientation label
draw.rectangle([0, H-22, 155, H], fill=BLACK_T)
draw.text((4, H-20), "Transverse - Vocal cord level", fill=WHITE, font=font_tiny)

result = Image.alpha_composite(img, overlay).convert('RGB')
out = '/tmp/workspace/usg-larynx/labeled_larynx_usg.jpg'
result.save(out, quality=96)
print("Saved:", out, result.size)