Update app.py

app.py

@@ -173,6 +173,10 @@ with gr.Blocks() as demo:
             if history[-1][1].endswith('</s>'):
                 history[-1][1] = history[-1][1][:-4]
                 yield history
+
+            if refusal_condition(history[-1][1]):
+                history = [['[安全拒答啟動]', '[安全拒答啟動] 請清除再開啟對話']]
+                yield history
     
         print('== Record ==\nQuery: {query}\nResponse: {response}'.format(query=repr(message), response=repr(history[-1][1])))