Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +1,7 @@
|
|
1 |
from flask import Flask, request, jsonify, url_for, send_from_directory
|
2 |
-
import base64
|
3 |
-
from io import BytesIO
|
4 |
import os
|
5 |
from datetime import datetime
|
6 |
import logging
|
7 |
-
import re
|
8 |
|
9 |
app = Flask(__name__)
|
10 |
|
@@ -18,32 +15,26 @@ handler = logging.StreamHandler()
|
|
18 |
handler.setLevel(logging.DEBUG)
|
19 |
app.logger.addHandler(handler)
|
20 |
|
21 |
-
# Валидация строки base64
|
22 |
-
def is_base64_encoded(s):
|
23 |
-
pattern = r'^([A-Za-z0-9+/]{4})*([A-Za-z0-9+/]{2}==|[A-Za-z0-9+/]{3}=)?$'
|
24 |
-
return re.match(pattern, s) is not None
|
25 |
-
|
26 |
# Главный маршрут, теперь принимающий POST-запросы
|
27 |
@app.route('/', methods=['POST'])
|
28 |
-
def
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
return jsonify({"error": "Invalid base64 data"}), 400
|
34 |
-
|
35 |
-
try:
|
36 |
-
input_bytes = base64.b64decode(input_str) # Декодируем из base64
|
37 |
-
bytes_io = BytesIO(input_bytes)
|
38 |
|
|
|
|
|
|
|
|
|
|
|
39 |
# Генерируем уникальное имя файла для сохранения изображения
|
40 |
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
|
41 |
filename = f'image_{timestamp}.png'
|
42 |
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
43 |
|
44 |
# Сохраняем изображение
|
45 |
-
|
46 |
-
f.write(bytes_io.getbuffer())
|
47 |
|
48 |
# Генерируем URL для доступа к изображению
|
49 |
image_url = url_for('uploaded_file', filename=filename, _external=True)
|
@@ -54,7 +45,7 @@ def process_bytesio():
|
|
54 |
app.logger.error(f'An error occurred: {e}')
|
55 |
return jsonify({"error": "An error occurred while processing the image"}), 500
|
56 |
else:
|
57 |
-
return jsonify({"error": "
|
58 |
|
59 |
# Отдача статических файлов (изображений)
|
60 |
@app.route('/static/images/<filename>')
|
|
|
1 |
from flask import Flask, request, jsonify, url_for, send_from_directory
|
|
|
|
|
2 |
import os
|
3 |
from datetime import datetime
|
4 |
import logging
|
|
|
5 |
|
6 |
app = Flask(__name__)
|
7 |
|
|
|
15 |
handler.setLevel(logging.DEBUG)
|
16 |
app.logger.addHandler(handler)
|
17 |
|
|
|
|
|
|
|
|
|
|
|
18 |
# Главный маршрут, теперь принимающий POST-запросы
|
19 |
@app.route('/', methods=['POST'])
|
20 |
+
def process_image():
|
21 |
+
if 'image' not in request.files:
|
22 |
+
return jsonify({"error": "No image part in the request"}), 400
|
23 |
+
|
24 |
+
file = request.files['image']
|
|
|
|
|
|
|
|
|
|
|
25 |
|
26 |
+
if file.filename == '':
|
27 |
+
return jsonify({"error": "No image selected for uploading"}), 400
|
28 |
+
|
29 |
+
if file:
|
30 |
+
try:
|
31 |
# Генерируем уникальное имя файла для сохранения изображения
|
32 |
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
|
33 |
filename = f'image_{timestamp}.png'
|
34 |
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
35 |
|
36 |
# Сохраняем изображение
|
37 |
+
file.save(filepath)
|
|
|
38 |
|
39 |
# Генерируем URL для доступа к изображению
|
40 |
image_url = url_for('uploaded_file', filename=filename, _external=True)
|
|
|
45 |
app.logger.error(f'An error occurred: {e}')
|
46 |
return jsonify({"error": "An error occurred while processing the image"}), 500
|
47 |
else:
|
48 |
+
return jsonify({"error": "Invalid image data"}), 400
|
49 |
|
50 |
# Отдача статических файлов (изображений)
|
51 |
@app.route('/static/images/<filename>')
|