Update app.py

app.py

@@ -19,6 +19,8 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
     if prompt == "" or prompt == None:
         return None
 
+    key = random.randint(0, 999)
+    
     if gpt:
         payload = {
             "model": "gpt-4-vision-preview",
@@ -57,6 +59,7 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
             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}")
         else:
@@ -66,7 +69,6 @@ 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)
     
     prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
     print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')