You can use the language SQL instead of PYTHON as its the supported language for stored procedures.

SQL stored procedures are good for scripting & creating modular SQL based workflows within boundary of Unity Catalog. It's a secure governed way to execute SQL without leaving warehouse.

However, if the workflows need advanced code logic, complex loops or manipulation of the Spark session, you can use databricks notebook running Python spark code as it gives access to the full power of Spark helping you create scalable & complex data pipelines that SQL cannot support.

More details here