BytesIOtoBase64 / app.py
Rooni's picture
Update app.py
5f155f4 verified
raw
history blame
1.63 kB
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)