Add libraries to record screen when needed

Iw tests are recorded in staged testing which helps in debugging.

Test: go/abtd/run/L82800000960365904
Bug: 280563885
Change-Id: Idecfc69c37626d5991f09a12d0a43573c898ff00