diff --git a/config.py b/config.py index d625895..90d1a8b 100644 --- a/config.py +++ b/config.py @@ -78,6 +78,16 @@ class AlltalkConfigDebug: debug_tts_variables = False debug_rvc = False +@dataclass +class AlltalkConfigGradioPages: + Generate_Help_page = False + Voice2RVC_page = False + TTS_Generator_page = False + TTS_Engines_Settings_page = False + alltalk_documentation_page = False + api_documentation_page = False + + @dataclass class AlltalkAvailableEngine: name = "" @@ -214,6 +224,7 @@ class AlltalkConfig(AbstractConfig): self.tgwui = AlltalkConfigTgwUi() self.api_def = AlltalkConfigApiDef() self.debugging = AlltalkConfigDebug() + self.gradio_pages = AlltalkConfigGradioPages() self._load_config() @staticmethod diff --git a/test/test_config.py b/test/test_config.py index df1f4dd..0bc9c8a 100644 --- a/test/test_config.py +++ b/test/test_config.py @@ -107,6 +107,14 @@ class TestAlltalkConfig(unittest.TestCase): self.assertFalse(self.config.debugging.debug_tts_variables) self.assertFalse(self.config.debugging.debug_rvc) + def test_gradio_pages(self): + self.assertTrue(self.config.gradio_pages.Generate_Help_page) + self.assertTrue(self.config.gradio_pages.Voice2RVC_page) + self.assertTrue(self.config.gradio_pages.TTS_Generator_page) + self.assertTrue(self.config.gradio_pages.TTS_Engines_Settings_page) + self.assertTrue(self.config.gradio_pages.alltalk_documentation_page) + self.assertTrue(self.config.gradio_pages.api_documentation_page) + def test_save_config(self): with tempfile.NamedTemporaryFile(suffix=".json") as tmp: self.config.branding = "foo" @@ -117,7 +125,7 @@ class TestAlltalkConfig(unittest.TestCase): def test_tts_engines(self): self.assertEqual(self.ttsEnginesConfig.engine_loaded, "piper") self.assertEqual(self.ttsEnginesConfig.selected_model, "piper") - self.assertListEqual(self.ttsEnginesConfig.get_engine_names_available(), ["parler", "piper", "vits", "xtts"]) + self.assertListEqual(self.ttsEnginesConfig.get_engine_names_available(), ["parler", "piper", "vits", "xtts", "f5tts"]) def test_tts_engines_default_config_path(self): expected_config_path = os.path.join(Path(__file__).parent.parent.resolve(), "system", "tts_engines", "tts_engines.json")