Update app.py

app.py

@@ -75,11 +75,20 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
         "seed": seed if seed != -1 else random.randint(1, 1000000000)
         }
 
-    image_bytes = requests.post(API_URL, headers=headers, json=payload, timeout=timeout).content
-    image = Image.open(io.BytesIO(image_bytes))
-    print(f'\033[1mГенерация {key} завершена!\033[0m ({prompt})')
-    return image
-
+    response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
+    if response.status_code != 200:
+        print(f"Ошибка: Не удалось получить изображение. Статус ответа: {response.status_code}")
+        print(f"Содержимое ответа: {response.text}")
+        return None
+    
+    try:
+        image_bytes = response.content
+        image = Image.open(io.BytesIO(image_bytes))
+        print(f'\033[1mГенерация {key} завершена!\033[0m ({prompt})')
+        return image
+    except Exception as e:
+        print(f"Ошибка при попытке открыть изображение: {e}")
+        return None
 
 css = """
 * {}