main.py logging

[Sukanyasingh3]

In the main.py code, Instead of printing success messages for each stage, consider logging more information about the stage, such as the time it took to complete, any important metrics, etc.