translator / app.py
Rooni's picture
Update app.py
0b17da2
raw
history blame
1.73 kB
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,
capture_session=True,
)
# Запускаем приложение Gradio
iface.launch()