Parse bucket name from ckpt directory name instead of separate parameter for bucket name

[Yash9060]

Internal Issue: b/352786281

Parse bucket_name from ckpt_dir_path instead of setting it separately. Updated Unit tests, benchmarks code, demos and documentation.

It's a good idea to open an issue first for discussion.

• Tests pass
• Appropriate changes to documentation are included in the PR