Update app.py

app.py

@@ -656,19 +656,55 @@ def upload_file_to_memory():
         return jsonify({"error": str(e)}), 500
 
 # 📤 Маршрут для получения последнего изображения из памяти
+latest_image = {
+    "data": None,
+    "filename": "",
+    "color_percentages": {"green": 0, "yellow": 0, "brown": 0}
+}
+
+def analyze_colors(image_bytes):
+    image_bytes.seek(0)
+    file_bytes = np.asarray(bytearray(image_bytes.read()), dtype=np.uint8)
+    img = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
+
+    # Преобразуем в HSV
+    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
+
+    # Диапазоны цветов в HSV
+    green_mask = cv2.inRange(hsv, (36, 40, 40), (86, 255, 255))
+    yellow_mask = cv2.inRange(hsv, (15, 40, 40), (35, 255, 255))
+    brown_mask = cv2.inRange(hsv, (10, 100, 20), (20, 255, 200))
+
+    total_pixels = img.shape[0] * img.shape[1]
+
+    green_percent = int((cv2.countNonZero(green_mask) / total_pixels) * 100)
+    yellow_percent = int((cv2.countNonZero(yellow_mask) / total_pixels) * 100)
+    brown_percent = int((cv2.countNonZero(brown_mask) / total_pixels) * 100)
+
+    return {
+        "green": green_percent,
+        "yellow": yellow_percent,
+        "brown": brown_percent
+    }
+
 @app.route('/last_image', methods=['GET'])
 def get_last_image():
     if latest_image["data"] is None:
         return jsonify({"error": "No image available"}), 404
 
-    # Создаем копию потока
+    # Анализ цвета
     latest_image["data"].seek(0)
     image_copy = BytesIO(latest_image["data"].read())
     image_copy.seek(0)
+    latest_image["color_percentages"] = analyze_colors(BytesIO(image_copy.getvalue()))
 
+    # Возврат изображения
+    image_copy.seek(0)
     return send_file(image_copy, mimetype='image/jpeg', download_name=latest_image["filename"])
 
-    return send_file(latest_image["data"], mimetype='image/jpeg', download_name=latest_image["filename"])
+@app.route('/color_stats', methods=['GET'])
+def color_stats():
+    return jsonify(latest_image["color_percentages"])
 
 
 @app.route('/view_image', methods=['GET'])