Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@ from flask import Flask, request, jsonify, url_for, send_from_directory
|
|
| 2 |
import os
|
| 3 |
from datetime import datetime
|
| 4 |
import logging
|
|
|
|
| 5 |
|
| 6 |
app = Flask(__name__)
|
| 7 |
|
|
@@ -18,15 +19,13 @@ app.logger.addHandler(handler)
|
|
| 18 |
# Главный маршрут, теперь принимающий POST-запросы
|
| 19 |
@app.route('/', methods=['POST'])
|
| 20 |
def process_image():
|
| 21 |
-
|
| 22 |
-
|
|
|
|
| 23 |
|
| 24 |
-
|
|
|
|
| 25 |
|
| 26 |
-
if file.filename == '':
|
| 27 |
-
return jsonify({"error": "No image selected for uploading"}), 400
|
| 28 |
-
|
| 29 |
-
if file:
|
| 30 |
try:
|
| 31 |
# Генерируем уникальное имя файла для сохранения изображения
|
| 32 |
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
|
|
@@ -34,7 +33,8 @@ def process_image():
|
|
| 34 |
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
| 35 |
|
| 36 |
# Сохраняем изображение
|
| 37 |
-
|
|
|
|
| 38 |
|
| 39 |
# Генерируем URL для доступа к изображению
|
| 40 |
image_url = url_for('uploaded_file', filename=filename, _external=True)
|
|
@@ -45,7 +45,7 @@ def process_image():
|
|
| 45 |
app.logger.error(f'An error occurred: {e}')
|
| 46 |
return jsonify({"error": "An error occurred while processing the image"}), 500
|
| 47 |
else:
|
| 48 |
-
return jsonify({"error": "
|
| 49 |
|
| 50 |
# Отдача статических файлов (изображений)
|
| 51 |
@app.route('/static/images/<filename>')
|
|
|
|
| 2 |
import os
|
| 3 |
from datetime import datetime
|
| 4 |
import logging
|
| 5 |
+
from io import BytesIO
|
| 6 |
|
| 7 |
app = Flask(__name__)
|
| 8 |
|
|
|
|
| 19 |
# Главный маршрут, теперь принимающий POST-запросы
|
| 20 |
@app.route('/', methods=['POST'])
|
| 21 |
def process_image():
|
| 22 |
+
data = request.json # Получаем JSON из запроса
|
| 23 |
+
if data and "image" in data:
|
| 24 |
+
image_data = data["image"]
|
| 25 |
|
| 26 |
+
# Преобразуем строку с двоичными данными в объект BytesIO
|
| 27 |
+
image_bytes_io = BytesIO(image_data.encode('utf-8'))
|
| 28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
try:
|
| 30 |
# Генерируем уникальное имя файла для сохранения изображения
|
| 31 |
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
|
|
|
|
| 33 |
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
| 34 |
|
| 35 |
# Сохраняем изображение
|
| 36 |
+
with open(filepath, 'wb') as f:
|
| 37 |
+
f.write(image_bytes_io.read())
|
| 38 |
|
| 39 |
# Генерируем URL для доступа к изображению
|
| 40 |
image_url = url_for('uploaded_file', filename=filename, _external=True)
|
|
|
|
| 45 |
app.logger.error(f'An error occurred: {e}')
|
| 46 |
return jsonify({"error": "An error occurred while processing the image"}), 500
|
| 47 |
else:
|
| 48 |
+
return jsonify({"error": "No image data provided"}), 400
|
| 49 |
|
| 50 |
# Отдача статических файлов (изображений)
|
| 51 |
@app.route('/static/images/<filename>')
|