Crear ventanas para introducir datos

[Macas]

Muy buenas,

Estoy utilizando el pygame para hacer una pequeña interface gràfica de un programa que tengo echo.

Os explico un poco de que va el tema, es un programa que calcula el menor coste de un punto a otro en este caso es un mapa de metro. Le introduces los nombres de las estaciones iniciales y finales y te saca el trayecto de menor coste

Con el pygame tengo echo que me cambien de color los puntos por los cuales estoy pasando.

Lo que me gustaría hacer es que cuando ejecute el programa me salga una ventana con diferentes botones y inputs para poder introducir los valores que quiera.
Rollo menú, introduzca el incio aquí, final y un botón para ejecutar.

Con el pygame se puede hacer? Me gustaría hacerlo con este ya que ya tengo parte echa.

Un saludo