Adding gradio_pages to config.py
This commit is contained in:
1 parent
37e665c5d8
commit
1e4e18f977
2 files changed
+20
-1
No files matched your search
@@ -78,6 +78,16 @@ class AlltalkConfigDebug:
|
|||||||
debug_tts_variables = False
|
debug_tts_variables = False
|
||||||
debug_rvc = 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
|
@dataclass
|
||||||
class AlltalkAvailableEngine:
|
class AlltalkAvailableEngine:
|
||||||
name = ""
|
name = ""
|
||||||
@@ -214,6 +224,7 @@ class AlltalkConfig(AbstractConfig):
|
|||||||
self.tgwui = AlltalkConfigTgwUi()
|
self.tgwui = AlltalkConfigTgwUi()
|
||||||
self.api_def = AlltalkConfigApiDef()
|
self.api_def = AlltalkConfigApiDef()
|
||||||
self.debugging = AlltalkConfigDebug()
|
self.debugging = AlltalkConfigDebug()
|
||||||
|
self.gradio_pages = AlltalkConfigGradioPages()
|
||||||
self._load_config()
|
self._load_config()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
|
|||||||
+9
-1
@@ -107,6 +107,14 @@ class TestAlltalkConfig(unittest.TestCase):
|
|||||||
self.assertFalse(self.config.debugging.debug_tts_variables)
|
self.assertFalse(self.config.debugging.debug_tts_variables)
|
||||||
self.assertFalse(self.config.debugging.debug_rvc)
|
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):
|
def test_save_config(self):
|
||||||
with tempfile.NamedTemporaryFile(suffix=".json") as tmp:
|
with tempfile.NamedTemporaryFile(suffix=".json") as tmp:
|
||||||
self.config.branding = "foo"
|
self.config.branding = "foo"
|
||||||
@@ -117,7 +125,7 @@ class TestAlltalkConfig(unittest.TestCase):
|
|||||||
def test_tts_engines(self):
|
def test_tts_engines(self):
|
||||||
self.assertEqual(self.ttsEnginesConfig.engine_loaded, "piper")
|
self.assertEqual(self.ttsEnginesConfig.engine_loaded, "piper")
|
||||||
self.assertEqual(self.ttsEnginesConfig.selected_model, "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):
|
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")
|
expected_config_path = os.path.join(Path(__file__).parent.parent.resolve(), "system", "tts_engines", "tts_engines.json")
|
||||||
|
|||||||
Reference in new issue
Block a user