如何在python中返回主菜单前显示绘图

import turtle

def draw_menu():
    # 绘制主菜单的代码
    # ...

def draw_graph():
    # 绘制图形的代码
    # ...

# 绘制主菜单
draw_menu()

# 等待用户选择菜单项
menu_choice = input("请选择菜单项：")

if menu_choice == "1":
    # 绘制图形
    draw_graph()
    # 显示绘图并等待用户关闭窗口
    turtle.done()
elif menu_choice == "2":
    # 其他菜单项的处理
    # ...
else:
    print("无效的菜单项")