Update app.py

app.py

@@ -1,77 +1,35 @@
 import gradio as gr
-from PIL import Image
 import requests
 import json
 import os
-import io
-import base64
 
-def process_image(image_data):
+def generate_minecraft_command(description, image):
     try:
-        image = Image.open(io.BytesIO(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()
-
         headers = {
-            'Content-Type': 'application/json',
             'Authorization': f'Bearer {os.getenv("API_KEY")}'
         }
 
-        payload_text = {
-            'messages': [{'role': 'system', 'content': f'{description}'}],
+        files = {'image': ('image.png', image.read(), 'image/png')}
+
+        data = {
+            'messages': [{'role': 'system', 'content': description}],
             'max_tokens': 10000,
             'model': os.getenv("MODEL")
         }
 
-        response_text = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload_text)
-        response_text.raise_for_status()
-        data_text = json.loads(response_text.text)
-
-        processed_image = process_image(image_data)
-
-        if processed_image is not None:
-            # Конвертируем обработанное изображение в base64 и добавляем к запросу изображения
-            buffered = io.BytesIO()
-            processed_image.save(buffered, format="PNG")
-            img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
-
-            payload_image = {
-                'messages': [{'role': 'user', 'content': f'data:image/png;base64,{img_str}'}],
-                'max_tokens': 10000,
-                'model': os.getenv("MODEL")
-            }
-
-            response_image = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload_image)
-            response_image.raise_for_status()
-            data_image = json.loads(response_image.text)
+        response = requests.post(os.getenv("BASE_URL"), headers=headers, data=data, files=files)
+        response.raise_for_status()
 
-            if 'choices' in data_image and len(data_image['choices']) > 0:
-                command = data_image['choices'][0]['message']['content'].strip()
-                return command
-            elif 'error' in data_image:
-                error_message = data_image['error']['message']
-                return f'Ошибка: {error_message}'
-            else:
-                return f'Не удалось сгенерировать команду. {data_image}'
+        data = json.loads(response.text)
 
-        elif 'choices' in data_text and len(data_text['choices']) > 0:
-            command = data_text['choices'][0]['message']['content'].strip()
+        if 'choices' in data and len(data['choices']) > 0:
+            command = data['choices'][0]['message']['content'].strip()
             return command
-        elif 'error' in data_text:
-            error_message = data_text['error']['message']
+        elif 'error' in data:
+            error_message = data['error']['message']
             return f'Ошибка: {error_message}'
         else:
-            return f'Не удалось сгенерировать команду. {data_text}'
+            return f'Не удалось сгенерировать команду. {data}'
 
     except Exception as e:
         return f'Произошла ошибка: {str(e)}'
@@ -80,7 +38,7 @@ iface = gr.Interface(
     fn=generate_minecraft_command,
     inputs=[
         gr.Textbox(label="Запрос"),
-        gr.File(label="Загрузить изображение", type="binary")
+        gr.Image(label="Изображение")
     ],
     outputs=gr.Textbox(label="Ответ"),
     title="GPT"