Update app.py

app.py

@@ -1,15 +1,27 @@
 import gradio as gr
+from PIL import Image
 import requests
 import json
 import os
 
-def generate_minecraft_command(input_data):
+def process_image(image_data):
+    try:
+        image = Image.open(image_data)
+        # Вы можете добавить здесь дополнительные шаги обработки изображения, если это необходимо
+        processed_image = image
+        return processed_image
+    except Exception as e:
+        return None
+
+def generate_minecraft_command(input_data, image_data=None):
     try:
         if isinstance(input_data, str):
             description = input_data
         else:
             description = input_data.read()
 
+        processed_image = process_image(image_data)
+
         headers = {
             'Content-Type': 'application/json',
             'Authorization': f'Bearer {os.getenv("API_KEY")}'
@@ -21,8 +33,12 @@ def generate_minecraft_command(input_data):
             'model': os.getenv("MODEL")
         }
 
+        if processed_image is not None:
+            # Добавим обработанное изображение к запросу, если оно доступно
+            payload['image'] = processed_image
+
         response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
-        response.raise_for_status()  # Raises an HTTPError if the HTTP request returned an unsuccessful status code
+        response.raise_for_status()
 
         data = json.loads(response.text)