Update modules/chat_func.py

modules/chat_func.py

@@ -253,6 +253,14 @@ def predict_all(
         status_text = standard_error_msg + str(response)
         return chatbot, history, status_text, all_token_counts
 
+def is_repeated_string(s):
+    n = len(s)
+    for i in range(1, n // 2 + 1):
+        if n % i == 0:
+            sub = s[:i]
+            if sub * (n // i) == s:
+                return True
+    return False
 
 def predict(
     openai_api_key,
@@ -271,8 +279,8 @@ def predict(
     should_check_token_count=True,
 ):  # repetition_penalty, top_k
     logging.info("输入为：" + colorama.Fore.BLUE + f"{inputs}" + colorama.Style.RESET_ALL)
-    if "我是来浪费的" in inputs:
-        print("有人来浪费了")
+    if is_repeated_string(inputs):
+        print("================== 有人来浪费了 ======================")
         yield chatbot+[(inputs, "🖕️🖕️🖕️🖕️🖕️看不起你")], history, "🖕️🖕️🖕️🖕️🖕️🖕️", all_token_counts
         return
     if should_check_token_count: