Update app.py

app.py

@@ -48,6 +48,14 @@ FREQUENCY_PENALTY=0
 model_name = "MediaTek-Research/Breeze-7B-Instruct-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
+def refusal_condition(query):
+    is_including_tw = '台灣' in query or '台湾' in query or 'taiwan' in query.lower() or 'tw' in query.lower()
+    is_including_cn = '中國' in query or '中国' in query or 'china' in query.lower() or 'cn' in query.lower()
+    if is_including_tw and is_including_cn:
+        return True
+
+    return False
+
 with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
 
@@ -113,6 +121,10 @@ with gr.Blocks() as demo:
             if assistant_msg is not None:
                 chat_data.append({"role": "assistant", "content": assistant_msg})
 
+        if refusal_condition(history[-1][0]):
+            history[-1][1] = '抱歉，我無法回答您這個問題'
+            return history
+
         message = tokenizer.apply_chat_template(chat_data, tokenize=False)
         message = message[3:]  # remove SOT token