Rooni commited on
Commit
df67f71
·
1 Parent(s): 65c5950

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -3,11 +3,12 @@ from PIL import Image
3
  import requests
4
  import json
5
  import os
 
6
 
7
  def process_image(image_data):
8
  try:
9
- image = Image.open(image_data)
10
- # Вы можете добавить здесь дополнительные шаги обработки изображения, если это необходимо
11
  processed_image = image
12
  return processed_image
13
  except Exception as e:
@@ -34,8 +35,11 @@ def generate_minecraft_command(input_data, image_data=None):
34
  }
35
 
36
  if processed_image is not None:
37
- # Добавим обработанное изображение к запросу, если оно доступно
38
- payload['image'] = processed_image
 
 
 
39
 
40
  response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
41
  response.raise_for_status()
 
3
  import requests
4
  import json
5
  import os
6
+ import io
7
 
8
  def process_image(image_data):
9
  try:
10
+ image = Image.open(io.BytesIO(image_data))
11
+ # Дополнительная обработка изображения, если требуется
12
  processed_image = image
13
  return processed_image
14
  except Exception as e:
 
35
  }
36
 
37
  if processed_image is not None:
38
+ # Конвертируем обработанное изображение в base64 и добавляем к запросу
39
+ buffered = io.BytesIO()
40
+ processed_image.save(buffered, format="PNG")
41
+ img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
42
+ payload['image'] = f"data:image/png;base64,{img_str}"
43
 
44
  response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
45
  response.raise_for_status()