Spaces:
Runtime error
Runtime error
File size: 2,923 Bytes
6ada2bc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | #!/usr/bin/env python3
"""CLI — Generate image from text prompt (T2I).
Usage:
python -m cli.image "A cat wearing sunglasses on a beach"
python -m cli.image "Dragon in cyberpunk city" --aspect landscape --count 4
python -m cli.image "Logo design" --aspect square -o logo.png
"""
import argparse
import asyncio
import os
import sys
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from omniflash import ExtensionBridge, DEFAULT_PROJECT
from omniflash.generators.t2i import generate_image, download_image, IMAGE_ASPECTS
async def run(args):
aspect = args.aspect
bridge = ExtensionBridge()
await bridge.start()
if not await bridge.wait_for_extension(timeout=30):
return
# Handle ref images: auto-upload local files
ref_ids = []
if args.ref:
from omniflash.generators.i2v import upload_image
for ref in args.ref:
if os.path.exists(ref):
print(f"📤 Uploading reference: {ref}")
mid = await upload_image(bridge, ref)
if mid:
ref_ids.append(mid)
print(f" media_id={mid[:12]}...")
else:
# Assume it's already a media_id
ref_ids.append(ref)
results = await generate_image(
bridge, args.prompt, aspect, args.project_id,
count=args.count, ref_media_ids=ref_ids or None
)
if not results:
await bridge.close()
return
os.makedirs(os.path.dirname(args.output) or ".", exist_ok=True)
for i, r in enumerate(results):
if not r.get("image_url"):
print(f"⚠️ Image {i+1}: no URL")
continue
if len(results) == 1:
out_path = args.output
else:
base, ext = os.path.splitext(args.output)
out_path = f"{base}_{i+1}{ext}"
if await download_image(bridge, r["image_url"], out_path):
print(f"🎉 Done! {out_path}")
await bridge.close()
def main():
parser = argparse.ArgumentParser(description="Flow Agent — Image Generator")
parser.add_argument("prompt", help="Text prompt for image")
parser.add_argument("--output", "-o", default="output/image.png", help="Output file")
parser.add_argument("--aspect", "-a",
choices=list(IMAGE_ASPECTS.keys()),
default="portrait",
help="Aspect ratio")
parser.add_argument("--count", "-c", type=int, choices=[1, 2, 3, 4], default=1,
help="Generate 1-4 variations")
parser.add_argument("--ref", "-r", nargs="+", metavar="IMAGE",
help="Reference image(s): file path or media_id")
parser.add_argument("--project-id", "-p", default=DEFAULT_PROJECT)
args = parser.parse_args()
asyncio.run(run(args))
if __name__ == "__main__":
main()
|