Image Generation with context

@moaz mujahid