Refactor main.py

[Apoorve73]

I've selected main.py for refactoring, which is a unit of 73 lines of code and 8 branch points. Addressing this will make our codebase more maintainable and improve Better Code Hub's Write Simple Units of Code guideline rating! 👍

Here's the gist of this guideline:

• Definition 📖
  Limit the number of branch points (if, for, while, etc.) per unit to 4.
• Why❓
  Keeping the number of branch points low makes units easier to modify and test.
• How 🔧
  Split complex units with a high number of branch points into smaller and simpler ones.

You can find more info about this guideline in Building Maintainable Software. 📖

---

ℹ️ To know how many other refactoring candidates need addressing to get a guideline compliant, select some by clicking on the 🔲 next to them. The risk profile below the candidates signals (✅) when it's enough! 🏁

---

Good luck and happy coding! ✨ 💯