Fix printing when provided initial messagesv0.2.2

1 files changed, 4 insertions, 3 deletions

diff --git a/src/gpt_chat_cli/cmd.py b/src/gpt_chat_cli/cmd.py
index 899c705..83cd298 100644
--- a/src/gpt_chat_cli/cmd.py
+++ b/src/gpt_chat_cli/cmd.py
@@ -297,10 +297,10 @@ def interactive(args : Arguments):
 
     initial_message = resolve_initial_message(args.initial_message, interactive=True)
 
-    if initial_message:
-        print( PROMPT, initial_message, sep='', flush=True )
-
     with prompter as prompt:
+        if initial_message:
+            print( prompt.prompt, initial_message, sep='', flush=True )
+
         while True:
             try:
                 if initial_message:
@@ -339,6 +339,7 @@ def interactive(args : Arguments):
 
                 hist.append( ChatMessage(Role.ASSISTANT, response) )
             except KeyboardInterrupt: # Skip to next prompt
+                print()
                 continue
             except EOFError: # Exit on Control-D
                 print()