* Fix install/uninstall in extension manager (#481)

* Add Kokoro TTS extension

Author: rsxdalv

README.md

@@ -74,6 +74,10 @@
 
 ## Changelog
 
+Apr 20:
+* Fix install/uninstall in extension manager
+* Add Kokoro TTS extension
+
 Apr 18:
 * Fix extension manager startup
 * Convert most models to extensions, install the classic ones by default

extensions.json

@@ -433,6 +433,21 @@
             "extension_website": "https://github.com/rsxdalv/extension_mars5",
             "extension_platform_version": "0.0.1"
         },
+        {
+            "package_name": "extension_kokoro",
+            "name": "Kokoro",
+            "version": "0.0.1",
+            "requirements": "git+https://github.com/rsxdalv/extension_kokoro@main",
+            "description": "Kokoro: A small, fast, and high-quality TTS model",
+            "extension_type": "interface",
+            "extension_class": "text-to-speech",
+            "author": "hexgrad",
+            "extension_author": "rsxdalv",
+            "license": "MIT",
+            "website": "https://huggingface.co/hexgrad/Kokoro-82M",
+            "extension_website": "https://github.com/rsxdalv/extension_kokoro",
+            "extension_platform_version": "0.0.1"
+        },
         {
             "package_name": "extensions.builtin.extension_model_downloader",
             "name": "Model Downloader",

tools/extension_manager.py

@@ -7,7 +7,7 @@
 import importlib
 
 # Add the project root directory to the Python path
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
 from tts_webui.css.css import full_css
 
@@ -321,23 +321,27 @@ def display_extension_ui(x, is_installed, is_disabled):
                 # Check if this is a built-in extension
                 is_builtin = package_name.startswith("extensions.builtin")
 
+                def install_extension():
+                    yield from pip_install_wrapper(x["requirements"], package_name)()
+
                 if is_installed:
                     if not is_builtin:
                         # Update button (only for non-builtin extensions)
                         update_btn = gr.Button("Update", variant="primary")
                         update_btn.click(
-                            fn=lambda pkg=package_name, req=x[
-                                "requirements"
-                            ]: pip_install_wrapper(req, pkg)(),
+                            fn=install_extension,
                             outputs=[gr.HTML()],
                         )
 
+                        def uninstall_extension():
+                            yield from pip_uninstall_wrapper(
+                                package_name, package_name
+                            )()
+
                         # Uninstall button (only for non-builtin extensions)
                         uninstall_btn = gr.Button("Uninstall", variant="stop")
                         uninstall_btn.click(
-                            fn=lambda pkg=package_name: pip_uninstall_wrapper(
-                                pkg, pkg
-                            )(),
+                            fn=uninstall_extension,
                             outputs=[gr.HTML()],
                         )
 
@@ -361,6 +365,12 @@ def toggle_extension(package_name=package_name):
                     toggle_btn.click(
                         fn=toggle_extension,
                         outputs=[gr.HTML()],
+                    ).then(
+                        fn=lambda button_text: gr.Button(
+                            "Enable" if button_text == "Disable" else "Disable"
+                        ),
+                        inputs=[toggle_btn],
+                        outputs=[toggle_btn],
                     )
 
                     if is_builtin:
@@ -371,9 +381,7 @@ def toggle_extension(package_name=package_name):
                     # Install button
                     install_btn = gr.Button("Install", variant="primary")
                     install_btn.click(
-                        fn=lambda pkg=package_name, req=x[
-                            "requirements"
-                        ]: pip_install_wrapper(req, pkg)(),
+                        fn=install_extension,
                         outputs=[gr.HTML()],
                     )