Update app.py

app.py

@@ -3,6 +3,8 @@ import base64
 from io import BytesIO
 import os
 from datetime import datetime
+import logging
+import re
 
 app = Flask(__name__)
 
@@ -10,16 +12,27 @@ app = Flask(__name__)
 app.config['UPLOAD_FOLDER'] = 'static/images'
 
 # Настройка логирования
-import logging
 logging.basicConfig(level=logging.DEBUG)
 
+handler = logging.StreamHandler()
+handler.setLevel(logging.DEBUG)
+app.logger.addHandler(handler)
+
+# Валидация строки base64
+def is_base64_encoded(s):
+    pattern = r'^([A-Za-z0-9+/]{4})*([A-Za-z0-9+/]{2}==|[A-Za-z0-9+/]{3}=)?$'
+    return re.match(pattern, s) is not None
+
 # Главный маршрут, теперь принимающий POST-запросы
 @app.route('/', methods=['POST'])
 def process_bytesio():
     data = request.json  # Получаем JSON из запроса
     if data and "BytesIO" in data:
+        input_str = data["BytesIO"]
+        if not is_base64_encoded(input_str):
+            return jsonify({"error": "Invalid base64 data"}), 400
+        
         try:
-            input_str = data["BytesIO"]
             input_bytes = base64.b64decode(input_str)  # Декодируем из base64
             bytes_io = BytesIO(input_bytes)
 
@@ -33,7 +46,7 @@ def process_bytesio():
                 f.write(bytes_io.getbuffer())
 
             # Генерируем URL для доступа к изображению
-            image_url = url_for('static', filename=os.path.join('images', filename), _external=True)
+            image_url = url_for('uploaded_file', filename=filename, _external=True)
 
             # Отправляем ответ с URL изображения
             return jsonify({"image_url": image_url})