|
|
|
|
|
import base64 |
|
from io import BytesIO |
|
|
|
import requests |
|
from PIL import Image |
|
|
|
|
|
class ServiceError(Exception): |
|
def __init__(self, status_code): |
|
self.status_code = status_code |
|
|
|
|
|
def get_images_from_backend(prompt, backend_url): |
|
r = requests.post(backend_url, json={"prompt": prompt}) |
|
if r.status_code == 200: |
|
json = r.json() |
|
images = json["images"] |
|
images = [Image.open(BytesIO(base64.b64decode(img))) for img in images] |
|
version = json.get("version", "unknown") |
|
return {"images": images, "version": version} |
|
else: |
|
raise ServiceError(r.status_code) |
|
|
|
|
|
def get_model_version(url): |
|
r = requests.get(url) |
|
if r.status_code == 200: |
|
version = r.json()["version"] |
|
return version |
|
else: |
|
raise ServiceError(r.status_code) |
|
|
