Update app.py

app.py

@@ -170,15 +170,15 @@ with gr.Blocks() as demo:
                             else:
                                 history[-1][1] += delta
 
-                            # if refusal_condition(history[-1][1]):
-                            #     history = [['[安全拒答啟動]', '[安全拒答啟動] 請清除再開啟對話']]
-                            #     yield history
-                            #     break
                             yield history
                     
             if history[-1][1].endswith('</s>'):
                 history[-1][1] = history[-1][1][:-4]
                 yield history
+
+            if refusal_condition(history[-1][1]):
+                history[-1][1] = history[-1][1] + '\n\n**系統指示：以上內容有錯誤的疑慮**'
+                yield history
     
         print('== Record ==\nQuery: {query}\nResponse: {response}'.format(query=repr(message), response=repr(history[-1][1])))