import gradio as gr
import os

# Функция для отправки запроса к OpenAI GPT-3.5-turbo
def translate_text(input_text, target_language):
    # Здесь должен быть ваш ключ API OpenAI GPT-3.5-turbo
    api_key = os.getenv("API_KEY")

    # Формируем текст запроса
    prompt = f"Переведи пожалуйста правильно, понятно, сохраняя мысль, идеально на {target_language}: \"{input_text}\""

    # Отправляем запрос к OpenAI GPT-3.5-turbo
    # Замените URL на актуальный URL для вашего аккаунта OpenAI
    response = gr.Interface.api.api_post(
        os.getenv("BASE_URL"),
        data={
            "prompt": prompt,
            "max_tokens": 10000,  # Можете настроить максимальное количество токенов
        },
        headers={"Authorization": f"Bearer {api_key}"},
    )

    # Получаем ответ от OpenAI и извлекаем переведенный текст
    translated_text = response.json()["choices"][0]["text"].strip()
    return translated_text

# Список поддерживаемых языков
languages = ["English", "Spanish", "French", "German", "Italian", "Russian"]

# Создаем интерфейс Gradio
iface = gr.Interface(
    fn=translate_text,
    inputs=[
        gr.Textbox("Введите текст для перевода:"),
        gr.Dropdown(languages, label="Выберите язык перевода (на):"),
    ],
    outputs="text",
    live=True,
)

# Запускаем приложение Gradio
iface.launch()