Update app.py

app.py

@@ -23,7 +23,14 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
     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}"}
     key = random.randint(0, 999)
+    
+    language = detect(prompt)
+    
+    if language == 'ru':
+        prompt = GoogleTranslator(source='ru', 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}')
     if model == 'DALL-E 3 XL':
         API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
@@ -57,19 +64,13 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         API_URL = "https://api-inference.huggingface.co/models/DoctorDiffusion/doctor-diffusion-s-controllable-vector-art-xl-lora"
     if model == 'Disney':
         API_URL = "https://api-inference.huggingface.co/models/goofyai/disney_style_xl"
-        prompt = f"Стиль Disney. {prompt}"
+        prompt = f"Disney style. {prompt}"
     if model == 'CleanLinearMix':
         API_URL = "https://api-inference.huggingface.co/models/digiplay/CleanLinearMix_nsfw"
     if model == 'OrangeMixs':
         API_URL = "https://api-inference.huggingface.co/models/WarriorMama777/OrangeMixs"
 
-    prompt = f"{prompt} | ультра детализация, ультра проработка, ультра качество, идеально."
     
-    language = detect(prompt)
-    
-    if language == 'ru':
-        prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
-        print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')
     
     
     payload = {
@@ -125,7 +126,7 @@ with gr.Blocks(css=css) as dalle:
 
     with gr.Tab("Информация"):
         with gr.Row():
-            gr.Textbox(label="Шаблон prompt", value="{prompt} | ультра детализация, ультра проработка, ультра качество, идеально.")
+            gr.Textbox(label="Шаблон prompt", value="{prompt} | ultra detail, ultra elaboration, ultra quality, perfect.")
 
     with gr.Row():
         text_button = gr.Button("Генерация", variant='primary', elem_id="gen-button")