Update functions.py

functions.py

@@ -7,6 +7,7 @@ import random
 import time
 import os
 import re
+import sys
 import json
 import copy
 # import sqlite3
@@ -190,7 +191,7 @@ def make_user_message_list_fn(
 
 
 def sequential_chat_once_fn(payload, api_key_text, history, history_md_stable, history_md_stream, tips):
-    print("\n\n")
+    # print("\n\n")
 
     assistant_message = ""
     tips = ""
@@ -200,26 +201,32 @@ def sequential_chat_once_fn(payload, api_key_text, history, history_md_stable, h
         completion = openai.ChatCompletion.create(**payload)
 
         if payload.get('stream'):
+            print('assistant:')
+            # print('->>>')
             is_first=True
             for chunk in completion:
                 if is_first:
                     is_first = False
                     continue
                 if chunk.choices[0].finish_reason is None:
-                    print(chunk.choices[0].delta.content or '', end="")
+                    sys.stdout.write("\r")
+                    # print(chunk.choices[0].delta.content or '', end="")
                     assistant_message += chunk.choices[0].delta.content or ''
+                    print(assistant_message, end="")
                     history_md_stream = make_md_line('assistant', assistant_message)
                     tips = 'streaming'
                     yield assistant_message, history_md_stream, tips, history
                 else:
                     pass
                 pass
+            # print('=>>>')
             print('')
             pass
         else:
             assistant_message = completion.choices[0].message.content
             history_md_stream = make_md_line('assistant', assistant_message)
             tips = 'got'
+            print('assistant:')
             print(assistant_message)
             yield assistant_message, history_md_stream, tips, history
         pass
@@ -231,7 +238,7 @@ def sequential_chat_once_fn(payload, api_key_text, history, history_md_stable, h
         yield assistant_message, history_md_stream, tips, history
         pass
 
-    print("\n\n")
+    # print("\n\n")
 
 
 def sequential_chat_fn(
@@ -297,6 +304,7 @@ def sequential_chat_fn(
 
         for user_message in user_message_list:
             print('')
+            print(f'user({token_text}):')
             print(user_message)
             print('')
             # make the_messages to sent