Update app.py

app.py

@@ -8,6 +8,11 @@ from deep_translator import GoogleTranslator
 import json
 from langdetect import detect
 
+last_prompt = None
+last_negative_prompt = None
+last_translation = None
+last_image_description = None
+
 api_base = os.getenv("API_BASE")
 mmodels = {
     "1": "openskyml/dalle-3-xl",
@@ -31,64 +36,75 @@ mmodels = {
     "Arcane": "nitrosocke/Arcane-Diffusion"
 }
 def query(prompt, is_negative=False, model="1", gpt=True, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
+    global last_prompt, last_negative_prompt, last_translation, last_image_description
     if prompt == "" or prompt == None:
         raise gr.Error("🙏 Опиши пожалуйста как должно выглядеть изображение")
         return None
 
     key = random.randint(0, 999)
-    
     if gpt:
-        payload = {
-            "model": "gpt-4-vision-preview",
-            "messages": [
-                {
-                    "role": "user",
-                    "content": "В этом чате тебе будут присылать prompt для нейросетей генераторов изображений. А ты должен будешь пожалуйста улучшить этот prompt. Если во входном сообщении будет что то NSFW, то придумай свой prompt, а не улучшай этот. Твоя задача только улучшить его и всё. Отвечай чисто (ТОЛЬКО улучшеный prompt, без лишнего). Prompt пиши ТОЛЬКО на английском языке пожалуйста.",
-                },
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
-            "max_tokens": 4095,
-        }    
-
-     
-        # API ключ для OpenAI
-        api_key_oi = os.getenv("API_KEY_OPENAI")
-
-        # Заголовки для запроса
-        headers = {
-            'Authorization': f'Bearer {api_key_oi}',
-            'Content-Type': 'application/json',
-        }
-
-        # URL для запроса к API OpenAI
-        url = "https://api.openai.com/v1/chat/completions"
-
-        # Отправляем запрос в OpenAI
-        response = requests.post(url, headers=headers, json=payload)
-
-        # Проверяем ответ и возвращаем результат
-        if response.status_code == 200:
-            response_json = response.json()
-            try:
-                # Пытаемся извлечь текст из ответа
-                prompt = response_json["choices"][0]["message"]["content"]
-                print(f'Генерация {key} gpt: {prompt}')
-            except Exception as e:
-                print(f"Error processing the image response: {e}")
-                raise gr.Info("🙄 Помогатор куда-то отошёл. Придётся без него...")
+        if prompt != last_prompt:
+            last_prompt = prompt
+            payload = {
+                "model": "gpt-4-vision-preview",
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": "В этом чате тебе будут присылать prompt для нейросетей генераторов изображений. А ты должен будешь пожалуйста улучшить этот prompt. Если во входном сообщении будет что то NSFW, то придумай свой prompt, а не улучшай этот. Твоя задача только улучшить его и всё. Отвечай чисто (ТОЛЬКО улучшеный prompt, без лишнего). Prompt пиши ТОЛЬКО на английском языке пожалуйста.",
+                    },
+                    {
+                        "role": "user",
+                        "content": prompt,
+                    }
+                ],
+                "max_tokens": 4095,
+            }    
+    
+         
+            # API ключ для OpenAI
+            api_key_oi = os.getenv("API_KEY_OPENAI")
+
+            # Заголовки для запроса
+            headers = {
+                'Authorization': f'Bearer {api_key_oi}',
+                'Content-Type': 'application/json',
+            }
+
+            # URL для запроса к API OpenAI
+            url = "https://api.openai.com/v1/chat/completions"
+
+            # Отправляем запрос в OpenAI
+            response = requests.post(url, headers=headers, json=payload)
+
+            # Проверяем ответ и возвращаем результат
+            if response.status_code == 200:
+                response_json = response.json()
+                try:
+                    # Пытаемся извлечь текст из ответа
+                    prompt = response_json["choices"][0]["message"]["content"]
+                    print(f'Генерация {key} gpt: {prompt}')
+                except Exception as e:
+                    print(f"Error processing the image response: {e}")
+                    raise gr.Info("🙄 Помогатор куда-то отошёл. Придётся без него...")
+            else:
+                # Если произошла ошибка, возвращаем сообщение об ошибке
+                print(f"Error: {response.status_code} - {response.text}")
+            last_image_description = prompt
         else:
-            # Если произошла ошибка, возвращаем сообщение об ошибке
-            print(f"Error: {response.status_code} - {response.text}")
+            prompt = last_image_description
+            print(f"Используем предыдущее описание изображения: {prompt}")
+        
     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}"}
-    language = detect(is_negative)
-    
-    if language != 'en':
-        is_negative = GoogleTranslator(source=language, target='en').translate(is_negative)
-        print(f'\033[1mГенерация {key} перевод negative:\033[0m {is_negative}')
+    if is_negative != last_negative_prompt:
+        last_negative_prompt = is_negative
+        language = detect(is_negative)
+        
+        if language != 'en':
+            last_translation = GoogleTranslator(source=language, target='en').translate(is_negative)
+            print(f'\033[1mГенерация {key} перевод negative:\033[0m {last_translation}')
+    else:
+        print(f"Используем предыдущий перевод для negative prompt: {last_translation}")
 
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
     print(f'\033[1mГенерация {key}:\033[0m {prompt}')