帮我写个简易火车票订购系统Python简单的代码加个退票功能

好的，以下是一个简易的火车票订购系统的Python代码，包括订票和退票功能：

# 火车票订购系统

# 火车票库存
tickets = {
    '北京-上海': 5,
    '北京-广州': 3,
    '北京-深圳': 2,
    '上海-广州': 4,
    '上海-深圳': 3,
    '广州-深圳': 6
}

# 订票函数
def book_ticket():
    print('欢迎使用火车票订购系统！')
    print('目前的火车票库存如下：')
    for k, v in tickets.items():
        print(k, v)
    start = input('请输入起点站：')
    end = input('请输入终点站：')
    key = start + '-' + end
    if key not in tickets:
        print('抱歉，该线路目前无法预订。')
    elif tickets[key] == 0:
        print('抱歉，该线路目前已售完。')
    else:
        num = int(input('请输入要订购的票数：'))
        if num > tickets[key]:
            print('抱歉，该线路目前只剩余{}张票，请重新输入。'.format(tickets[key]))
        else:
            tickets[key] -= num
            print('恭喜，订票成功！您订购了{}张从{}到{}的火车票。'.format(num, start, end))
            print('目前的火车票库存如下：')
            for k, v in tickets.items():
                print(k, v)

# 退票函数
def return_ticket():
    print('欢迎使用火车票退票系统！')
    print('目前您已订购的火车票如下：')
    for k, v in tickets.items():
        if v < 5:
            print(k, 5 - v)
    start = input('请输入要退票的起点站：')
    end = input('请输入要退票的终点站：')
    key = start + '-' + end
    if key not in tickets:
        print('抱歉，您没有订购过该线路的火车票。')
    else:
        num = int(input('请输入要退票的张数：'))
        tickets[key] += num
        print('恭喜，退票成功！您已成功退掉了{}张从{}到{}的火车票。'.format(num, start, end))
        print('目前的火车票库存如下：')
        for k, v in tickets.items():
            print(k, v)

# 主程序
while True:
    print('欢迎使用火车票订购系统！')
    print('请选择您要使用的功能：')
    print('1. 订票')
    print('2. 退票')
    print('3. 退出系统')
    choice = input()
    if choice == '1':
        book_ticket()
    elif choice == '2':
        return_ticket()
    elif choice == '3':
        print('感谢使用火车票订购系统，再见！')
        break
    else:
        print('输入有误，请重新输入。')

希望能帮到你！