Spaces:
Sleeping
Sleeping
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' # Папка для сохранения изображений | |
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) |