Update app.py

app.py

@@ -623,6 +623,7 @@ def uploaded_file(filename):
 
 
 # 🧠 2. Маршрут: сохраняет файл только в память (BytesIO)
+# Маршрут для загрузки файла в память
 @app.route('/upload_memory', methods=['POST'])
 def upload_file_to_memory():
     if 'file' not in request.files:
@@ -632,80 +633,45 @@ def upload_file_to_memory():
     if file.filename == '':
         return jsonify({"error": "No selected file"}), 400
 
-    if not allowed_file(file.filename):
+    # Проверка типа файла (разрешаем только изображения)
+    if not file.filename.lower().endswith(('.jpg', '.jpeg', '.png')):
         return jsonify({"error": "Invalid file type"}), 400
 
+    # Создаем имя файла с timestamp
     timestamp = datetime.now().strftime('%Y.%m.%d_%H:%M:%S_')
     filename = timestamp + file.filename
 
+    # Сохраняем в память
     memory_buffer = BytesIO()
     try:
-        while chunk := file.read(1024):
-            memory_buffer.write(chunk)
-
+        file.save(memory_buffer)
         memory_buffer.seek(0)
+        
+        # Обновляем последнее изображение
         latest_image["data"] = memory_buffer
         latest_image["filename"] = filename
 
         return jsonify({
-            "message": "Image uploaded to memory",
+            "message": "Image uploaded successfully",
             "filename": filename
         }), 200
 
     except Exception as e:
         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"])
-
-@app.route('/color_stats')
-def color_stats():
-    stats = latest_image["color_percentages"]
-    print(f"Raw color stats: {stats}")  # Для отладки в консоли
-    return jsonify(stats)
+    return send_file(
+        latest_image["data"],
+        mimetype='image/jpeg',
+        download_name=latest_image["filename"]
+    )
+
 
 
 @app.route('/view_image', methods=['GET'])