| import json | |
| import os | |
| import yaml | |
| import requests | |
| import pathlib | |
| from aiohttp import web | |
| root_path = pathlib.Path(__file__).parent.parent.parent | |
| config_path = os.path.join(root_path,'config.yaml') | |
| class FluxAIAPI: | |
| def __init__(self): | |
| self.api_url = "https://fluxaiimagegenerator.com/api" | |
| self.origin = "https://fluxaiimagegenerator.com" | |
| self.user_agent = None | |
| self.cookie = None | |
| def promptGenerate(self, text, cookies=None): | |
| cookie = self.cookie if cookies is None else cookies | |
| if cookie is None: | |
| if os.path.isfile(config_path): | |
| with open(config_path, 'r') as f: | |
| data = yaml.load(f, Loader=yaml.FullLoader) | |
| if 'FLUXAI_COOKIE' not in data: | |
| raise Exception("Please add FLUXAI_COOKIE to config.yaml") | |
| if "FLUXAI_USER_AGENT" in data: | |
| self.user_agent = data["FLUXAI_USER_AGENT"] | |
| self.cookie = cookie = data['FLUXAI_COOKIE'] | |
| headers = { | |
| "Cookie": cookie, | |
| "Referer": "https://fluxaiimagegenerator.com/flux-prompt-generator", | |
| "Origin": self.origin, | |
| "Content-Type": "application/json", | |
| } | |
| if self.user_agent is not None: | |
| headers['User-Agent'] = self.user_agent | |
| url = self.api_url + '/prompt' | |
| json = { | |
| "prompt": text | |
| } | |
| response = requests.post(url, json=json, headers=headers) | |
| res = response.json() | |
| data = res['data'] | |
| if "error" in data: | |
| return data['error'] | |
| elif "prompt" in data: | |
| return data['prompt'] | |
| fluxaiAPI = FluxAIAPI() | |