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) |