Update app.py

app.py

@@ -1,26 +1,38 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, url_for
 import base64
 from io import BytesIO
-import json
+import os
+from datetime import datetime
 
 app = Flask(__name__)
+app.config['UPLOAD_FOLDER'] = 'static/images'  # Папка для сохранения изображений
 
 @app.route('/')
 def process_bytesio():
     data = request.json  # Получаем JSON из запроса
     if data and "BytesIO" in data:
-        # Преобразуем строку в BytesIO
         input_str = data["BytesIO"]
         input_bytes = base64.b64decode(input_str)  # Декодируем из base64
         bytes_io = BytesIO(input_bytes)
 
-        # Конвертируем BytesIO в base64 (мы можем это сделать сразу, т.к. input_bytes уже в байтах)
-        b64_encoded = base64.b64encode(bytes_io.getvalue()).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)
 
-        # Отправляем ответ
-        return jsonify({"base64": b64_encoded})
+        # Сохраняем изображение
+        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 изображения
+        return jsonify({"image_url": image_url})
     else:
         return jsonify({"error": "No data provided or missing 'BytesIO' key"}), 400
 
 if __name__ == '__main__':
+    # Убедитесь, что папка для загрузки существует
+    os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
     app.run(host='0.0.0.0', port=7860)