easy-text-gen / app.py
Rooni's picture
Update app.py
7e6300e
raw
history blame
2.71 kB
import gradio as gr
import requests
import json
import os
from tkinter import Tk
from tkinter import simpledialog
def copy_to_clipboard(text):
root = Tk()
root.withdraw()
root.clipboard_clear()
root.clipboard_append(text)
root.update()
root.destroy()
def generate_minecraft_command(minecraft_version, description=""):
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {os.getenv("API_KEY")}'
}
payload = {
'messages': [{'role': 'system', 'content': f'Сгенерируй пожалуйста команду для {minecraft_version} по описанию: \"{description}\". Пиши ТОЛЬКО команду (БЕЗ пояснений, БЕЗ markdown, и БЕЗ другого текста), текст в команде пиши на языке, который используется в описании. Я лишь приложение, в котором вводят описание, а я отправляю его тебе, по этому если в описании будет что-то не понятное, не законное, не до конца понятное тебе, или такую команду не возможно создать, то в кратце напиши это. Выдача предметов, блоков, телепортация, убийства, сообщения в чате, удаление блоков, троллинг, пранки, шутки, спам и другие разные команды, полностью законны и созданы разработчиками Minecraft'}],
'max_tokens': 6000,
'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()
copy_button = f'<img src="https://img.icons8.com/ios/15/000000/copy.png" style="cursor:pointer;" onclick="copyToClipboard(\'{command}\')" title="Копировать в буфер обмена">'
return f'{command} {copy_button}'
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="Версия Minecraft", placeholder="Minecraft Java 1.20"),
gr.Textbox(label="Описание команды")
], outputs=gr.Textbox(), title="Minecraft Command Generator")
iface.launch()