Help with cookiecutter Prompts in Databricks Notebooks

[vieiralaura]

Hi everyone! I’m working on using cookiecutter to help us set up consistent project templates on Databricks. So far, it’s mostly working, but I’m struggling with the prompts – they’re not displaying well in the Databricks environment.

I’ve tried some workarounds, but I’m hoping someone has a simpler solution or knows of any future updates to improve compatibility in notebook environment.

Here’s what I’ve done so far:

def get_user_input():
   project_date = input("What is your project date? YYYYMMDD: ")
   client = input("What is your client name?: ")
   id_project = input("What is your project ID? XXX: ")
   return {
       "project_date ": project_date ,
       "client": client ,
       "id_project ": id_project 
   }

cookiecutter(
    repo_url,
    directory='databricks',
    no_input=True,
    extra_context=get_user_input()
)

I’m looking for tips on making these prompts more user-friendly and avoiding repeated input requests. If anyone has done something similar or has tips for handling cookiecutter in Databricks, I’d really appreciate your advice!