Update app.py

app.py

@@ -6,6 +6,7 @@ import os
 from PIL import Image
 from deep_translator import GoogleTranslator
 import json
+from langdetect import detect
 
 api_base = os.getenv("API_BASE")
 mmodels = {
@@ -83,8 +84,11 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
             print(f"Error: {response.status_code} - {response.text}")
     API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN"), os.getenv("HF_READ_TOKEN_2"), os.getenv("HF_READ_TOKEN_3"), os.getenv("HF_READ_TOKEN_4"), os.getenv("HF_READ_TOKEN_5")]) # it is free
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
-    prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
-    print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')
+    language = detect(prompt)
+    
+    if language != 'en':
+        prompt = GoogleTranslator(source=language, target='en').translate(prompt)
+        print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')
 
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
     print(f'\033[1mГенерация {key}:\033[0m {prompt}')