Bruce Willis "wanted to work" and was "capable of continuing to work" after being diagnosed with aphasia, his lawyer has insisted.