Update app.py

app.py

@@ -1,38 +1,56 @@
-from flask import Flask, request, jsonify, url_for
+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'  # Папка для сохранения изображений
 
+# Настройка папки для загрузки изображений
+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:
-        input_str = data["BytesIO"]
-        input_bytes = base64.b64decode(input_str)  # Декодируем из base64
-        bytes_io = BytesIO(input_bytes)
+        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)
+            # Генерируем уникальное имя файла для сохранения изображения
+            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())
+            # Сохраняем изображение
+            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 для доступа к изображению
+            image_url = url_for('static', filename=os.path.join('images', filename), _external=True)
 
-        # Отправляем ответ с URL изображения
-        return jsonify({"image_url": image_url})
+            # Отправляем ответ с 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)
-    app.run(host='0.0.0.0', port=7860)
+    
+    # Запуск Flask-приложения с включенным режимом отладки
+    app.run(host='0.0.0.0', port=7860, debug=True)