Improve `widget` module under `proof_system`

[ZER0]

Currently there is a lot of code repetition and complexity under proof_system/widget modules, it should be definitely possible to reduce and improve the ergonomics of the code.