from flask import Flask, request, jsonify, url_for, send_from_directory
import os
from datetime import datetime
import logging
import base64
import traceback

app = Flask(__name__)

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

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

handler = logging.StreamHandler()
handler.setLevel(logging.DEBUG)
app.logger.addHandler(handler)

# Главный маршрут, теперь принимающий POST-запросы
@app.route('/', methods=['POST'])
def process_image():
    data = request.json  # Получаем JSON из запроса
    if data and "image" in data:
        image_data_str = data["image"]
        try:
            # Преобразуем строку с двоичными данными в байты
            image_data_bytes = image_data_str.encode('utf-8').decode('unicode_escape').encode('latin1')

            # Кодируем байты в строку base64
            image_data_base64 = base64.b64encode(image_data_bytes).decode('utf-8')

            # Генерируем уникальное имя файла для сохранения изображения
            timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
            filename = f'image_{timestamp}.png'
            filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)

            # Сохраняем изображение в формате base64
            with open(filepath, 'wb') as f:
                f.write(base64.b64decode(image_data_base64))

            # Генерируем URL для доступа к изображению
            image_url = url_for('uploaded_file', filename=filename, _external=True)

            # Отправляем ответ с URL изображения
            return jsonify({"image_url": image_url})
        except Exception as e:
            # Выводим подробную информацию об ошибке
            error_info = traceback.format_exc()
            app.logger.error('An error occurred: %s', error_info)
            return jsonify({"error": "An error occurred while processing the image", "info": error_info}), 500
    else:
        return jsonify({"error": "No image data provided"}), 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)