' Set up the TFT display
TFT.INIT

' Main program loop
LOOP:
  TFT.FILL 0  ' Clear the screen with the default background color
  DrawMandelbrot -2, -1.5, 3, 3, 100  ' Draw Mandelbrot set
  WAIT 5000    ' Wait for 5 seconds before redrawing (optional)
GOTO LOOP

' Subroutine to draw the Mandelbrot set
SUB DrawMandelbrot(xMin, yMin, xMax, yMax, maxIter)
  LOCAL x, y, x0, y0, xtemp, iteration, color

  FOR px = 0 TO 319  ' Width of the display
    FOR py = 0 TO 239  ' Height of the display
      x0 = xMin + (xMax - xMin) * px / 319
      y0 = yMin + (yMax - yMin) * py / 239
      x = 0
      y = 0
      iteration = 0

      WHILE (x * x + y * y <= 4) AND (iteration < maxIter)
        xtemp = x * x - y * y + x0
        y = 2 * x * y + y0
        x = xtemp
        iteration = iteration + 1
      WEND

      ' Color based on the number of iterations
      IF iteration < maxIter THEN
        color = TFT.COLOR(iteration * 255 / maxIter, 0, iteration * 255 / maxIter) ' Gradient color based on iterations
        TFT.PIXEL px, py, color
      ELSE
        TFT.PIXEL px, py, TFT.COLOR(0) ' Black for points in the set
      ENDIF
    NEXT py
  NEXT px
RETURN