From 6e73a52eb7cf182bc9801bce1d9ede919c54483e Mon Sep 17 00:00:00 2001 From: Benjamyn Love Date: Mon, 21 Aug 2023 11:15:50 +1000 Subject: [PATCH] removed import requests.py --- modules/import requests.py | 86 -------------------------------------- 1 file changed, 86 deletions(-) delete mode 100644 modules/import requests.py diff --git a/modules/import requests.py b/modules/import requests.py deleted file mode 100644 index e18cde9..0000000 --- a/modules/import requests.py +++ /dev/null @@ -1,86 +0,0 @@ -import requests -import io -import os.path -from PIL import Image -from config import Config -import uuid -import openai -from base64 import b64decode - -cfg = Config() - -working_directory = "auto_gpt_workspace" - - -def generate_image(prompt): - - filename = str(uuid.uuid4()) + ".jpg" - - # DALL-E - if cfg.image_provider == 'dalle': - - openai.api_key = cfg.openai_api_key - - response = openai.Image.create( - prompt=prompt, - n=1, - size="256x256", - response_format="b64_json", - ) - - print("Image Generated for prompt:" + prompt) - - image_data = b64decode(response["data"][0]["b64_json"]) - - with open(working_directory + "/" + filename, mode="wb") as png: - png.write(image_data) - - return "Saved to disk:" + filename - - # STABLE DIFFUSION - elif cfg.image_provider == 'sd': - - API_URL = "https://api-inference.huggingface.co/models/CompVis/stable-diffusion-v1-4" - headers = {"Authorization": "Bearer " + cfg.huggingface_api_token} - - response = requests.post(API_URL, headers=headers, json={ - "inputs": prompt, - }) - - image = Image.open(io.BytesIO(response.content)) - print("Image Generated for prompt:" + prompt) - - image.save(os.path.join(working_directory, filename)) - - return "Saved to disk:" + filename - - elif cfg.image_provider == 'sdwu': - API_URL = "http://10.6.9.69:7860" - if payload is None: - payload = { - "prompt": prompt, - } - - payload['prompt'] = prompt - - settings = { - "enable_pnginfo": False, - } - - override_payload = { - "override_settings": settings - } - - payload.update(override_payload) - - request = requests.post("/sdapi/v1/txt2img", json=payload) - - image_data = request.json() - image = Image.open(io.BytesIO(image_data["images"][0])) - print("Image Generated for prompt:" + prompt) - - image.save(os.path.join(working_directory, filename)) - - return "Saved to disk:" + filename - else: - return "No Image Provider Set" \ No newline at end of file