add skipH

GalTransl/Backend/BingGPT4Translate.py

@@ -21,6 +21,7 @@
     NewBing_NAME_PROMPT3,
     NewBing_PROOFREAD_PROMPT,
     NewBing_TRANS_PROMPT,
+    H_WORDS_LIST
 )
 
 
@@ -60,6 +61,11 @@ def __init__(
             self.force_NewBing_hs_mode = val
         else:
             self.force_NewBing_hs_mode = False
+        # 跳过h
+        if val := config.getKey("skipH"):
+            self.skipH = val
+        else:
+            self.skipH = False
         # 跳过重试
         if val := config.getKey("skipRetry"):
             self.skipRetry = val
@@ -385,6 +391,14 @@ async def batch_translate(
             proofread=proofread,
             retran_key=retran_key,
         )
+        if self.skipH:
+            LOGGER.warning("skipH: 将跳过含有敏感词的句子")
+            trans_list_unhit = [
+                tran
+                for tran in trans_list_unhit
+                if not any(word in tran.post_jp for word in H_WORDS_LIST)
+            ]
+
         if len(trans_list_unhit) == 0:
             return []
         # 新文件重置chatbot

GalTransl/Backend/GPT3Translate.py

@@ -28,6 +28,7 @@
     GPT35_1106_TRANS_PROMPT,
     GPT35_0125_SYSTEM_PROMPT,
     GPT35_0125_TRANS_PROMPT,
+    H_WORDS_LIST,
 )
 
 
@@ -71,6 +72,11 @@ def __init__(
             self.restore_context_mode = val
         else:
             self.restore_context_mode = False
+        # 跳过h
+        if val := config.getKey("skipH"):
+            self.skipH = val
+        else:
+            self.skipH = False
         # 跳过重试
         if val := config.getKey("skipRetry"):
             self.skipRetry = val
@@ -355,7 +361,9 @@ async def asyncTranslate(self, content: CTransList, gptdict="") -> CTransList:
                     LOGGER.warning("-> 单句仍错，重置会话")
                 # 单句5次重试则中止
                 if self.retry_count == 5:
-                    raise RuntimeError(f"-> 单句反复出错，已中止。最后错误为：{error_message}")
+                    raise RuntimeError(
+                        f"-> 单句反复出错，已中止。最后错误为：{error_message}"
+                    )
                 continue
 
             if warn_flag:
@@ -494,6 +502,15 @@ async def batch_translate(
             retry_failed=retry_failed,
             retran_key=retran_key,
         )
+
+        if self.skipH:
+            LOGGER.warning("skipH: 将跳过含有敏感词的句子")
+            trans_list_unhit = [
+                tran
+                for tran in trans_list_unhit
+                if not any(word in tran.post_jp for word in H_WORDS_LIST)
+            ]
+
         if len(trans_list_unhit) == 0:
             return []
 

GalTransl/Backend/GPT4Translate.py

@@ -25,6 +25,7 @@
     GPT4Turbo_TRANS_PROMPT,
     GPT4Turbo_CONF_PROMPT,
     GPT4Turbo_PROOFREAD_PROMPT,
+    H_WORDS_LIST
 )
 
 
@@ -86,6 +87,11 @@ def __init__(
             self.skipRetry = val
         else:
             self.skipRetry = False
+        # 跳过h
+        if val := config.getKey("skipH"):
+            self.skipH = val
+        else:
+            self.skipH = False
         # 流式输出模式
         if val := config.getKey("gpt.streamOutputMode"):
             self.streamOutputMode = val
@@ -436,9 +442,14 @@ async def batch_translate(
             retran_key=retran_key,
         )
 
-        # 校对模式多喂上一行
-        # if proofread and trans_list_unhit[0].prev_tran != None:
-        #    trans_list_unhit.insert(0, trans_list_unhit[0].prev_tran)
+        if self.skipH:
+            LOGGER.warning("skipH: 将跳过含有敏感词的句子")
+            trans_list_unhit = [
+                tran
+                for tran in trans_list_unhit
+                if not any(word in tran.post_jp for word in H_WORDS_LIST)
+            ]
+
         if len(trans_list_unhit) == 0:
             return []
         # 新文件重置chatbot