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()