File size: 1,066 Bytes
f92a51b
 
6eccccd
ec1e4af
f92a51b
fde6e17
d8abe74
6eccccd
23a6746
d8abe74
6eccccd
 
b2592e7
fde6e17
01868c2
d8abe74
6eccccd
ec1e4af
6eccccd
2ed3994
6eccccd
18f89c2
ed9a233
d74acb3
2ed3994
f443314
10954e8
f443314
10954e8
adc769b
fde6e17
f443314
79300dc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import gradio as gr
import requests
import json
import os

def generate_minecraft_command(description):
    headers = {
        'Content-Type': 'application/json',
        'Authorization': f'Bearer {os.getenv("API_KEY")}'
    }

    payload = {
        'messages': [{'role': 'system', 'content': f'{description}'}],
        'max_tokens': 10000,
        'model': os.getenv("MODEL")
    }

    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=generate_minecraft_command, inputs=[
    gr.Textbox(label="Запрос")
], outputs=gr.Textbox(label="Ответ", markup=True), title="GPT")
iface.launch()