Изменение основной программы для теста нескольких моделей

run.py

@@ -1,27 +1,61 @@
-from transformers import pipeline
-from datasets import load_dataset
-import torch
 import streamlit as st
 
+from mulyavin_aa import langdetector
+from mulyavin_aa import translator
+
+LANG_DETECTOR = "LANG_DETECTOR"
+TRANSLATOR = "TRANSLATOR"
+
+
 @st.cache_resource
-def load_speech_model():
-    synthesiser = pipeline("text-to-speech", "microsoft/speecht5_tts")
+def load_models() -> dict:
+    """
+    Получение справочника моделей
+    :return: Справочник моделей
+    """
+    models = dict()
+    models[LANG_DETECTOR] = langdetector.load_text_detection_model()
+    models[TRANSLATOR] = translator.load_text_translator_model()
+
+    return models
+
+
+def main_app():
+    """
+    Основная программа
+    """
+
+    models = load_models()
+
+    st.title = 'Домашнее задание'
+
+    # Оформление заголовка
+    st.header('Домашнее задание', divider='gray')
 
-    embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
-    speaker_embedding = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)
+    input_text = st.text_area(
+        'Введите текст на русском или английском языке и нажмите кнопку генератора:')
 
-    return synthesiser, speaker_embedding
+    if st.button('Генерировать!!!'):
+        # Определение языка
+        text_lang = langdetector.lang_detect(input_text, models[LANG_DETECTOR])
+        if text_lang not in ['ru', 'en']:
+            st.error('Язык текста не может быть определен')
+            return
 
-def speech_elements():
-    synthesiser, speaker_embedding = load_speech_model()
+        # Перевод языка если не en
+        if text_lang in ['ru']:
+            input_text = translator.translate_to_en(input_text, models[TRANSLATOR])
 
-    text = st.text_area('Enter English text here')
-    st.write(f'You wrote {len(text)} characters.')
+        tab1, tab2, tab3 = st.tabs(['Озвученный текст', 'Таб 2', 'Таб 3'])
+        with tab1:
+            st.header("Озвученный текст на английском языке")
+            # st.audio()
 
-    if st.button('Speech'):
-        speech = synthesiser(text, forward_params={"speaker_embeddings": speaker_embedding})
+        with tab2:
+            st.header("Таб 2")
 
-        st.audio(speech['audio'], sample_rate=speech['sampling_rate'])
+        with tab3:
+            st.header("Таб 3")
 
 
-speech_elements()
+main_app()