Spaces:
Sleeping
Sleeping
| 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-запросы | |
| 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 | |
| # Отдача статических файлов (изображений) | |
| 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) |