Update app.py

app.py

@@ -142,7 +142,7 @@ with gr.Blocks() as demo:
 
 
     def user(user_message, history):
-        return "", history + [[user_message, None]]
+        return "", history + [[user_message, '']]
 
 
     def bot(history, max_new_tokens, temperature, top_p, system_prompt):
@@ -159,8 +159,10 @@ with gr.Blocks() as demo:
         message = tokenizer.apply_chat_template(chat_data, tokenize=False)
         message = message[3:]  # remove SOT token
 
+        response = None
         if refusal_condition(history[-1][0]):
             history = [['[安全拒答啟動]', '[安全拒答啟動] 請清除再開啟對話']]
+            response = '[REFUSAL]'
             yield history
         else:
             data = {
@@ -198,12 +200,10 @@ with gr.Blocks() as demo:
                         except Exception as e:
                             raise e
 
-                        if history[-1][1] is None:
-                            history[-1][1] = delta
-                        else:
-                            history[-1][1] += delta
+                        history[-1][1] += delta
                         yield history
-                    
+
+            response = history[-1][1]
             if history[-1][1].endswith('</s>'):
                 history[-1][1] = history[-1][1][:-4]
                 yield history
@@ -213,8 +213,8 @@ with gr.Blocks() as demo:
                 yield history
 
         print('== Record ==\nQuery: {query}\nResponse: {response}'.format(query=repr(message), response=repr(history[-1][1])))
-        insert_to_db(message, history[-1][1], float(temperature), float(top_p))
-
+        insert_to_db(message, response, float(temperature), float(top_p))
+        
     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         fn=bot,
         inputs=[