Update app.py

app.py

@@ -10,18 +10,22 @@ import os
 from PIL import Image
 import io
 import base64
+import re
 
 
 def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     result = {"prompt": prompt,"negative_prompt": negative_prompt,"task": task,"steps": steps,"sampler": sampler,"cfg_scale": cfg_scale,"seed": seed}
     print(result)
+    try:
+        language = detect(prompt)
+        if language == 'ru':
+            prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
+            print(prompt)
+    except:
+        pass
 
-    language = detect(prompt)
+    prompt = re.sub(r'[^a-zA-Zа-яА-Я\s]', '', prompt)
     
-    if language == 'ru':
-        prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
-        print(prompt)
-
     cfg = int(cfg_scale)
     steps = int(steps)
     seed = int(seed)
@@ -65,8 +69,6 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             return photo
 
 
-
-
 def flipp():
     if task == 'Stable Diffusion XL 1.0':
         model = 'sd_xl_base_1.0'