Adding gradio_pages to config.py

This commit is contained in:
ekoc committed 2024-11-15 10:02:16 +01:00
1 parent 37e665c5d8
commit 1e4e18f977
2 files changed
+20 -1

No files matched your search

+11
View File
@@ -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
+9 -1
View File
@@ -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")