如何在Python端口扫描程序中添加提示，要求用户输入要扫描的端口？

import socket

def scan_port(host, port):
    try:
        # 创建socket对象
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        # 设置超时时间为1秒
        s.settimeout(1)
        # 尝试连接目标主机和端口
        result = s.connect_ex((host, port))
        if result == 0:
            print(f"端口 {port} 开放")
        else:
            print(f"端口 {port} 关闭")
        s.close()
    except socket.error:
        print("连接失败")

if __name__ == "__main__":
    host = input("请输入要扫描的主机IP地址或域名: ")
    port = int(input("请输入要扫描的端口号: "))
    scan_port(host, port)