This paper presents the development of a computational tool for wellbore stability analysis and mud weight optimization, written in Python language. The tool utilizes well log data as input parameters and correlations to calculate stresses, and performs wellbore stability analysis to create a mud weight window. The computational tool’s performance is evaluated and compared to traditional methods, and a case study is presented. The results demonstrate that the computational tool provides an efficient and accurate solution for wellbore stability analysis and mud weight optimization. Future research directions are also discussed.