chatGPT_DALLE.ino

// This project works on the ESP32-S3-N16R2 (8MB of PSRAM)
// PSRAM is needed for this project.
// The round display (https://amzn.to/3L4pud6) use the GC9A01 driver from the TFT_eSPI library
// The Micro SD Card Module (optional) is this one : https://amzn.to/46tSgvn
// The Rotary Encoder (optional) is this one: https://amzn.to/3Cj7t61
//
// See the full tutorial here : https://youtu.be/zBzIBRsckTw

#include <PNGdec.h>           // Install this library with the Arduino IDE Library Manager
#include <TFT_eSPI.h>         // Install this library with the Arduino IDE Library Manager
                              // Don't forget to configure the driver for your display!
#include <AnimatedGIF.h>      // Install this library with the Arduino IDE Library Manager
#include <SimpleRotary.h>     // Install this library with the Arduino IDE Library Manager
#include <SD.h>               // Install this library with the Arduino IDE Library Manager
#include <WiFiClientSecure.h> // Install this library with the Arduino IDE Library Manager
#include "secrets.h"
#include "arduino_base64.hpp"
#include "display.h"
#include "switch.h"
#include "images\ai.h"             // AI Animated GIF
#include "images\readyPng.h"       // Ready PNG
#include "images\readyAnimation.h" // Ready Animated GIF

#define SIMULATE_CALL_DALLE // Test images will be used, uncomment this line to make the real call to the DALLE API
#define DEBUG_ON            // Comment this line if you don't want detailed messages on the serial monitor, all errors will be printed

#ifndef SIMULATE_CALL_DALLE
// The images folder and index file are created automatically, 
// just make your sure SD card is 32GB Max and formatted as FAT32.  See this video that shows how to format it: https://youtu.be/Np0cbretbns?t=110
#define USE_SD_CARD // Comment this line if you don't have an SD Card module
#endif

#ifdef USE_SD_CARD
// The rotary encoder can be used only with the SD Card Module
#define USE_ROTARY_ENCODER // Comment this line if you don't want to use the rotary encoder, it is used to navigate files on the SD card module
#endif

#ifdef DEBUG_ON
#define DEBUG_PRINT(x) Serial.print(x)
#define DEBUG_PRINTLN(x) Serial.println(x)
#define DEBUG_PRINTF(...) Serial.printf(__VA_ARGS__)
#else
#define DEBUG_PRINT(x)
#define DEBUG_PRINTLN(x)
#define DEBUG_PRINTF(...)
#endif

#ifdef SIMULATE_CALL_DALLE
#include "base64_test_images\mandalaBase64Png.h"
#include "base64_test_images\rainbowBase64Png.h"
#include "base64_test_images\resistance.h"
#include "base64_test_images\spaceship.h"
#include "base64_test_images\hal.h"

// Test PNG Images
const char *testPngImages[] = {mandalaBase64Png, resistance64Png, spaceship64Png, hal64Png};
const int testImagesCount = 4;
#endif

#ifdef USE_ROTARY_ENCODER
#define ROTARY_PIN_A 41
#define ROTARY_PIN_B 42
#define ROTARY_PUSH_BUTTON 42 // Not used
SimpleRotary rotary(ROTARY_PIN_A, ROTARY_PIN_B, ROTARY_PUSH_BUTTON);
#endif

#ifdef USE_SD_CARD
#define SD_CARD_CS_PIN 5 // Chip Select Pin for the SD Card Module
const char *ID_FILENAME = "id.txt";
const char *IMAGES_FOLDER_NAME = "/images";
int idForNewFile = 1;
unsigned int nbfOfimagesOnSdCard = -1;
#endif

#define ANIMATED_AI_IMAGE ai
AnimatedGIF gif;
TaskHandle_t taskHandlePlayGif = NULL;
void GIFDraw(GIFDRAW *pDraw);

// Switch
SwitchReader generationSwitch(1);
bool runImageGeneration = false; // Flag to indicate if generation of images is running or stopped

#define MAX_IMAGE_WIDTH 1024
#define PSRAM_BUFFER_DECODED_LENGTH 4000000L      // Length of buffer for base64 data decoding in PSRAM
#define PSRAM_BUFFER_READ_ENCODED_LENGTH 2000000L // Length of buffer for reading the base64 encoded data in PSRAM
#define BUFFER_RESPONSE_LENGTH 1024               // Length of buffer for reading api response in chunks

// Delimiters to extract base64 data in Json
const char *startToken = "\"b64_json\": \"";
const char *endToken = "\"";

// Prompts
const int promptsCount = 11;
char *prompts[promptsCount] = {"An Alien Planet with Ships Orbiting", "A Futuristic City", "A Star Wars Spaceship", "A Futurisctic HUD Screen",
                               "A Star Wars Empire Spaceship Attacking", "The Interior of a Spaceship", "Control Panels of an Alien Spaceship", "An Alien World",
                               "A Star Trek LCARS Interface", "Battle in Space between Spaceships", "A Star Wars Vessel Cockpit View in Space"};

enum imageGenerationMode
{
    MODE_SEQUENTIAL, // Images are generated in sequential order of the prompts
    MODE_RANDOM      // Images are generated by using a random prompt
};
imageGenerationMode generationMode = MODE_SEQUENTIAL; // Choose your mode here
int currentPrompt = 0;                                // Used when MODE_SEQUENTIAL is used

int16_t xpos = 0;
int16_t ypos = 0;

PNG png; // PNG decoder instance

// Display - You must disable chip select pin definitions in the user_setup.h and the driver setup (ex.: Setup46_GC9A01_ESP32.h)
TFT_eSPI tft = TFT_eSPI();  // Make sure SPI_FREQUENCY is 20000000 in your TFT_eSPI driver for your display if on a breadboard
const int NUM_DISPLAYS = 4; // Adjust this value based on the number of displays
Display display[NUM_DISPLAYS] = {
    Display(15), // Assign a chip select pin for each display
    Display(7),
    Display(6),
    Display(16)};

String base64Data;          // String buffer for base64 encoded Data
uint8_t *decodedBase64Data; // Buffer to decode base64 data

void setup()
{
    Serial.begin(115200);
    connectToWifiNetwork();
    if (!initDisplayPinsAndStorage())
    {
        Serial.println("!!! Cannot allocate enough PSRAM to store images");
        Serial.println("!!! Code Execution stopped!");
        while (true)
        {
            // Infinite loop, code execution useless without PSRAM
        }
    }
    gif.begin(BIG_ENDIAN_PIXELS);

    if (!allocatePsramMemory())
    {
        Serial.println("!!! Code Execution stopped!");
        while (true)
        {
            // Infinite loop, code execution useless without PSRAM
        }
    }

#ifdef USE_SD_CARD
    if (!initSDCard())
    {
        Serial.println("!!! Code Execution stopped!");
        while (true)
        {
            // Infinite loop, code execution useless without SD Card Module
        }
    }
    idForNewFile = readNextId(SD) + 1;
    DEBUG_PRINTF("ID for the next file is %d\n", idForNewFile);
    createDir(SD, IMAGES_FOLDER_NAME);
#endif
    createTaskCore();
    // delay(5000); // You can safely remove this delay
    playReadyOnScreens();
}

void loop()
{
    if (runImageGeneration)
    {
        generateAIImages(generationMode);
    }
#ifdef USE_ROTARY_ENCODER
    else
    {
        if (nbfOfimagesOnSdCard > NUM_DISPLAYS - 1)
        {
            readRotaryEncoder();
        }
        else
        {
            Serial.printf("Not enough images on SD Card = %u\n", nbfOfimagesOnSdCard);
        }
    }
#endif
}

#ifdef USE_ROTARY_ENCODER
void readRotaryEncoder(void)
{
    byte i;
    int fileIndex;
    i = rotary.rotate();

    if (i == 1)
    {
        DEBUG_PRINTLN("Clockwise");
        fileIndex = findPreviousFileIndexOnSDCard(display[NUM_DISPLAYS - 1].fileIndex);
        shifImagesOnDisplayRight();
        displayPngFileFromSDCard(fileIndex, NUM_DISPLAYS - 1);
    }

    if (i == 2)
    {
        DEBUG_PRINTLN("Counter Clockwise");
        fileIndex = findNextFileIndexOnSDCard(display[0].fileIndex);
        shifImagesOnDisplayLeft();
        displayPngFileFromSDCard(fileIndex, 0);
    }
}

int findPreviousFileIndexOnSDCard(int fileIndex)
{
    fileIndex -= 1;
    while (fileIndex > 0)
    {
        String filename = String(IMAGES_FOLDER_NAME) + "/" + String(fileIndex) + ".png";
        if (!SD.exists(filename))
        {
            fileIndex -= 1;
        }
        else
        {
            return fileIndex;
        }
    }

    return findPreviousFileIndexOnSDCard(idForNewFile); // Cycle back
}
int findNextFileIndexOnSDCard(int fileIndex)
{
    fileIndex += 1;
    while (fileIndex < idForNewFile)
    {
        String filename = String(IMAGES_FOLDER_NAME) + "/" + String(fileIndex) + ".png";
        if (!SD.exists(filename))
        {
            fileIndex += 1;
        }
        else
        {
            return fileIndex;
        }
    }

    return findNextFileIndexOnSDCard(0); // Cycle back
}

void displayPngFileFromSDCard(int fileIndex, int screenIndex)
{
    size_t imageSize;
    const uint8_t *image = NULL;
    if (verifyScreenIndex(screenIndex))
    {
        String filename = String(IMAGES_FOLDER_NAME) + "/" + String(fileIndex) + ".png";
        DEBUG_PRINTF("Reading %s from SD Card\n", filename.c_str());
        image = readPNGImageFromSDCard(filename.c_str(), &imageSize);
        if (image != NULL)
        {
            displayPngFromRam(image, imageSize, screenIndex);
            display[screenIndex].storeImage((uint8_t *)image, imageSize);
            display[screenIndex].fileIndex = fileIndex;
            delete image; // delete buffer for image
        }
    }
}
#endif

// Play an animated GIF asynchronously using a CPU Task
// This function is called by the task playAIGifTask
void playAIGif(void *parameter)
{
    for (;;)
    {
        gif.playFrame(true, NULL);
        vTaskDelay(5 / portTICK_PERIOD_MS);
    }
}

// Start the task to play an animated GIF asynchronously
void startPlayAIGifAsync(void)
{
    gif.open((uint8_t *)ANIMATED_AI_IMAGE, sizeof(ANIMATED_AI_IMAGE), GIFDraw);
    display[0].activate();
    tft.startWrite();
    if (taskHandlePlayGif == NULL)
    {
        taskHandlePlayGif = playAIGifTask();
    }
}

// Stop the task to play an animated GIF asynchronously
void stopPlayAIGifAsync(void)
{
    if (taskHandlePlayGif != NULL)
    {
        vTaskDelete(taskHandlePlayGif);
        taskHandlePlayGif = NULL; // Set to NULL to indicate that the task is no longer running
    }
    gif.close();
    tft.endWrite();
    display[0].deActivate();
}

// Create the task to play an animated GIF asynchronously
TaskHandle_t playAIGifTask()
{
    TaskHandle_t taskHandle = NULL;
    xTaskCreate(
        playAIGif,   // Task function
        "playAIGif", // Name of task
        2048,        // Stack size of task (adjust as needed)
        NULL,        // Parameter of the task
        1,           // Priority of the task
        &taskHandle  // Task handle
    );
    return taskHandle;
}

// Play an animated GIF synchronously
void playAnimatedGIFSync(uint8_t *image, size_t imageSize, int screenIndex)
{
    if (verifyScreenIndex(screenIndex))
    {
        gif.open(image, imageSize, GIFDraw);
        display[screenIndex].activate();
        tft.startWrite();
        while (gif.playFrame(true, NULL))
        {
            yield();
        }
        gif.close();
        tft.endWrite();
        display[screenIndex].deActivate();
    }
}

#ifdef USE_SD_CARD
// Initialize Micro SD card Module
bool initSDCard(void)
{
    pinMode(SD_CARD_CS_PIN, OUTPUT);
    digitalWrite(SD_CARD_CS_PIN, HIGH);
    if (!SD.begin(SD_CARD_CS_PIN))
    {
        // You can get this error if no Micro SD card is inserted into the module
        Serial.println("!!! SD Card initialization failed!");
        return false;
    }
    else
    {
        Serial.println("SD Card initialization success");
    }
    uint8_t cardType = SD.cardType();

    if (cardType == CARD_NONE)
    {
        Serial.println("!!! No SD card attached");
        return false;
    }

    DEBUG_PRINTLN("SD Card Type: ");
    if (cardType == CARD_MMC)
    {
        DEBUG_PRINTLN("MMC");
    }
    else if (cardType == CARD_SD)
    {
        DEBUG_PRINTLN("SDSC");
    }
    else if (cardType == CARD_SDHC)
    {
        DEBUG_PRINTLN("SDHC");
    }
    else
    {
        DEBUG_PRINTLN("UNKNOWN");
    }

    uint64_t cardSize = SD.cardSize() / (1024 * 1024);
    DEBUG_PRINTF("SD Card Size: %lluMB\n", cardSize);
    nbfOfimagesOnSdCard = countFilesInDirectory(IMAGES_FOLDER_NAME);
    return true;
}

// Read a PNG image stored on the SD card
uint8_t *readPNGImageFromSDCard(const char *filename, size_t *imageSize)
{
    DEBUG_PRINTLN("Reading PNG Image");

    File file = SD.open(filename, FILE_READ);
    if (!file)
    {
        Serial.println("!!! Failed to open PNG file for reading");
        return NULL;
    }

    // Get the size of the file
    size_t fileSize = file.size();
    if (fileSize == 0)
    {
        Serial.println("!!! PNG file is empty");
        file.close();
        return NULL;
    }
    DEBUG_PRINTF("File Size on SD Card is %u\n", fileSize);

    // Allocate memory for the file content
    uint8_t *buffer = new uint8_t[fileSize];
    if (!buffer)
    {
        Serial.println("!!! Failed to allocate memory for PNG image");
        file.close();
        return NULL;
    }

    // Read the file into the buffer
    size_t bytesRead = file.read(buffer, fileSize);
    file.close();

    if (bytesRead != fileSize)
    {
        Serial.println("!!! Failed to read the entire PNG file");
        delete[] buffer;
        return NULL;
    }

    // Set the output parameters
    *imageSize = fileSize;

    return buffer;
}

// Read the file ID to use on the SD card, for naming the next image to be stored on the SD Card
int readNextId(fs::FS &fs)
{
    DEBUG_PRINTLN("Reading next ID");
    String idFilename = ID_FILENAME;
    File file = fs.open("/" + idFilename);
    if (!file)
    {
        Serial.println("!!! Failed to open ID file for reading");
        return 0;
    }

    String fileContent = "";
    while (file.available())
    {
        fileContent += (char)file.read();
    }
    file.close();

    return fileContent.toInt(); // Convert string to integer and return
}

// Write the ID on the SD card to be used for naming the next image to be stored on the SD card
void writeNextId(fs::FS &fs, int id)
{
    DEBUG_PRINTLN("Writing next ID");
    String idFilename = ID_FILENAME;

    File file = fs.open("/" + idFilename, FILE_WRITE);
    if (!file)
    {
        Serial.println("!!! Failed to open file for writing");
        return;
    }

    // Convert the integer to a string and write it to the file
    file.print(id);

    file.close();
    DEBUG_PRINTLN("ID written successfully");
}

// Store a PNG file on the SD card
void writeImage(fs::FS &fs, const char *path, uint8_t *image, size_t length)
{
    DEBUG_PRINTF("Writing file: %s\n", path);

    File file = fs.open(path, FILE_WRITE);
    if (!file)
    {
        Serial.println("!!! Failed to open file for writing");
        return;
    }
    if (file.write(image, length))
    {
        DEBUG_PRINTLN("File written");
    }
    else
    {
        Serial.println("!!! Write failed");
    }
    file.close();
}

// Create a folder on the SD card,
void createDir(fs::FS &fs, const char *path)
{
    DEBUG_PRINTF("Creating Dir: %s\n", path);
    if (fs.mkdir(path))
    {
        DEBUG_PRINTLN("Dir created"); // if the folder already exist it reports that it was created without touching it
    }
    else
    {
        Serial.println("!!! mkdir failed");
    }
}

// Count the number of files in a folder
unsigned int countFilesInDirectory(const char *dirPath)
{
    unsigned int fileCount = 0;

    File dir = SD.open(dirPath);
    if (!dir)
    {
        Serial.println("Failed to open directory");
        return 0;
    }
    if (!dir.isDirectory())
    {
        Serial.println("Not a directory");
        return 0;
    }

    File file = dir.openNextFile();
    while (file)
    {
        if (!file.isDirectory())
        {
            fileCount++; // Count only files, not sub-directories
        }
        file.close();
        file = dir.openNextFile();
    }

    dir.close();
    DEBUG_PRINTF("Number of images on SD card=%u\n", fileCount);
    return fileCount;
}
#endif

// CPU Task to read the momentary switch asynchronously
void generationSwitchTask(void *parameter)
{
    for (;;)
    { // Task loop
        if (generationSwitch.read())
        {
            runImageGeneration = !runImageGeneration;
            if (runImageGeneration)
            {
                Serial.println("Image generation starting...");
            }
            else
            {
                Serial.println("Image generation will be stopped");
            }
        }
        vTaskDelay(50 / portTICK_PERIOD_MS); // Delay to prevent task from running too frequently
    }
}

// Generate an image on a screen
void generateAIImages(imageGenerationMode mode)
{
#ifdef SIMULATE_CALL_DALLE
    // Test images are used instead of calling the DALLE API
    display[0].activate();
    tft.fillScreen(TFT_BLACK);
    tft.setTextColor(TFT_WHITE);
    printTextWithWordWrap("Generating test image", 70, 70, 100);
    display[0].deActivate();
    startPlayAIGifAsync();
    unsigned long t = millis();
    while (millis() - t < 5000)
    {
    }
    stopPlayAIGifAsync();
    const char *image = testPngImages[myRandom(testImagesCount)];
    size_t length = displayPngImage(image, 0);
    display[0].storeImage(decodedBase64Data, length);
#else
    // Calling the DALLE API to generate images using prompts
    size_t length;
    if (mode == MODE_SEQUENTIAL)
    {
        // Sequential mode
        length = genereteDalleImage(prompts[currentPrompt]);
        currentPrompt++;
        if (currentPrompt >= promptsCount)
        {
            currentPrompt = 0;
        }
    }
    else
    {
        // Random mode
        length = generateDalleImageRandomPrompt();
    }

    display[0].storeImage(decodedBase64Data, length);
#ifdef USE_SD_CARD // IF a SD card module is used, the image is stored on the SD card
    String filename = String(IMAGES_FOLDER_NAME) + "/" + String(idForNewFile) + ".png";
    idForNewFile += 1;
    writeImage(SD, filename.c_str(), decodedBase64Data, length);
    writeNextId(SD, idForNewFile);
#endif
#endif
    delay(5000);
    if (runImageGeneration)
    {
        shifImagesOnDisplayLeft();
    }
    else
    {
        Serial.println("Image generation stopped");
    }
}

// Shift all images to the left on the screens to make place for the next image
void shifImagesOnDisplayLeft(void)
{
    for (int i = NUM_DISPLAYS - 1; i >= 0; i--)
    {
        int displaySource = i - 1;
        if (i == 0)
        {
            return;
        }
        switchImageOnDisplay(displaySource, i);
    }
}

// Shift all images to the right on the screens to make place for the next image
void shifImagesOnDisplayRight(void)
{
    for (int i = 0; i < NUM_DISPLAYS; i++)
    {
        int displaySource = i - 1;
        if (i == NUM_DISPLAYS)
        {
            return;
        }
        switchImageOnDisplay(i, displaySource);
    }
}

// Switch images between screens, the image shown on sourceDisplay is copied to destinationDisplay
void switchImageOnDisplay(int sourceDisplay, int destinationDisplay)
{
    if (sourceDisplay == destinationDisplay)
    {
        Serial.printf("!!! Source and destination display are the same (%d)\n", sourceDisplay);
        return;
    }

    if (verifyScreenIndex(sourceDisplay) && verifyScreenIndex(destinationDisplay))
    {
        if (display[sourceDisplay].imageSize() > 0)
        {
            display[destinationDisplay].fileIndex = display[sourceDisplay].fileIndex;
            display[destinationDisplay].storeImage(display[sourceDisplay].image(), display[sourceDisplay].imageSize());
            displayPngFromRam(display[destinationDisplay].image(), display[destinationDisplay].imageSize(), destinationDisplay);
            display[sourceDisplay].activate();
            tft.fillScreen(TFT_BLACK);
            display[sourceDisplay].deActivate();
        }
        else
        {
            display[destinationDisplay].activate();
            tft.fillScreen(TFT_BLACK);
            display[destinationDisplay].deActivate();
        }
        return;
    }
}

// Verify screen index, should be 0 to NUM_DISPLAYS-1
bool verifyScreenIndex(int screenIndex)
{
    if (screenIndex < 0 || screenIndex >= NUM_DISPLAYS)
    {
        Serial.printf("!!! Wrong screen index=%d, must from 0 to %d\n", screenIndex, NUM_DISPLAYS - 1);
        return false;
    }
    return true;
}

// Initialization of all the displays including buffer space to store the image currently shown
bool initDisplayPinsAndStorage(void)
{
    tft.init();
    tft.setFreeFont(&FreeSans9pt7b);

    for (int i = 0; i < NUM_DISPLAYS; i++)
    {
        pinMode(display[i].chipSelectPin(), OUTPUT);
        display[i].activate();
        tft.setRotation(2); // Adjust orientation as needed (0-3)
        tft.fillScreen(TFT_BLACK);
        display[i].deActivate();
        display[i].fileIndex = i;
    }
    for (int i = 0; i < NUM_DISPLAYS; i++)
    {
        if (!display[i].reserveMemoryForStorage())
        {
            return false;
        }
    }
    return true;
}

// Play the "Ready" animated GIF on each screen in sequence
void playReadyOnScreens(void)
{
    for (int i = 0; i < NUM_DISPLAYS; i++)
    {
        playAnimatedGIFSync((uint8_t *)readyAnimation, sizeof(readyAnimation), i);
        size_t length = base64::decodeLength(ready64Png);
        base64::decode(ready64Png, decodedBase64Data);
        display[i].storeImage(decodedBase64Data, length);
    }
}

// Return a random PNG image generated by DALLE encoded in base64
size_t generateDalleImageRandomPrompt(void)
{
    char *randomPrompt = prompts[myRandom(promptsCount)];
    return genereteDalleImage(randomPrompt);
}

// Call DALLE to generate an image using a prompt
size_t genereteDalleImage(char *prompt)
{
    display[0].activate();
    tft.fillScreen(TFT_BLACK);
    tft.setTextColor(TFT_WHITE);
    printTextWithWordWrap(prompt, 70, 70, 100);
    display[0].deActivate();
    callOpenAIAPIDalle(&base64Data, prompt);
    size_t length = displayPngImage(base64Data.c_str(), 0);

    return length;
}

// Call the DALLE API to generate an image using a prompt
// readBuffer will contain the generated image encoded in base64
void callOpenAIAPIDalle(String *readBuffer, const char *prompt)
{
    startPlayAIGifAsync();
    *readBuffer = ""; // Clear the buffer

    WiFiClientSecure client;
    client.setInsecure(); // Only for demonstration, use a proper certificate validation in production

    const char *host = "api.openai.com";
    const int httpsPort = 443;

    if (!client.connect(host, httpsPort))
    {
        // playAIGif();
        Serial.println("!!! Connection failed");
        stopPlayAIGifAsync();
        return;
    }

    String jsonPayload = "{\"model\": \"dall-e-2\", \"prompt\": \"";
    jsonPayload += prompt;
    jsonPayload += "\", \"n\": 1, \"size\": \"256x256\", \"response_format\": \"b64_json\"}";

    String request = "POST /v1/images/generations HTTP/1.1\r\n";
    request += "Host: " + String(host) + "\r\n";
    request += "Content-Type: application/json\r\n";
    request += "Authorization: Bearer " + String(chatGPT_APIKey) + "\r\n";
    request += "Content-Length: " + String(jsonPayload.length()) + "\r\n";
    request += "Connection: close\r\n\r\n";
    request += jsonPayload;

    client.print(request);

    DEBUG_PRINTF("Request sent with prompt : %s\n", prompt);

    while (client.connected())
    {
        String line = client.readStringUntil('\n');
        if (line == "\r")
        {
            DEBUG_PRINTLN("headers received");
            break;
        }
    }

    // Read the response in chunks
    int bufferLength = BUFFER_RESPONSE_LENGTH;
    char buffer[bufferLength];
    bool base64StartFound = false, base64EndFound = false;

    // Read and process the response in chunks
    while (client.available() && !base64EndFound)
    {
        // playAIGif();
        bufferLength = client.readBytes(buffer, sizeof(buffer) - 1);
        buffer[bufferLength] = '\0';
        if (!base64StartFound)
        {
            char *base64Start = strstr(buffer, startToken);
            if (base64Start)
            {
                base64StartFound = true;
                char *startOfData = base64Start + strlen(startToken); // Skip "base64,"

                // Check if the end of base64 data is in the same buffer
                char *endOfData = strstr(startOfData, endToken);
                if (endOfData)
                {
                    *endOfData = '\0'; // Replace the quote with a null terminator
                    *readBuffer += startOfData;
                    base64EndFound = true;
                }
                else
                {
                    *readBuffer += startOfData;
                }
            }
        }
        else
        {
            char *endOfData = strstr(buffer, endToken);
            if (endOfData)
            {
                *endOfData = '\0'; // Replace the quote with a null terminator
                *readBuffer += buffer;
                base64EndFound = true;
            }
            else
            {
                *readBuffer += buffer;
            }
        }
    }

    client.stop();
    if (!base64StartFound)
    {
        Serial.println("!!! No Json Base64 data in Response:");
        Serial.println(buffer);
    }
    DEBUG_PRINTLN("Request call completed");
    stopPlayAIGifAsync();
}

// Display a PNG image (decoded) on a specific screen (screenIndex)
void displayPngFromRam(const unsigned char *pngImageinC, size_t length, int screenIndex)
{
    if (verifyScreenIndex(screenIndex))
    {
        int res = png.openRAM((uint8_t *)pngImageinC, length, pngDraw);
        if (res == PNG_SUCCESS)
        {
            DEBUG_PRINTLN("Successfully opened png file");
            DEBUG_PRINTF("image specs: (%d x %d), %d bpp, pixel type: %d\n", png.getWidth(), png.getHeight(), png.getBpp(), png.getPixelType());
            DEBUG_PRINTF("Image size: %d\n", length);
            DEBUG_PRINTF("Buffer size: %d\n", png.getBufferSize());
            display[screenIndex].activate();
            tft.startWrite();
            uint32_t dt = millis();
            res = png.decode(NULL, 0);
            if (res != PNG_SUCCESS)
            {
                printPngError(png.getLastError());
            }
            DEBUG_PRINT(millis() - dt);
            DEBUG_PRINTLN("ms");

            tft.endWrite();
            display[screenIndex].deActivate();
        }
        else
        {
            printPngError(res);
        }
    }
}

// Allocate memory in PSRAM
bool allocatePsramMemory(void)
{
    if (psramFound())
    {

        DEBUG_PRINTF("PSRAM Size=%ld\n", ESP.getPsramSize());

        decodedBase64Data = (uint8_t *)ps_malloc(PSRAM_BUFFER_DECODED_LENGTH);
        if (decodedBase64Data == NULL)
        {
            Serial.println("!!! Failed to allocate memory in PSRAM for image Buffer");
            return false;
        }

        if (!base64Data.reserve(PSRAM_BUFFER_READ_ENCODED_LENGTH))
        {
            Serial.println("!!! Failed to allocate memory in PSRAM for deconding base64 Data");
            return false;
        }
    }
    else
    {
        Serial.println("!!! No PSRAM detected, needed to perform reading and decoding large base64 encoded images");
        return false;
    }
    return true;
}

// Create the CPU task to read the momentary switch asynchronously
void createTaskCore(void)
{
    /* Core where the task should run */
    xTaskCreatePinnedToCore(
        generationSwitchTask,   /* Function to implement the task */
        "generationSwitchTask", /* Name of the task */
        10000,                  /* Stack size in words */
        NULL,                   /* Task input parameter */
        1,                      /* Priority of the task */
        NULL,                   /* Task handle. */
        1);                     /* Core where the task should run */
}

// Function to print text on the screen with word wrapping
void printTextWithWordWrap(const String &text, int16_t x, int16_t y, uint16_t maxWidth)
{
    int16_t cursorX = x;
    int16_t cursorY = y;
    String currentLine = "";
    String word = "";
    char currentChar;

    tft.setTextWrap(false); // Disable automatic text wrapping

    for (int i = 0; i < text.length(); i++)
    {
        currentChar = text.charAt(i);

        // Check for space or newline
        if (currentChar == ' ' || currentChar == '\n')
        {
            if (tft.textWidth(currentLine + word) <= maxWidth)
            {
                currentLine += word + (currentChar == ' ' ? " " : "");
                word = "";
            }
            else
            {
                tft.drawString(currentLine, cursorX, cursorY);
                cursorY += tft.fontHeight();
                currentLine = word + " ";
                word = "";
            }
            if (currentChar == '\n')
            {
                tft.drawString(currentLine, cursorX, cursorY);
                cursorY += tft.fontHeight();
                currentLine = "";
            }
        }
        else
        {
            word += currentChar;
        }
    }

    if (tft.textWidth(currentLine + word) <= maxWidth)
    {
        tft.drawString(currentLine + word, cursorX, cursorY);
    }
    else
    {
        tft.drawString(currentLine, cursorX, cursorY);
        cursorY += tft.fontHeight();
        tft.drawString(word, cursorX, cursorY);
    }
}

//=========================================v==========================================
//                                      pngDraw
//====================================================================================
// This next function will be called during decoding of the png file to
// render each image line to the TFT.  If you use a different TFT library
// you will need to adapt this function to suit.
// Callback function to draw pixels to the display
void pngDraw(PNGDRAW *pDraw)
{
    uint16_t lineBuffer[MAX_IMAGE_WIDTH];
    png.getLineAsRGB565(pDraw, lineBuffer, PNG_RGB565_BIG_ENDIAN, 0xffffffff);
    // png.getLineAsRGB565(pDraw, lineBuffer, PNG_RGB565_BIG_ENDIAN, -1);
    tft.pushImage(xpos, ypos + pDraw->y, pDraw->iWidth, 1, lineBuffer);
}

void printPngError(int errorCode)
{
    switch (errorCode)
    {
    case PNG_SUCCESS:
        Serial.println("!!! PNG Success!");
        break;
    case PNG_INVALID_PARAMETER:
        Serial.println("!!! PNG Error Invalid Parameter");
        break;
    case PNG_DECODE_ERROR:
        Serial.println("!!! PNG Error Decode");
        break;
    case PNG_MEM_ERROR:
        Serial.println("!!! PNG Error Memory");
        break;
    case PNG_NO_BUFFER:
        Serial.println("!!! PNG Error No Buffer");
        break;
    case PNG_UNSUPPORTED_FEATURE:
        Serial.println("!!! PNG Error Unsupported Feature");
        break;
    case PNG_INVALID_FILE:
        Serial.println("!!! PNG Error Invalid File");
        break;
    case PNG_TOO_BIG:
        Serial.println("!!! PNG Error Too Big");
        break;
    default:
        Serial.println("!!! PNG Error Unknown");
        break;
    }
}

// Display a PNG image (encoded) on a specific screen (screenIndex)
size_t displayPngImage(const char *imageBase64Png, int displayIndex)
{
    size_t length = base64::decodeLength(imageBase64Png);
    base64::decode(imageBase64Png, decodedBase64Data);

    DEBUG_PRINTF("base64 encoded length = %ld\n", strlen(imageBase64Png));
    DEBUG_PRINTF("base64 decoded length = %ld\n", length);

    displayPngFromRam(decodedBase64Data, length, displayIndex);
    return length;
}

// Generate a random numbre using the ESP32 random number generator from 0 to howbig-1
long myRandom(long howbig)
{
    if (howbig == 0)
    {
        return 0;
    }
    return esp_random() % howbig;
}

// Generate a random numbre using the ESP32 random number generator from howsmall to howbig-1
long myRandom(long howsmall, long howbig)
{
    if (howsmall >= howbig)
    {
        return howsmall;
    }
    long diff = howbig - howsmall;
    return esp_random() % diff + howsmall;
}

// Connect to the Wifi network
void connectToWifiNetwork()
{
    WiFi.mode(WIFI_STA);
    WiFi.begin(ssid, password);
    Serial.println("");

    while (WiFi.status() != WL_CONNECTED)
    {
        delay(100);
        Serial.print(".");
    }

    Serial.println("");
    Serial.print("Connected to ");
    Serial.println(ssid);
    Serial.print("IP address: ");
    Serial.println(WiFi.localIP());
}