|
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'Generate Minecraft command for description: {description}'}], |
|
'max_tokens': 500, |
|
'model': 'gpt-3.5-turbo' |
|
} |
|
|
|
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'Не удалось сгенерировать команду для Minecraft. Причина: {data}' |
|
|
|
iface = gr.Interface(fn=generate_minecraft_command, inputs="text", outputs="text", title="Minecraft Command Generator") |
|
iface.launch() |