//***********************************************************************************************
//*
//* openAI - blocking function to fetch openAI TTS response and play the audio
//* by SaKiE 2024 - Doha Qatar
//*
//***********************************************************************************************
#pragma once 
#include <HTTPClient.h>
#include <ArduinoJson.h>
#include "AudioTools.h" // Main AudioTools library
#include "AudioTools/AudioCodecs/CodecMP3Helix.h"
#include "AudioTools/AudioLibs/AudioBoardStream.h"
#define I2S_BCK 26
#define I2S_WS 25
#define I2S_DAT 22
//***************************************************************
//** Defaults openAI
//***************************************************************
const char* model_OpenAI = "tts-1";
const char* voice_OpenAI = "alloy";
const char* apiKey = "*************************";
const char* apiUrl = "https://api.openai.com/v1/audio/speech";
//***************************************************************
//** Defaults wifi
//***************************************************************
// WiFi credentials
const char* ssid = "****************";
const char* password = "****************";
HTTPClient http;
//***************************************************************
//** Objects i2s Pipeline
//***************************************************************
AudioInfo info(44100, 2, 16);
AudioBoardStream i2s(NoBoard);
EncodedAudioStream decoderTTS(&i2s, new MP3DecoderHelix()); // Decoding stream
void setup() {
 Serial.begin(115200);
 Serial.flush();
 delay(2000);
 //clear the terminal
 for (int i = 0; i < 10; i++) {
 Serial.println();
 }
 Serial.println("\t\tWelcome !");
 //*************************************************************
 //****** Comms
 //*************************************************************
 // Connect to WiFi
 WiFi.begin(ssid, password);
 Serial.print("Connecting to WiFi...");
 while (WiFi.status() != WL_CONNECTED) {
 delay(500);
 Serial.print(".");
 }
 Serial.println("\nWiFi connected.");
 //*************************************************************
 //****** i2s
 //*************************************************************
 //I2S Output Intercafe Config
 Serial.println("\t\tSetting up I2S Speaker Environment");
 auto cfg = i2s.defaultConfig(TX_MODE);
 cfg.sample_rate = 44100;
 cfg.bits_per_sample = 16;
 cfg.channels = 2;
 cfg.buffer_count = 20;
 cfg.buffer_size = 512;
 // Custom I2S output pins
 cfg.pin_bck = I2S_BCK;
 cfg.pin_ws = I2S_WS;
 cfg.pin_data = I2S_DAT;
 i2s.begin(cfg);
 auto config = decoderTTS.defaultConfig();
 //Requires adjustment depending on selected model
 config.sample_rate = 24000; 
 config.bits_per_sample = 16;
 config.channels = 1;
 if (!decoderTTS.begin(config)) {
 Serial.println("Failed to initialize decoder.");
 } else {
 if (fetchOpenAI("I am a wonderful openAI Text To Speech voice !")) {
 Serial.println("Successful openAI TTS-1 retrieval.");
 } else {
 Serial.println("Failed to process openAI TTS requst !");
 }
 }
}
bool fetchOpenAI(const char* requestStr) {
 
 HTTPClient http;
 if (WiFi.status() == WL_CONNECTED) {
 // Prepare the JSON payload
 DynamicJsonDocument jsonDoc(512);
 jsonDoc["model"] = model_OpenAI;
 jsonDoc["input"] = requestStr;
 jsonDoc["voice"] = voice_OpenAI;
 String requestBody;
 serializeJson(jsonDoc, requestBody);
 // Set up the HTTPS request
 http.begin(apiUrl);
 http.addHeader("Authorization", String("Bearer ") + apiKey);
 http.addHeader("Content-Type", "application/json");
 Serial.println("Sending POST request...");
 int httpResponseCode = http.POST(requestBody);
 if (httpResponseCode) {// == 200
 Serial.println("Request successful. Streaming audio...");
 // Use the HTTP response stream directly
 WiFiClient* audioStream = http.getStreamPtr();
 // Stream and play audio
 StreamCopy copier(decoderTTS, *audioStream);
 copier.begin();
 unsigned long lastCopyTime = millis(); // Track the time of last copy to detect stalls
 unsigned long maxIdleTime = 5000; // Maximum time without data copy before timing out
 Serial.println("Start playing TTS audio...");
 do {
 if (copier.copy()) {
 // Reset the last copy time whenever data is successfully copied
 lastCopyTime = millis();
 }
 
 // Check if the stream has been idle too long (without data copied)
 if (millis() - lastCopyTime > maxIdleTime) {
 Serial.println("Timeout while copying audio due to inactivity.");
 break; // Exit the loop if we timeout
 }
 } while (copier.available()); // Continue while data is available to copy
 Serial.println("End playing TTS audio.");
 Serial.println("Audio playback complete.");
 } else {
 Serial.printf("HTTP Request failed: %d\n", httpResponseCode);
 return false;
 }
 http.end(); // Close connection
 return true;
 } else {
 Serial.println("WiFi not connected.");
 return false;
 }
}
void loop() {
 yield();
}