Spaces:
Sleeping
Sleeping
File size: 1,630 Bytes
bfd5b9c a584bd9 bfd5b9c a584bd9 bfd5b9c a584bd9 5f155f4 a584bd9 045a9fb bfd5b9c 045a9fb bfd5b9c a584bd9 045a9fb a584bd9 bfd5b9c 084f889 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
from flask import Flask, request, jsonify, url_for
import base64
from io import BytesIO
import os
from datetime import datetime
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = 'static/images' # Папка для сохранения изображений
@app.route('/', methods=['POST'])
def process_bytesio():
data = request.json # Получаем JSON из запроса
if data and "BytesIO" in data:
input_str = data["BytesIO"]
input_bytes = base64.b64decode(input_str) # Декодируем из base64
bytes_io = BytesIO(input_bytes)
# Генерируем уникальное имя файла для сохранения изображения
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
filename = f'image_{timestamp}.png'
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
# Сохраняем изображение
with open(filepath, 'wb') as f:
f.write(bytes_io.getbuffer())
# Генерируем URL для доступа к изображению
image_url = url_for('static', filename=os.path.join('images', filename), _external=True)
# Отправляем ответ с URL изображения
return jsonify({"image_url": image_url})
else:
return jsonify({"error": "No data provided or missing 'BytesIO' key"}), 400
if __name__ == '__main__':
# Убедитесь, что папка для загрузки существует
os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
app.run(host='0.0.0.0', port=7860) |