Added support for AiColors

This commit is contained in:
ggindinson 2023-05-26 21:43:25 +03:00
parent e35264d3db
commit 97346e6257
2 changed files with 39 additions and 0 deletions

View File

@ -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)

View File

@ -0,0 +1,9 @@
from dataclasses import dataclass
@dataclass
class AiColorsResponse:
background: str
primary: str
accent: str
text: str