from pocketflow import Flow
from nodes import GenerateImageNode

def create_generation_flow():
    """Creates a flow for image generation (initial or regeneration)."""
    generate_image_node = GenerateImageNode()
    return Flow(start=generate_image_node)