Spaces:
Sleeping
Sleeping
from flask import Flask, request, jsonify, url_for, send_from_directory | |
import os | |
from datetime import datetime | |
import logging | |
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(): | |
if 'image' not in request.files: | |
return jsonify({"error": "No image part in the request"}), 400 | |
file = request.files['image'] | |
if file.filename == '': | |
return jsonify({"error": "No image selected for uploading"}), 400 | |
if file: | |
try: | |
# Генерируем уникальное имя файла для сохранения изображения | |
timestamp = datetime.now().strftime("%Y%m%d%H%M%S") | |
filename = f'image_{timestamp}.png' | |
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename) | |
# Сохраняем изображение | |
file.save(filepath) | |
# Генерируем URL для доступа к изображению | |
image_url = url_for('uploaded_file', filename=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": "Invalid image data"}), 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) |