Update app.py

app.py

@@ -436,6 +436,7 @@ def to_tensor(image):
 
 import unicodedata
 
+
 def full2half(text):
     half = []
     for char in text:
@@ -448,8 +449,21 @@ def full2half(text):
             half.append(char)
     return ''.join(half)
 
+def has_chinese_char(string):
+    pattern = re.compile('[\u4e00-\u9fa5]')
+    if pattern.search(string):
+        return True
+    else:
+        return False
+
+image_404 = Image.open('file/images/404.jpg')
+
 def text_to_image(prompt,slider_step,slider_guidance,slider_batch):
 
+    if has_chinese_char(prompt):
+        print('trigger')
+        return image_404, None
+
     prompt = full2half(prompt)
     prompt = prompt.replace('"', "'")
     prompt = prompt.replace('‘', "'")
@@ -562,6 +576,10 @@ print(f'{colored("[√]", "green")} Text segmenter is successfully loaded.')
 
 def text_to_image_with_template(prompt,template_image,slider_step,slider_guidance,slider_batch, binary):
 
+    if has_chinese_char(prompt):
+        print('trigger')
+        return image_404, None
+    
     if slider_step>=50:
         slider_step = 50
         
@@ -605,7 +623,8 @@ def text_to_image_with_template(prompt,template_image,slider_step,slider_guidanc
         binary = gray.point(lambda x: 255 if x > 96 else 0, '1')
         template_image = binary.convert('RGB')
 
-    # to_tensor = transforms.ToTensor()
+    # 
+        = transforms.ToTensor()
     image_tensor = to_tensor(template_image).unsqueeze(0).cuda().sub_(0.5).div_(0.5) # (b, 3, 256, 256)
             
     with torch.no_grad():
@@ -663,6 +682,10 @@ def text_to_image_with_template(prompt,template_image,slider_step,slider_guidanc
 
 def text_inpainting(prompt,orig_image,mask_image,slider_step,slider_guidance,slider_batch):
 
+    if has_chinese_char(prompt):
+        print('trigger')
+        return image_404, None
+        
     if slider_step>=50:
         slider_step = 50