Do not show JJ's prompt twice in GUI

doctor.py

@@ -8,11 +8,16 @@ from time import sleep
 import sys
 import os
 
-def slowprint(s):
+logfile = None
+
+
+def slowprint(s, logfile=None):
     for c in s+'\n':
         sys.stdout.write(c)
         sys.stdout.flush()
         sleep(0.01)
+    if logfile:
+        logfile.write(s+'\n')
 
 MODEL_NAME = "gpt-3.5-turbo-16k-0613"
 
@@ -40,17 +45,25 @@ conversation = ConversationChain(
 
 sys.stderr.write("{esc}[2J{esc}[H".format(esc=chr(27))) # Clear screen ;)
 sys.stderr.flush()
+
+if len(sys.argv)>1:
+    try:
+        logfile = open(sys.argv[1], "w")
+    except:
+        slowprint("Couldn't open log file '{}'!!!\n".format(sys.argv[1]))
+
 slowprint("""=====================================================================
 Interactive Talk-Program loaded and started.
 Users:
 * [Doctor Kernel]
 * [JJ] <- you
 Hit <Enter> to exit
-=====================================================================""")
-slowprint("Doctor Kernel: {}".format(conversation.predict(input="Hello again, doctor. Please remind me what we were talking about last time.")))
+=====================================================================""", logfile)
+slowprint("Doctor Kernel: {}\n".format(
+    conversation.predict(input="Hello again, doctor. Please remind me what we were talking about last time."), logfile))
 while True:
     try:
-        prompt = input("> ").strip()
+        prompt = input("JJ: > ").strip()
     except EOFError:
         prompt = ""
     if not prompt:
@@ -59,5 +72,6 @@ while True:
 End of Talk
 =====================================================================""")
             sys.exit(0)
-    slowprint("\nJJ: {}".format(prompt))
-    slowprint("Doctor Kernel: {}".format(conversation.predict(input=prompt)))
+    if logfile:
+        logfile.write("JJ: {}\n".format(prompt))
+    slowprint("Doctor Kernel: {}".format(conversation.predict(input=prompt)), logfile)

run.sh

@@ -5,7 +5,7 @@ cd "$(dirname "$0")"
 trap 'echo "Caught Ctrl-C"' INT
 while [ 1 ] ; do
     LOGFILE=log-$(date +%Y%m%d-%H%M%S).txt
-    python doctor.py|tee logs/$LOGFILE
+    python doctor.py logs/$LOGFILE
     # We expect at least
     # * '<Enter>' in the header
     # * first '>' prompt