Update app.py

app.py

@@ -624,6 +624,7 @@ def uploaded_file(filename):
 
 # 🧠 2. Маршрут: сохраняет файл только в память (BytesIO)
 # Маршрут для загрузки файла в память
+# Загрузка изображения в память (в виде байтов)
 @app.route('/upload_memory', methods=['POST'])
 def upload_file_to_memory():
     if 'file' not in request.files:
@@ -633,22 +634,18 @@ def upload_file_to_memory():
     if file.filename == '':
         return jsonify({"error": "No selected file"}), 400
 
-    # Проверка типа файла (разрешаем только изображения)
     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:
-        file.save(memory_buffer)
-        memory_buffer.seek(0)
-        
-        # Обновляем последнее изображение
-        latest_image["data"] = memory_buffer
+        # Читаем весь файл в байты
+        file_bytes = file.read()
+
+        # Сохраняем в переменную
+        latest_image["data"] = file_bytes
         latest_image["filename"] = filename
 
         return jsonify({
@@ -659,21 +656,19 @@ def upload_file_to_memory():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-# Маршрут для получения последнего изображения
+# Получение последнего изображения
 @app.route('/last_image', methods=['GET'])
 def get_last_image():
-    if latest_image["data"] is None:
+    if not latest_image["data"]:
         return jsonify({"error": "No image available"}), 404
 
-    latest_image["data"].seek(0)
+    # Возвращаем через новый BytesIO каждый раз
     return send_file(
-        latest_image["data"],
+        BytesIO(latest_image["data"]),
         mimetype='image/jpeg',
         download_name=latest_image["filename"]
     )
 
-
-
 @app.route('/view_image', methods=['GET'])
 def view_image():
     return render_template('show_image.html')