From 97346e625709ff147a2c5a3884b0457aecd4a2e3 Mon Sep 17 00:00:00 2001
From: ggindinson <grisha.gindinson@gmail.com>
Date: Fri, 26 May 2023 21:43:25 +0300
Subject: [PATCH] Added support for AiColors

---
 gpt4free/aicolors/__init__.py         | 30 +++++++++++++++++++++++++++
 gpt4free/aicolors/typings/__init__.py |  9 ++++++++
 2 files changed, 39 insertions(+)
 create mode 100644 gpt4free/aicolors/__init__.py
 create mode 100644 gpt4free/aicolors/typings/__init__.py

diff --git a/gpt4free/aicolors/__init__.py b/gpt4free/aicolors/__init__.py
new file mode 100644
index 00000000..f3afa265
--- /dev/null
+++ b/gpt4free/aicolors/__init__.py
@@ -0,0 +1,30 @@
+import fake_useragent
+import requests
+import json
+from typings import AiColorsResponse
+
+
+class Completion:
+    @staticmethod
+    def create(
+        query: str = "",
+    ) -> AiColorsResponse:
+        headers = {
+            "authority": "jsuifmbqefnxytqwmaoy.functions.supabase.co",
+            "accept": "*/*",
+            "accept-language": "en-US,en;q=0.5",
+            "cache-control": "no-cache",
+            "sec-fetch-dest": "empty",
+            "sec-fetch-mode": "cors",
+            "sec-fetch-site": "same-origin",
+            "user-agent": fake_useragent.UserAgent().random,
+        }
+
+        json_data = {"query": query}
+
+        url = "https://jsuifmbqefnxytqwmaoy.functions.supabase.co/chatgpt"
+        request = requests.post(url, headers=headers, json=json_data, timeout=30)
+        data = request.json().get("text").get("content")
+        json_data = json.loads(data.replace("\n  ", ""))
+
+        return AiColorsResponse(**json_data)
diff --git a/gpt4free/aicolors/typings/__init__.py b/gpt4free/aicolors/typings/__init__.py
new file mode 100644
index 00000000..8c4f29d1
--- /dev/null
+++ b/gpt4free/aicolors/typings/__init__.py
@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class AiColorsResponse:
+    background: str
+    primary: str
+    accent: str
+    text: str