Rooni commited on
Commit
045a9fb
·
verified ·
1 Parent(s): 084f889

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from flask import Flask, request, jsonify
2
  import base64
3
  from io import BytesIO
 
4
 
5
  app = Flask(__name__)
6
 
@@ -10,17 +11,20 @@ def index():
10
 
11
  @app.route('/process', methods=['POST'])
12
  def process_bytesio():
13
- if request.data:
14
- # Преобразуем полученные байты в BytesIO
15
- bytes_io = BytesIO(request.data)
16
-
17
- # Конвертируем BytesIO в base64
 
 
 
18
  b64_encoded = base64.b64encode(bytes_io.getvalue()).decode('utf-8')
19
-
20
  # Отправляем ответ
21
  return jsonify({"base64": b64_encoded})
22
  else:
23
- return jsonify({"error": "No data provided"}), 400
24
 
25
  if __name__ == '__main__':
26
  app.run(host='0.0.0.0', port=7860)
 
1
  from flask import Flask, request, jsonify
2
  import base64
3
  from io import BytesIO
4
+ import json
5
 
6
  app = Flask(__name__)
7
 
 
11
 
12
  @app.route('/process', methods=['POST'])
13
  def process_bytesio():
14
+ data = request.json # Получаем JSON из запроса
15
+ if data and "BytesIO" in data:
16
+ # Преобразуем строку в BytesIO
17
+ input_str = data["BytesIO"]
18
+ input_bytes = base64.b64decode(input_str) # Декодируем из base64
19
+ bytes_io = BytesIO(input_bytes)
20
+
21
+ # Конвертируем BytesIO в base64 (мы можем это сделать сразу, т.к. input_bytes уже в байтах)
22
  b64_encoded = base64.b64encode(bytes_io.getvalue()).decode('utf-8')
23
+
24
  # Отправляем ответ
25
  return jsonify({"base64": b64_encoded})
26
  else:
27
+ return jsonify({"error": "No data provided or missing 'BytesIO' key"}), 400
28
 
29
  if __name__ == '__main__':
30
  app.run(host='0.0.0.0', port=7860)