zmqTextClient.py

#!/usr/bin/env python

## Life AI Text to Speech test for ZMQ module
#
# Chris Kennedy 2023 (C) GPL
#
# Free to use for any use as in truly free software
# as Richard Stallman intended it to be.
#

import zmq
import argparse
import uuid
import time

def main():
    context = zmq.Context()
    socket = None

    # Socket to send messages on
    if args.pub:
        socket = context.socket(zmq.PUB)
        print("Send message: %s:%d" % (args.output_host, args.output_port))
        if args.bind_output:
            socket.bind(f"tcp://{args.output_host}:{args.output_port}")
        else:
            socket.connect(f"tcp://{args.output_host}:{args.output_port}")
    else:
        socket = context.socket(zmq.PUSH)
        print("Send message: %s:%d" % (args.output_host, args.output_port))
        if args.bind_output:
            socket.bind(f"tcp://{args.output_host}:{args.output_port}")
        else:
            socket.connect(f"tcp://{args.output_host}:{args.output_port}")

    history = []
    aipersonality = args.ai_personality
    ainame = args.ai_name

    # Send the message
    client_request = {
        "segment_number": 0,
        "mediaid": args.id,
        "mediatype": args.type,
        "username": args.username,
        "source": args.source,
        "message": args.message,
        "aipersonality": aipersonality,
        "ainame": ainame,
        "episode": args.episode,
        "history": history,
        "text": args.text,
    }

    print(f"Looping every {args.seconds} seconds. Press Ctrl-C to exit.")
    while True:
        socket.send_json(client_request)
        time.sleep(args.seconds)

if __name__ == "__main__":
    default_id = uuid.uuid4().hex[:8]

    parser = argparse.ArgumentParser()
    parser.add_argument("--output_port", type=int, default=1500, required=False, help="Port to send message to")
    parser.add_argument("--output_host", type=str, default="127.0.0.1", required=False, help="Host for sending message to.")
    parser.add_argument("--message", type=str, required=False, default="Story about naruto.", help="The original prompt generating the text.")
    parser.add_argument("--id", type=str, required=False, default=default_id, help="ID of the message")
    parser.add_argument("--type", type=str, required=False, default="chat", help="Type of message")
    parser.add_argument("--username", type=str, required=False, default="anonymous", help="Username of sender")
    parser.add_argument("--source", type=str, required=False, default="lifeAI", help="Source of message")
    parser.add_argument("--ai_personality", type=str, required=False, default="I am GAIB the AI Bot of Life AI, I am helpful and approach the chat with love, compassion, equinimity, joy and courage.", help="AI personality")
    parser.add_argument("--ai_name", type=str, required=False, default="GAIB", help="AI name")
    parser.add_argument("--text", type=str, required=False, default="As he walked down the street he saw a firehydrant. It reminded him of the past.", help="AI name")
    parser.add_argument("-ll", "--loglevel", type=str, default="info", help="Logging level: debug, info...")
    parser.add_argument("--seconds", type=int, default=60, help="Seconds to wait between messages")
    parser.add_argument("--episode", action="store_true", default=False, help="Episode mode")
    parser.add_argument("--pub", action="store_true", default=False, help="Publish to a topic")
    parser.add_argument("--bind_output", action="store_true", default=False, help="Bind to a topic")

    args = parser.parse_args()

    main()