removed import requests.py

This commit is contained in:
Benjamyn Love 2023-08-21 11:15:50 +10:00
parent f4718c83db
commit 6e73a52eb7

View File

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