control of context segment values depending on document type or document style

Summary:

I have a case that i need to limit context value depending on document type or style for example we have two contexts values X & Y we need context X to appear on PO only and context Y to appear on Agreement only , So can we achieve that ? i tries sandboxes but it didnt work i may write something wrong in the groovy!

Content (required):

I have a case that i need to limit context value depending on document type or style for example we have two contexts values X & Y we need context X to appear on PO only and context Y to appear on Agreement only , So can we achieve that ? i tries sandboxes but it didnt work i may write something wrong in the groovy!