Update app.py

app.py

@@ -67,7 +67,7 @@ def generate_cached_image(cache_key, prompt, width, height, seed, model_name):
         return image
     except requests.exceptions.RequestException as e:
         app.logger.error(f"Error generating image: {e}")
-        abort(500, description="Error generating image")
+        return None
 
 @app.route('/prompt/<path:prompt>')
 def get_image(prompt):
@@ -84,7 +84,7 @@ def get_image(prompt):
         language = detect(prompt)
     except Exception as e:
         app.logger.error(f"Error detecting language: {e}")
-        abort(500, description="Error detecting language")
+        return send_error_image()
 
     # Переводим промпт, если он не на английском языке
     if language != 'en':
@@ -93,7 +93,7 @@ def get_image(prompt):
             prompt = translator.translate(prompt)
         except Exception as e:
             app.logger.error(f"Error translating prompt: {e}")
-            abort(500, description="Error translating prompt")
+            return send_error_image()
 
     # Масштабируем размеры изображения, если они превышают максимальные значения
     width, height = scale_dimensions(width, height, MAX_WIDTH, MAX_HEIGHT)
@@ -103,9 +103,11 @@ def get_image(prompt):
 
     try:
         image = generate_cached_image(cache_key, prompt, width, height, seed, model_name)
+        if image is None:
+            return send_error_image()
     except Exception as e:
         app.logger.error(f"Error generating image: {e}")
-        abort(500, description="Error generating image")
+        return send_error_image()
 
     img_byte_arr = io.BytesIO()
     image.save(img_byte_arr, format='PNG')
@@ -120,5 +122,22 @@ def get_image(prompt):
 def health_check():
     return "OK", 200
 
+def send_error_image():
+    error_image_url = "https://raw.githubusercontent.com/Igroshka/-/refs/heads/main/img/nuai/errorimg.png"
+    try:
+        response = requests.get(error_image_url)
+        response.raise_for_status()
+        error_image = Image.open(io.BytesIO(response.content))
+        img_byte_arr = io.BytesIO()
+        error_image.save(img_byte_arr, format='PNG')
+        img_byte_arr = img_byte_arr.getvalue()
+        return send_file(
+            io.BytesIO(img_byte_arr),
+            mimetype='image/png'
+        )
+    except Exception as e:
+        app.logger.error(f"Error fetching error image: {e}")
+        abort(500, description="Error fetching error image")
+
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=False)
+    app.run(host='0.0.0.0', port=7860, debug=False)