highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-11-21 12:26:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

rehost dalle results

Browse Source
This commit is contained in:
Believethehype
2023-11-30 10:57:29 +01:00
parent fccaffa9fc
commit 023ba6d55f
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tasks/imagegeneration_openai_dalle.py
View File

@@ -1,10 +1,15 @@
import json import json
from io import BytesIO
import requests
from PIL import Image
from interfaces.dvmtaskinterface import DVMTaskInterface from interfaces.dvmtaskinterface import DVMTaskInterface
from utils.admin_utils import AdminConfig from utils.admin_utils import AdminConfig
from utils.definitions import EventDefinitions from utils.definitions import EventDefinitions
from utils.dvmconfig import DVMConfig from utils.dvmconfig import DVMConfig
from utils.nip89_utils import NIP89Config from utils.nip89_utils import NIP89Config
from utils.output_utils import upload_media_to_hoster
""" """
This File contains a Module to transform Text input on NOVA-Server and receive results back. This File contains a Module to transform Text input on NOVA-Server and receive results back.
@@ -101,7 +106,12 @@ class ImageGenerationDALLE(DVMTaskInterface):
) )
image_url = response.data[0].url image_url = response.data[0].url
return image_url # rehost the result instead of relying on the openai link
response = requests.get(image_url)
image = Image.open(BytesIO(response.content)).convert("RGB")
image.save("./outputs/image.jpg")
result = upload_media_to_hoster("./outputs/image.jpg")
return result
except Exception as e: except Exception as e:
print("Error in Module") print("Error in Module")