如何在tkinter中将用户输入目录作为函数返回？

from tkinter import Tk
from tkinter import filedialog

def get_directory():
    root = Tk()
    root.withdraw()  # 隐藏根窗口
    directory = filedialog.askdirectory()  # 打开目录选择对话框
    return directory

selected_directory = get_directory()
print(selected_directory)