Rooni's picture
Update app.py
9b01ba5
import gradio as gr
import openai
# Установите ваш API-ключ OpenAI
openai.api_key = api_key
# Функция для перевода текста с помощью GPT-3
def translate_text(text, source_language, target_language):
# Проверяем, нужно ли перевести с русского на английский язык или наоборот
if source_language == "ru" and target_language == "en":
prompt = "Translate the following Russian text to English: " + text
elif source_language == "en" and target_language == "ru":
prompt = "Translate the following English text to Russian: " + text
else:
return "Unsupported language pair"
# Вызываем GPT-3 API для получения перевода
response = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt,
max_tokens=100,
n=1,
stop=None,
)
# Извлекаем перевод из ответа API
translations = response.choices[0].text.strip().split("\n")
translation = translations[0]
return translation
# Создание интерфейса Gradio для взаимодействия с пользователем
input_text = gr.inputs.Textbox(label="Введите текст")
source_lang = gr.inputs.Dropdown(["ru", "en"], label="Исходный язык")
target_lang = gr.inputs.Dropdown(["ru", "en"], label="Целевой язык")
output_text = gr.outputs.Textbox(label="Перевод")
gr.Interface(fn=translate_text, inputs=[input_text, source_lang, target_lang], outputs=output_text).launch()