From ccf60130ede5bf021adc4937036144879fb89c4d Mon Sep 17 00:00:00 2001
From: Lukas Winkler <git@lw1.at>
Date: Sun, 27 Feb 2022 14:17:24 +0100
Subject: [PATCH] also commit API

---
 fifteen_ai_api.py | 65 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 fifteen_ai_api.py

diff --git a/fifteen_ai_api.py b/fifteen_ai_api.py
new file mode 100644
index 0000000..5ca8eea
--- /dev/null
+++ b/fifteen_ai_api.py
@@ -0,0 +1,65 @@
+"""
+based on https://github.com/wafflecomposite/15.ai-Python-API
+by wafflecomposite
+MIT license
+"""
+from pathlib import Path
+from typing import TypedDict, List, Union
+
+import requests
+from requests import Session
+
+
+class Response(TypedDict):
+    batch: List[float]
+    wavNames: List[str]
+    scores: List[float]
+    torchmoji: List[Union[float, str]]
+    text_parsed: List[str]
+    tokenized: List[str]
+    dict_exists: List[List[str]]
+
+
+
+
+class FifteenAPI:
+    tts_url = "https://api.15.ai/app/getAudioFile5"
+    audio_url = "https://cdn.15.ai/audio/"
+    max_text_len = 500
+
+    def __init__(self, character: str, emotion: str = "Contextual"):
+        self.character = character
+        self.emotion = emotion
+        self.s = Session()
+
+    def set_progress(self, num: int, of: int) -> None:
+        self.s.headers.update({"User-Agent": f"VacuumVoicePack ({num}/{of})"})
+
+
+    def get_tts(self, text: str) -> Response:
+        assert len(text) <= self.max_text_len
+
+        data = {
+            "text": text,
+            "character": self.character,
+            "emotion": self.emotion
+        }
+        print(data)
+        r = self.s.post(self.tts_url, data=data)
+        r.raise_for_status()
+        return r.json()
+
+    def tts_to_wavs(self, dir: Path, text: str) -> None:
+        data = self.get_tts(text)
+        dir.mkdir(exist_ok=True)
+        print("fetching wav files")
+        for i, filename in enumerate(data["wavNames"]):
+            print(filename)
+            r = requests.get(self.audio_url + filename)
+            with open(dir / f"{i}.wav", "wb") as f:
+                f.write(r.content)
+
+
+if __name__ == '__main__':
+    api = FifteenAPI(character="GLaDOS")
+    print(api.get_tts("This is an example!"))