From b16bffb1303f06dd363384a977cd9d4d011262b1 Mon Sep 17 00:00:00 2001 From: array-in-a-matrix Date: Mon, 22 Aug 2022 01:14:24 -0400 Subject: [PATCH] improvements --- textgen.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/textgen.py b/textgen.py index 6a41a58..ab55553 100644 --- a/textgen.py +++ b/textgen.py @@ -35,8 +35,13 @@ def train_ai(): match sys.argv[1]: case "generate": # ? send message to parent JS process - print(generate_message(sys.argv[2])) - sys.stdout.flush() + try: + prompt = ' '.join(map(str, sys.argv[2:])) + print(generate_message(prompt)) + except IndexError: + print(generate_message()) + finally: + sys.stdout.flush() case "train": train_ai() case _: