TCP client CLI with yield example in python

yield.py

import socket

def make_stream(addr):
    c = socket.create_connection(("0",5001))
    print("Connected!")
    reply = c
    while True:
        request = yield reply
        if not request.strip():
            break
        c.sendall(request)
        if not (reply := c.recv(1024)):
            break

def cli():
    stream = make_stream(("0", 5001))
    prepare = stream.send(None)
    while True:
        message = input("> ").encode()
        reply = stream.send(message)
        print(reply.decode())


cli()