| |
|
|
| 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) |
|
|
