帮我写一个python脚本，脚本的作用是把文件夹内的文件名导出，就是类似目录树一样的东西。在我运行该脚本是，它会先问我文件夹路径，在我输入之后，它就开始工作了，等工作完毕，它会问我如何导出，是导出为xmid文件还是导出为csv

文件。

以下是Python脚本示例：

import os
import csv

# 询问文件夹路径
folder_path = input("请输入文件夹路径：")

# 获取文件夹内所有文件名
file_names = []
for root, dirs, files in os.walk(folder_path):
    for file in files:
        file_names.append(os.path.join(root, file))

# 询问导出方式
export_type = input("请选择导出方式（输入xmid或csv）：")

# 导出为xmid文件
if export_type == "xmid":
    with open("file_names.xmid", "w") as f:
        for name in file_names:
            f.write(name + "\n")
    print("文件名已成功导出为file_names.xmid")

# 导出为csv文件
elif export_type == "csv":
    with open("file_names.csv", "w", newline="") as f:
        writer = csv.writer(f)
        writer.writerow(["文件名"])
        for name in file_names:
            writer.writerow([name])
    print("文件名已成功导出为file_names.csv")

# 导出方式输入错误
else:
    print("输入错误，请重新运行脚本并输入正确的导出方式。")

运行脚本后，会先要求输入文件夹路径，例如：

请输入文件夹路径：D:\Documents\Files

然后会要求选择导出方式，可以输入"xmid"或"csv"，例如：

请选择导出方式（输入xmid或csv）：csv

脚本会根据选择的导出方式将文件名导出为相应格式的文件。例如，选择导出为csv文件时，导出的文件内容如下：

文件名
D:\Documents\Files\file1.txt
D:\Documents\Files\file2.txt
D:\Documents\Files\subfolder1\file3.txt
D:\Documents\Files\subfolder2\file4.txt