import gradio as gr
import requests

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

    # URL API GPT-3.5-turbo
    api_url = os.getenv("BASE_URL")

    # Формирование заголовков запроса
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {api_key}"
    }

    # Формирование параметров запроса
    data = {
        "prompt": f"Переведи пожалуйста правильно, понятно, сохраняя мысль, идеально на {target_language}: \"{input_text}\"",
        "max_tokens": 15000
        "model": "gpt-3.5-turbo"
    }

    # Отправка запроса к OpenAI GPT-3.5-turbo
    response = requests.post(api_url, headers=headers, json=data)

    # Обработка ответа и возврат переведенного текста
    if response.status_code == 200:
        return response.json()["choices"][0]["text"]
    else:
        return f"Error {response.status_code}: {response.text}"

# Функция для создания интерфейса Gradio
def translate_interface(input_text, target_language):
    translated_text = translate_text(input_text, target_language)
    return translated_text

# Создание интерфейса Gradio
iface = gr.Interface(
    fn=translate_interface,
    inputs=["text", gr.DropDown(["English", "Spanish", "French", "German", "Italian", "Russian", "Chinese"])],
    outputs="text",
    live=True,
    title="OpenAI GPT-3.5-turbo Translator",
    description="Translate text to the selected language using GPT-3.5-turbo."
)

# Запуск интерфейса Gradio
iface.launch()