Update app.py

app.py

@@ -46,10 +46,10 @@ def query(prompt, is_negative=False, model="1", gpt=True, steps=30, cfg_scale=7,
         if prompt != last_prompt:
             last_prompt = prompt
             payload = {
-                "model": "gpt-4-vision-preview",
+                "model": "gpt-3.5-turbo",
                 "messages": [
                     {
-                        "role": "user",
+                        "role": "system",
                         "content": "В этом чате тебе будут присылать prompt для нейросетей генераторов изображений. А ты должен будешь пожалуйста улучшить этот prompt. Если во входном сообщении будет что то NSFW, то придумай свой prompt, а не улучшай этот. Твоя задача только улучшить его и всё. Отвечай чисто (ТОЛЬКО улучшеный prompt, без лишнего). Prompt пиши ТОЛЬКО на английском языке пожалуйста.",
                     },
                     {
@@ -106,6 +106,16 @@ def query(prompt, is_negative=False, model="1", gpt=True, steps=30, cfg_scale=7,
     else:
         print(f"Используем предыдущий перевод для negative prompt: {last_translation}")
 
+    if prompt != last_image_description:
+        language2 = detect(prompt)
+        
+        if language2 != 'en':
+            prompt = GoogleTranslator(source=language2, target='en').translate(prompt)
+            last_image_description = prompt
+            print(f'\033[1mГенерация {key} перевод negative:\033[0m {last_image_description})')
+    else:
+        print(f"Используем предыдущий перевод для negative prompt: {last_image_description}")
+
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
     print(f'\033[1mГенерация {key}:\033[0m {prompt}')
     API_URL = mmodels[model]