Woody Allen left ‘extremely distraught and surprised and upset’ at the Annie Hall star and ex-girlfriend’s death