import torch
from diffusers import BriaFiboEditPipeline
from PIL import Image
from src.edit_promptify import get_prompt

# Generate structured JSON using API mode
image = Image.open("photo.jpg")
mask_image = Image.open("mask.jpg").convert("L")
edit_json = get_prompt(image=image, instruction="make it look vintage", mask_image=mask_image)

# Use with Fibo Edit
pipe = BriaFiboEditPipeline.from_pretrained(
    "briaai/Fibo-Edit",
    torch_dtype=torch.bfloat16,
)
pipe.to("cuda")

result = pipe(
    image=image,
    mask=mask,
    prompt=edit_json,
    num_inference_steps=50,
    guidance_scale=5
).images[0]

result.save("edited.png")

import torch
from diffusers import BriaFiboEditPipeline
from PIL import Image
from src.edit_promptify import get_prompt

# Generate structured JSON using API mode
image = Image.open("photo.jpg")
edit_json = get_prompt(image, "make it look vintage")

# Use with Fibo Edit
pipe = BriaFiboEditPipeline.from_pretrained(
    "briaai/Fibo-Edit",
    torch_dtype=torch.bfloat16,
)
pipe.to("cuda")

result = pipe(
    image=image,
    prompt=edit_json,
    num_inference_steps=50,
    guidance_scale=5
).images[0]

result.save("edited.png")

import torch
from diffusers import BriaFiboEditPipeline
from PIL import Image
from src.edit_promptify import get_prompt

# Generate structured JSON using local VLM
image = Image.open("photo.jpg")
edit_json = get_prompt(
    image=image,
    instruction="make it look vintage",
    vlm_mode="local",
    model="briaai/FIBO-edit-prompt-to-JSON"
)

# Use with Fibo Edit
pipe = BriaFiboEditPipeline.from_pretrained(
    "briaai/Fibo-Edit",
    torch_dtype=torch.bfloat16,
)
pipe.to("cuda")

result = pipe(
    image=image,
    prompt=edit_json,
    num_inference_steps=50,
    guidance_scale=5
).images[0]

result.save("edited.png")

# API mode (default)
python src/example_edit.py --images photo.jpg --instructions "change the car color to green"

# Local VLM mode
python src/example_edit.py --vlm-mode local --vlm-model briaai/FIBO-edit-prompt-to-JSON \
    --images photo.jpg --instructions "change the car color to green"

More Examples

Or explicitly specify

edit_json = get_prompt( image=image, instruction="your edit instruction", vlm_mode="api", model="gemini/gemini-2.5-flash" )

Citation

@article{gutflaish2025generating,
  title={Generating an Image From 1,000 Words: Enhancing Text-to-Image With Structured Captions},
  author={Gutflaish, Eyal and Kachlon, Eliran and Zisman, Hezi and Hacham, Tal and Sarid, Nimrod and Visheratin, Alexander and Huberman, Saar and Davidi, Gal and Bukchin, Guy and Goldberg, Kfir and others},
  journal={arXiv preprint arXiv:2511.06876},
  year={2025}
}