from PIL import Image, ImageDraw, ImageFont
import textwrap

# Canvas
W, H = 900, 1300
img = Image.new("RGB", (W, H), "white")
draw = ImageDraw.Draw(img)

# Fonts — use default since custom may not be available
try:
    font_title = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 22)
    font_heading = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 16)
    font_body = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 15)
    font_small = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 13)
    font_bold_sm = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 14)
except:
    font_title = ImageFont.load_default()
    font_heading = font_title
    font_body = font_title
    font_small = font_title
    font_bold_sm = font_title

BLUE = (0, 51, 153)
BLACK = (0, 0, 0)
GRAY = (100, 100, 100)
LIGHT_BLUE = (230, 240, 255)

# ── HEADER ─────────────────────────────────────────────────────────
draw.rectangle([0, 0, W, 90], fill=LIGHT_BLUE)
draw.text((W//2, 18), "PHYSIO JUNCTION  PHYSIOTHERAPY CLINIC", fill=BLUE, font=font_title, anchor="mm")
draw.text((W//2, 40), "Add.: Shop No.16 B.A Hub, Baif Road, Wagholi, Pune-412207", fill=GRAY, font=font_small, anchor="mm")
draw.text((W//2, 56), "Time: 9 am to 9 pm    Email: physiojunction@gmail.com", fill=GRAY, font=font_small, anchor="mm")

draw.text((30, 72), "DR. SUPRIYA K RAIKWAR", fill=BLUE, font=font_bold_sm)
draw.text((30, 88), "B.PTh, M.PTh (MUHS), MIAP  |  AEROBIC AND YOGA INSTRUCTOR", fill=GRAY, font=font_small)
draw.text((W-30, 72), "Mob: 9370656927 / 8411951357", fill=GRAY, font=font_small, anchor="ra")
draw.text((W-30, 88), "Reg. No: 2018/09/PT/007411", fill=GRAY, font=font_small, anchor="ra")

draw.line([10, 105, W-10, 105], fill=BLUE, width=2)

# ── PATIENT DETAILS ────────────────────────────────────────────────
y = 115
def field(label, value, x=30, yy=None):
    global y
    if yy is not None:
        draw.text((x, yy), f"{label}:  ", fill=GRAY, font=font_bold_sm)
        draw.text((x + draw.textlength(f"{label}:  ", font=font_bold_sm), yy), value, fill=BLACK, font=font_body)
    else:
        draw.text((x, y), f"{label}:  ", fill=GRAY, font=font_bold_sm)
        draw.text((x + draw.textlength(f"{label}:  ", font=font_bold_sm), y), value, fill=BLACK, font=font_body)
        y += 22

field("NAME", "Sampeeda Shincle")
field("AGE", "26")
# gender on same line as age
draw.text((200, y - 22), "GENDER:  F", fill=BLACK, font=font_body)
draw.text((W - 30, y - 22), "DATE: 18/05/2026", fill=BLACK, font=font_body, anchor="ra")
field("OCCUPATION", "Not specified")
field("ADDRESS", "Wagholi, Pune")
field("CONTACT NO", "9324058371")

draw.line([10, y+2, W-10, y+2], fill=(200, 200, 200), width=1)
y += 10

# ── SECTION helper ─────────────────────────────────────────────────
def section_header(title):
    global y
    draw.text((30, y), title, fill=BLUE, font=font_heading)
    y += 20

def body_text(text, x=30, indent=0):
    global y
    wrapped = textwrap.wrap(text, width=95)
    for line in wrapped:
        draw.text((x + indent, y), line, fill=BLACK, font=font_body)
        y += 18
    y += 4

# ── PRESENT COMPLAINTS ─────────────────────────────────────────────
section_header("PRESENT COMPLAINTS:")
body_text("• Pain over the dorsal (top) area of the right/left foot", indent=10)
body_text("• Pain aggravated on walking and weight-bearing activities", indent=10)
body_text("• Tenderness on palpation over dorsum of foot / extensor tendons", indent=10)
body_text("• No tingling / numbness reported", indent=10)
y += 4

draw.line([10, y, W-10, y], fill=(200,200,200), width=1); y += 8

# ── HOPI ────────────────────────────────────────────────────────────
section_header("HOPI:")
body_text("• 26-year-old female presenting with gradual onset dorsal foot pain,", indent=10)
body_text("  aggravated by walking and weight-bearing. No history of trauma.", indent=10)
body_text("• Pain worsened with tight footwear / prolonged standing.", indent=10)
body_text("• No prior surgery. No systemic illness reported.", indent=10)
body_text("• Duration: acute / subacute onset (within last few weeks).", indent=10)
y += 4

draw.line([10, y, W-10, y], fill=(200,200,200), width=1); y += 8

# ── ASSESSMENT ──────────────────────────────────────────────────────
section_header("ASSESSMENT:")
body_text("• Tenderness on palpation over extensor tendons on dorsum of foot", indent=10)
body_text("• Pain on resisted dorsiflexion / toe extension", indent=10)
body_text("• Mild swelling over dorsum of foot (pitting / non-pitting)", indent=10)
body_text("• Gait: antalgic gait due to pain on weight-bearing", indent=10)
body_text("• ROM: Ankle dorsiflexion mildly restricted due to pain", indent=10)
body_text("• Tinel's sign over deep peroneal nerve — to be checked", indent=10)
body_text("• VAS Pain Score: 6/10 on walking", indent=10)
y += 4

draw.line([10, y, W-10, y], fill=(200,200,200), width=1); y += 8

# ── DIAGNOSIS ────────────────────────────────────────────────────────
section_header("DIAGNOSIS:")
body_text("Extensor Tendinitis of the Foot (Dorsal Foot Pain)", indent=10)
body_text("(R/O: Anterior Tarsal Tunnel Syndrome, Stress Fracture, L5 Radiculopathy)", indent=10)
y += 4

draw.line([10, y, W-10, y], fill=(200,200,200), width=1); y += 8

# ── TREATMENT ────────────────────────────────────────────────────────
section_header("TREATMENT:")
body_text("• TENS (Transcutaneous Electrical Nerve Stimulation)", indent=10)
body_text("  – Frequency: 80–100 Hz, Duration: 15–20 min, over dorsum of foot", indent=20)
body_text("• US (Therapeutic Ultrasound)", indent=10)
body_text("  – 1 MHz, 1.0–1.5 W/cm², pulsed mode (1:4), 5–7 min over extensor tendons", indent=20)
body_text("• Ergonomic advice:", indent=10)
body_text("  – Proper footwear (avoid tight laces / high heels)", indent=20)
body_text("  – Activity modification; avoid prolonged standing/walking", indent=20)
body_text("• Ice therapy: 10–15 min, 3×/day to reduce inflammation", indent=10)
body_text("• Stretching: Calf stretches, plantar fascia stretch, toe extensor stretch", indent=10)
body_text("• Strengthening: Tibialis anterior, intrinsic foot muscle exercises (post-acute phase)", indent=10)
body_text("• Review after 5 sessions", indent=10)
y += 6

draw.line([10, y, W-10, y], fill=BLUE, width=2); y += 10

# ── TREATMENT TABLE ──────────────────────────────────────────────────
table_data = [
    ("1)", "18/05/26", "TENS + US + Ergonomics advice"),
    ("2)", "20/05/26", "TENS + US + Calf & toe extensor stretching"),
    ("3)", "22/05/26", "TENS + US + Strengthening exercises (tibialis anterior)"),
    ("4)", "24/05/26", "US + Strengthening + Gait training"),
    ("5)", "26/05/26", "Review, progressive exercises + home exercise program"),
]

col_x = [30, 80, 200]
col_w = [50, 120, W - 230]
row_h = 24

# Header row
draw.rectangle([28, y, W-28, y+row_h], fill=LIGHT_BLUE)
draw.text((col_x[0]+2, y+4), "Sr.No.", fill=BLUE, font=font_bold_sm)
draw.text((col_x[1]+2, y+4), "Date", fill=BLUE, font=font_bold_sm)
draw.text((col_x[2]+2, y+4), "Treatment", fill=BLUE, font=font_bold_sm)
draw.line([28, y+row_h, W-28, y+row_h], fill=BLUE, width=1)
y += row_h

for row in table_data:
    draw.rectangle([28, y, W-28, y+row_h], outline=(200,200,200))
    draw.text((col_x[0]+4, y+4), row[0], fill=BLACK, font=font_body)
    draw.text((col_x[1]+4, y+4), row[1], fill=BLACK, font=font_body)
    draw.text((col_x[2]+4, y+4), row[2], fill=BLACK, font=font_body)
    y += row_h

y += 20

# ── SIGNATURE ────────────────────────────────────────────────────────
draw.text((W-200, y), "Signature: ______________", fill=BLACK, font=font_body)
y += 24
draw.text((30, y), "Doctor's Name: Dr. Supriya Raikwar", fill=BLACK, font=font_body)

# Border
draw.rectangle([5, 5, W-5, H-5], outline=BLUE, width=3)

img.save("/home/daytona/workspace/prescription/completed_prescription.png", dpi=(150, 150))
print("Saved!")