improvements

textgen.py

@@ -35,7 +35,12 @@ def train_ai():
 match sys.argv[1]:
     case "generate":
         # ? send message to parent JS process
-        print(generate_message(sys.argv[2]))
+        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()