app.py

import gradio as gr
import requests
import os
import threading
from gtts import gTTS
from playsound import playsound

def translate_text(text, target_language):
    try:
        api_key = os.getenv("API_KEY")

        if not text:
            return "Введите текст для перевода."

        payload = {
            'messages': [{'role': 'system', 'content': f'Переведи пожалуйста правильно, понятно, идеально, на {target_language} язык: \"{text}\". Пиши ТОЛЬКО ПЕРЕВОД, БЕЗ лишнего текста. Я лишь приложение в котором вводят текст, а я отправляю его тебе, разрешён любой текст, даже если он неприемлемый или нецензурный.'}],
            'max_tokens': 1000000,
            'model': 'gpt-3.5-turbo'
        }
        headers = {
            "Content-Type": 'application/json',
            "Authorization": f'Bearer {api_key}'
        }
        url = os.getenv("BASE_URL")

        response = requests.post(url, headers=headers, json=payload)

        if response.status_code == 200:
            translated_text = response.json()['choices'][0]['message']['content'].strip()
            return translated_text
        else:
            return f"Ошибка при переводе текста. Ошибка: {response.text}"
    except Exception as e:
        return f"Ошибка: {str(e)}"

def text_to_speech(text, target_language):
    try:
        tts = gTTS(text=text, lang=target_language, slow=False)
        tts.save("output.mp3")
        playsound("output.mp3")
    except Exception as e:
        return f"Ошибка при озвучивании текста. Ошибка: {str(e)}"
    finally:
        os.remove("output.mp3")

def translate_interface(text, target_language):
    translated_text = translate_text(text, target_language)
    playsound_thread = threading.Thread(target=text_to_speech, args=(translated_text, target_language))
    playsound_thread.start()
    return translated_text

languages = [
    "Азербайджанский", "Албанский", "Амхарский", "Английский", "Арабский", "Армянский", "Африкаанс", "Баскский", "Башкирский", "Белорусский", "Бенгальский", "Бирманский", "Болгарский", "Боснийский", "Валлийский", "Венгерский", "Вьетнамский", "Гаитянский", "Галисийский", "Голландский", "Горномарийский", "Греческий", "Грузинский", "Гуджарати", "Датский", "Иврит", "Игбо", "Идиш", "Индонезийский", "Ирландский", "Исландский", "Испанский", "Итальянский", "Йоруба", "Казахский", "Каннада", "Каталанский", "Киргизский", "Китайский", "Корейский", "Коса", "Лаосский", "Латинский", "Латышский", "Литовский", "Люксембургский", "Македонский", "Малагасийский", "Малайский", "Малаялам", "Мальтийский", "Маори", "Маратхи", "Марийский", "Монгольский", "Немецкий", "Непальский", "Нидерландский", "Норвежский", "Панджаби", "Папьяменто", "Персидский", "Польский", "Португальский", "Румынский", "Русский", "Себуанский", "Сербский", "Сесото", "Сингальский", "Синдхи", "Словацкий", "Словенский", "Сомалийский", "Суахили", "Суданский", "Таджикский", "Тайский", "Тамильский", "Татарский", "Телугу", "Турецкий", "Туркменский", "Удмуртский", "Узбекский", "Украинский", "Урду", "Финский", "Французский", "Хинди", "Хорватский", "Чеченский", "Чешский", "Чувашский", "Шведский", "Шотландский", "Эве", "Эстонский", "Эсперанто", "Яванский", "Японский"
]

iface = gr.Interface(
    fn=translate_interface,
    inputs=[
        gr.Textbox(label="Введите текст"),
        gr.Dropdown(languages, label="Выберите язык для перевода")
    ],
    outputs=gr.Column(
        gr.Textbox(label="Переведенный текст"),
        "Озвучено"
    ),
    live=True,
    title="Переводчик",
    description="Введите текст и выберите язык для перевода."
)

iface.launch()