basic ChatGPT terminal client

chatgpt.py

#!/bin/python3
import sys
from openai import OpenAI

myPrompt = ' '.join(sys.argv[1:]).strip()
if not myPrompt:
    print("Enter prompt. Enter newline when done")
    myPrompt = ""
    while True:
        a = input("> ")
        if a == "":
            break
        myPrompt +=a+"\n"

print("Me:", myPrompt, file=sys.stderr)

client = OpenAI(api_key = "YOUR KEY HERE")

stream = client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "system", "content": "Output clear, short and concise answers and use the less words possible unless told otherwise"},
        {"role": "system", "content": "If I ask for a script, assume I'm going to pipe it into a file, so don't output anything but the code"},
        {"role": "user", "content": myPrompt}
        ],
    stream=True,
)

try:
    print("ChatGPT: ", end="", flush=True, file=sys.stderr)
    for part in stream:
        print(part.choices[0].delta.content or "", end="", flush=True)
    print()
except KeyboardInterrupt:
    print("\n\nCtrl+C")