fifteen_api.py

import re
import time
import json
import logging

import requests
from requests.exceptions import ConnectionError


class FifteenAPI:

    logger = logging.getLogger('15API')
    logger.addHandler(logging.StreamHandler())

    max_text_len = 500

    tts_headers = {
        "accept": "application/json, text/plain, */*",
        "accept-encoding": "gzip, deflate, br",
        "accept-language": "en-US,en;q=0.9",
        "access-control-allow-origin": "*",
        "content-type": "application/json;charset=UTF-8",
        "origin": "https://fifteen.ai",
        "referer": "https://fifteen.ai/app",
        "sec-fetch-dest": "empty",
        "sec-fetch-mode": "cors",
        "sec-fetch-site": "same-site",
        "user-agent": "python-requests 15.ai-Python-API(https://github.com/wafflecomposite/15.ai-Python-API)"
    }

    tts_url = "https://api.15.ai/app/getAudioFile5"


    def __init__(self, show_debug = False):
        if show_debug:
            self.logger.setLevel(logging.DEBUG)
        else:
            self.logger.setLevel(logging.WARNING)
        self.logger.info("FifteenAPI initialization")


    def get_tts_raw(self, character, emotion, text):

        resp = {"status": "NOT SET", "data": None}


        text_len = len(text)
        if text_len > self.max_text_len:
            self.logger.warning(f'Text too long ({text_len} > {self.max_text_len}), trimming to {self.max_text_len} symbols')
            text = text[:self.max_text_len - 1]

        if not text.endswith(".") and not text.endswith("!") and not text.endswith("?"):
            if len(text) < 140:
                text += '.'
            else:
                text = text[:-1] + '.'

        self.logger.info(f'Target text: [{text}]')
        self.logger.info(f'Character: [{character}]')
        self.logger.info(f'Emotion: [{emotion}]')

        data = json.dumps({"text": text, "character": character, "emotion": emotion})

        self.logger.info('Waiting for 15.ai response...')

        try:
            response = requests.post(self.tts_url, data=data, headers=self.tts_headers)
        except requests.exceptions.ConnectionError as e:
            resp["status"] = f"ConnectionError ({e})"
            self.logger.error(f"ConnectionError ({e})")
            return resp

        if response.status_code == 200:
            resp["status"] = "OK"
            resp["data"] = response.content
            self.logger.info(f"15.ai API response success")
            return resp
        else:
            self.logger.error(f'15.ai API request error, Status code: {response.status_code}')
            resp["status"] = f'15.ai API request error, Status code: {response.status_code}'
        return resp
        
    def save_to_file(self, character, emotion, text, filename=None):
        tts = self.get_tts_raw(character, emotion, text)
        if tts["status"] == "OK" and tts["data"] is not None:
            if filename is None:
                char_filename_part = "".join(x for x in character[:10] if x.isalnum())
                text_filename_part = "".join(x for x in text[:16] if x.isalnum())
                filename = f"15ai-{char_filename_part}-{text_filename_part}-{round(time.time())}.wav"
            if not filename.endswith(".wav"):
                filename += ".wav"
            f = open(filename, 'wb')
            f.write(tts["data"])
            f.close()
            self.logger.info(f"File saved: {filename}")
            return {"status": tts["status"], "filename": filename}
        else:
            return {"status": tts["status"], "filename": None}
        


if __name__ == "__main__":
    fifteen = FifteenAPI(show_debug = True)

    print("Visit https://fifteen.ai/app to find available characters and their emotions.")

    input_str = None
    while input_str != "quit":
        print("Input character (Case sensitive!):")
        character = input()
        print(f"Input emotion (Case sensitive!):")
        emotion = input()
        print("Input text:")
        text = input()
        print("Processing...")
        fifteen.save_to_file(character, emotion, text)