[GAN] #4 Improve Code Documentation: Add Docstrings to Classes and Functions

[veeradi34]

Description: The code lacks docstrings for several classes and methods. Adding docstrings will help others understand the purpose and functionality of the code better.

Tasks:
Add a docstring to each class explaining its purpose.
Add docstrings to methods explaining their inputs, outputs, and any key logic.