package utils
import (
	"image"
	"image/color"
	"strings"
	"testing"
)
func TestWriteKittyImage_SmallImage(t *testing.T) {
	// 2x2 image produces small base64 payload — single chunk, no m= key
	img := image.NewGray(image.Rect(0, 0, 2, 2))
	img.SetGray(0, 0, color.Gray{Y: 128})
	var buf strings.Builder
	if err := WriteKittyImage(img, &buf); err != nil {
		t.Fatalf("WriteKittyImage: %v", err)
	}
	out := buf.String()
	if !strings.HasPrefix(out, "\x1b_Gf=100,a=T;") {
		t.Error("expected single-chunk header with f=100,a=T")
	}
	if strings.Contains(out, "m=") {
		t.Error("small image should not use chunked m= key")
	}
	if !strings.HasSuffix(out, "\x1b\\") {
		t.Error("expected escape sequence terminator")
	}
}
func TestWriteKittyImage_LargeImage_Chunked(t *testing.T) {
	// 448x448 grayscale image produces >4096 bytes of base64
	img := image.NewGray(image.Rect(0, 0, 448, 448))
	for y := range 448 {
		for x := range 448 {
			img.SetGray(x, y, color.Gray{Y: uint8((x + y) % 256)})
		}
	}
	var buf strings.Builder
	if err := WriteKittyImage(img, &buf); err != nil {
		t.Fatalf("WriteKittyImage: %v", err)
	}
	out := buf.String()
	// Should have multiple escape sequences
	chunks := strings.Split(out, "\x1b\\")
	// Last element is empty after final terminator
	chunks = chunks[:len(chunks)-1]
	if len(chunks) < 2 {
		t.Fatalf("expected multiple chunks, got %d", len(chunks))
	}
	// First chunk should have f=100,a=T,m=1
	if !strings.Contains(chunks[0], "f=100,a=T,m=1") {
		t.Errorf("first chunk missing f=100,a=T,m=1: %s", chunks[0][:min(80, len(chunks[0]))])
	}
	// Last chunk should have m=0
	last := chunks[len(chunks)-1]
	if !strings.Contains(last, "\x1b_Gm=0;") {
		t.Errorf("last chunk missing m=0: %s", last[:min(80, len(last))])
	}
	// Middle chunks should have m=1
	for i := 1; i < len(chunks)-1; i++ {
		if !strings.Contains(chunks[i], "\x1b_Gm=1;") {
			t.Errorf("middle chunk %d missing m=1", i)
		}
	}
}
func TestClearKittyImages(t *testing.T) {
	var buf strings.Builder
	ClearKittyImages(&buf)
	expected := "\x1b_Ga=d\x1b\\"
	if buf.String() != expected {
		t.Errorf("got %q, want %q", buf.String(), expected)
	}
}

// The image is encoded as PNG, base64'd, and sent via Kitty escape sequences.

// The image is encoded as PNG, base64'd, and sent via chunked Kitty escape sequences.
// Chunks are 4096 bytes per the Kitty protocol recommendation.

	// Write Kitty protocol: ESC _ G f=100,a=T;{base64_data} ESC \
	// f=100 = PNG format
	// a=T = transmit and display
	fmt.Fprintf(w, "\x1b_Gf=100,a=T;%s\x1b\\", base64Data)

	// Small payload: send in a single escape sequence (no m= key needed)
	const chunkSize = 4096
	if len(base64Data) <= chunkSize {
		fmt.Fprintf(w, "\x1b_Gf=100,a=T;%s\x1b\\", base64Data)
		return nil
	}
	// Chunked transmission for large payloads
	for offset := 0; offset < len(base64Data); offset += chunkSize {
		end := offset + chunkSize
		if end > len(base64Data) {
			end = len(base64Data)
		}
		chunk := base64Data[offset:end]
		if offset == 0 {
			// First chunk: include format and action, m=1 means more chunks follow
			fmt.Fprintf(w, "\x1b_Gf=100,a=T,m=1;%s\x1b\\", chunk)
		} else if end < len(base64Data) {
			// Middle chunk: m=1 means more chunks follow
			fmt.Fprintf(w, "\x1b_Gm=1;%s\x1b\\", chunk)
		} else {
			// Last chunk: m=0 means no more chunks
			fmt.Fprintf(w, "\x1b_Gm=0;%s\x1b\\", chunk)
		}
	}

		return m, nil

		return m, tea.ClearScreen

		return m, nil

		return m, tea.ClearScreen

				return m, tea.ClearScreen

	// Resize to 224x224
	resized := utils.ResizeImage(img, 224, 224)

	// Resize to 448x448
	resized := utils.ResizeImage(img, 448, 448)

		resized := utils.ResizeImage(img, 224, 224)

		resized := utils.ResizeImage(img, 448, 448)

## [2026-02-28] Fix Kitty Image Rendering at 448px in Classify TUI

**Bug fix:** Spectrogram display upgraded from 224x224 to 448x448 pixels. Old image artifacts persisted between segment navigations at the larger size.
- `utils/kitty_image.go` — Chunked Kitty protocol transmission (4096-byte chunks) per spec; small images still sent as single payload
- `tui/classify.go` — Return `tea.ClearScreen` on navigation keys (`,`, `.`, bindings) to force full redraw and reliable image clearing
- `tui/classify.go` — `ResizeImage` call updated from 224x224 to 448x448
- `utils/kitty_image_test.go` — Tests for single-chunk, multi-chunk, and clear behavior