Update app.py

app.py

@@ -3,44 +3,42 @@ import requests
 import json
 import os
 
-def gpt3_turbo(image=None, text=None):
+def generate_minecraft_command(input_data):
+    if isinstance(input_data, str):
+        description = input_data
+    else:
+        description = input_data.read()
 
-    api_key = os.getenv("API_KEY")
-    endpoint = os.getenv("BASE_URL")
-    
-    # Формируем данные для отправки запроса
-    data = {}
-    if image is not None:
-        with open(image.name, "rb") as img_file:
-            data['image'] = img_file.read()
-    if text:
-        data['text'] = text
-
-    # Отправляем запрос на API GPT-3.5-turbo
     headers = {
         'Content-Type': 'application/json',
-        'Authorization': f'Bearer {api_key}'
+        'Authorization': f'Bearer {os.getenv("API_KEY")}'
+    }
+
+    payload = {
+        'messages': [{'role': 'system', 'content': f'{description}'}],
+        'max_tokens': 10000,
+        'model': os.getenv("MODEL")
     }
-    
-    try:
-        response = requests.post(endpoint, headers=headers, json={"prompt": data, "max_tokens": 100, "model": 'gpt-3.5-turbo'})
-        response.raise_for_status()  # Проверка статуса ответа
-        result = response.json()
-        return result['choices'][0]['text']
-    except requests.RequestException as e:
-        return f"Ошибка при получении ответа от модели: {e}"
 
-# Создаем интерфейс Gradio
+    response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
+    data = json.loads(response.text)
+
+    if 'choices' in data and len(data['choices']) > 0:
+        command = data['choices'][0]['message']['content'].strip()
+        return command
+    elif 'error' in data:
+        error_message = data['error']['message']
+        return f'Ошибка: {error_message}'
+    else:
+        return f'Не удалось сгенерировать команду. {data}'
+
 iface = gr.Interface(
-    fn=gpt3_turbo,
+    fn=generate_minecraft_command,
     inputs=[
-        gr.Image(type="pil", label="Выберите изображение (необязательно)"),
-        "text"
+        gr.Textbox(label="Запрос"),
+        gr.File(label="Загрузить изображение", type="file")
     ],
-    outputs="text",
-    title="GPT",
-    description="Помощь ученикам, решение заданий"
+    outputs=gr.Textbox(label="Ответ"),
+    title="GPT"
 )
-
-# Запускаем интерфейс
 iface.launch()