Update app.py

app.py

@@ -1,6 +1,7 @@
 import numpy as np
 import os
 import re
+import jieba
 from io import BytesIO
 import datetime
 import time
@@ -60,11 +61,23 @@ class Reviewer:
         print("completion_token_used:", response.usage.completion_tokens)
         print("total_token_used:", response.usage.total_tokens)
         print("response_time:", response.response_ms/1000.0, 's')    
-        output_text = insert_sentence(result, '(Generated by ChatGPT, no copying allowed!)', 10)
+        output_text = insert_sentence(result, '**Generated by ChatGPT, no copying allowed!**', 10)
         return output_text, response.usage.total_tokens        
 
+    def contains_chinese(text):
+        for ch in text:
+            if u'\u4e00' <= ch <= u'\u9fff':
+                return True
+        return False
+    
     def insert_sentence(text, sentence, interval):
-        words = text.split()
+        if contains_chinese(text):
+            words = list(jieba.cut(text))
+            separator = ''
+        else:
+            words = text.split()
+            separator = ' '
+    
         new_words = []
         count = 0
     
@@ -75,7 +88,7 @@ class Reviewer:
             if count % interval == 0:
                 new_words.append(sentence)
     
-        return ' '.join(new_words)
+        return separator.join(new_words)