Update image saving with timestamps, fix upscaler steps placement, rename vectorize output to Image

This commit is contained in:
Sirshendu Ganguly
2026-01-23 23:21:13 -05:00
parent 91646ada84
commit 7b570f7b4b
47 changed files with 193 additions and 366 deletions
+11 -2
View File
@@ -552,6 +552,15 @@ def convertIMG2Tensor(b64img):
tensorImage = torch.from_numpy(imageNP).squeeze()
return tensorImage
def extractImageURLs(imageDataObject):
"""Extract image URLs from API response and return as comma-separated string"""
image_urls = []
for result in imageDataObject.get("data", []):
imageURL = result.get("imageURL")
if imageURL:
image_urls.append(imageURL)
return ",".join(image_urls) if image_urls else ""
def convertImageB64List(imageDataObject):
images = ()
for result in imageDataObject["data"]:
@@ -566,8 +575,8 @@ def convertImageB64List(imageDataObject):
generatedImage = convertIMG2Tensor(generatedImage)
images += (generatedImage,)
else:
# If no base64 data, try to get image URL (check both imageURL and mediaURL)
imageURL = result.get("imageURL") or result.get("mediaURL")
# If no base64 data, try to get image URL
imageURL = result.get("imageURL")
if imageURL:
# Download image from URL and convert to tensor
try: