File size: 2,436 Bytes
157a73d
a584bd9
 
bfd5b9c
 
a584bd9
 
 
157a73d
 
 
 
 
 
 
 
5f155f4
a584bd9
045a9fb
 
157a73d
 
 
 
045a9fb
157a73d
 
 
 
045a9fb
157a73d
 
 
bfd5b9c
157a73d
 
bfd5b9c
157a73d
 
 
 
 
a584bd9
045a9fb
a584bd9
157a73d
 
 
 
 
a584bd9
bfd5b9c
 
157a73d
 
 
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
from flask import Flask, request, jsonify, url_for, send_from_directory
import base64
from io import BytesIO
import os
from datetime import datetime

app = Flask(__name__)

# Настройка папки для загрузки изображений
app.config['UPLOAD_FOLDER'] = 'static/images'

# Настройка логирования
import logging
logging.basicConfig(level=logging.DEBUG)

# Главный маршрут, теперь принимающий POST-запросы
@app.route('/', methods=['POST'])
def process_bytesio():
    data = request.json  # Получаем JSON из запроса
    if data and "BytesIO" in data:
        try:
            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})
        except Exception as e:
            app.logger.error(f'An error occurred: {e}')
            return jsonify({"error": "An error occurred while processing the image"}), 500
    else:
        return jsonify({"error": "No data provided or missing 'BytesIO' key"}), 400

# Отдача статических файлов (изображений)
@app.route('/static/images/<filename>')
def uploaded_file(filename):
    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)

if __name__ == '__main__':
    # Убедитесь, что папка для загрузки существует
    os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
    
    # Запуск Flask-приложения с включенным режимом отладки
    app.run(host='0.0.0.0', port=7860, debug=True)