Update app.py

app.py

@@ -13,6 +13,10 @@ import uuid
 from io import BytesIO
 
 
+import cv2
+import numpy as np
+
+
 from werkzeug.utils import secure_filename
 
 
@@ -652,19 +656,62 @@ 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 = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
+
+    # Оптимизированные, но безопасные диапазоны цветов
+    color_ranges = {
+        "green": ((36, 40, 40), (86, 255, 255)),     # Зеленый (оставил как было)
+        "yellow": ((22, 100, 100), (32, 255, 255)),   # Желтый (немного расширил)
+        "orange": ((10, 120, 120), (20, 255, 255)),   # Оранжевый (чуть уменьшил насыщенность)
+        "brown": ((5, 50, 20), (15, 150, 150))        # Коричневый (оставил почти как было)
+    }
+
+    total_pixels = img.shape[0] * img.shape[1]
+    results = {}
+
+    for color, (lower, upper) in color_ranges.items():
+        mask = cv2.inRange(hsv, np.array(lower), np.array(upper))
+        percent = round(cv2.countNonZero(mask) / total_pixels * 100, 1)
+        results[color] = percent
+
+    return results
+
 @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')
+def color_stats():
+    stats = latest_image["color_percentages"]
+    print(f"Raw color stats: {stats}")  # Для отладки в консоли
+    return jsonify(stats)
+
+
+@app.route('/view_image', methods=['GET'])
+def view_image():
+    return render_template('show_image.html')
+