virtual-characters / src /image_engine.py
ShadowInk's picture
Upload complete Space runtime files
6bcddd0 verified
Raw
History Blame Contribute Delete
470 Bytes
import os
from pathlib import Path
def generate_character_image(prompt: str, output_path: str | Path) -> Path | None:
url = os.environ.get("VC_MODAL_IMAGE_URL")
if not url:
return None
import httpx
response = httpx.post(url, json={"prompt": prompt, "steps": 4}, timeout=240)
response.raise_for_status()
path = Path(output_path)
path.parent.mkdir(parents=True, exist_ok=True)
path.write_bytes(response.content)
return path