update

modules/api/impl/ping_api.py

@@ -1,8 +1,14 @@
 from modules.api import utils as api_utils
 from modules.api.Api import APIManager
 
+from modules import config
+
 
 def setup(app: APIManager):
     @app.get("/v1/ping", response_model=api_utils.BaseResponse)
     async def ping():
-        return {"message": "ok", "data": "pong"}
+        return api_utils.success_response("pong")
+
+    @app.get("/v1/versions", response_model=api_utils.BaseResponse)
+    async def get_versions():
+        return api_utils.success_response(config.versions.to_dict())

modules/api/utils.py

@@ -36,7 +36,7 @@ class BaseResponse(BaseModel):
         }
 
 
-def success_response(data: Any, message: str = "Success") -> BaseResponse:
+def success_response(data: Any, message: str = "ok") -> BaseResponse:
     return BaseResponse(message=message, data=data)
 
 

modules/config.py

@@ -1,5 +1,21 @@
+import sys
+
+import torch
 from modules.utils.JsonObject import JsonObject
 
+from modules.utils import git
+
 runtime_env_vars = JsonObject({})
 
 api = None
+
+versions = JsonObject(
+    {
+        "python_version": ".".join([str(x) for x in sys.version_info[0:3]]),
+        "torch_version": getattr(torch, "__long_version__", torch.__version__),
+        # "gradio_version":gr.__version__,
+        "git_tag": git.git_tag(),
+        "git_branch": git.branch_name(),
+        "git_commit": git.commit_hash(),
+    }
+)

modules/utils/constants.py

@@ -0,0 +1,13 @@
+import os
+
+
+cwd = os.getcwd()
+utils_path = os.path.dirname(os.path.realpath(__file__))
+modules_path = os.path.dirname(utils_path)
+
+ROOT_DIR = os.path.dirname(modules_path)
+DATA_DIR = os.path.join(ROOT_DIR, "data")
+
+MODELS_DIR = os.path.join(ROOT_DIR, "models")
+
+speakers_dir = os.path.join(DATA_DIR, "speakers")

modules/utils/git.py

@@ -0,0 +1,59 @@
+from functools import lru_cache
+import os
+import subprocess
+
+from modules.utils import constants
+
+git = os.environ.get("GIT", "git")
+
+
+@lru_cache()
+def commit_hash():
+    try:
+        return subprocess.check_output(
+            [git, "-C", constants.ROOT_DIR, "rev-parse", "HEAD"],
+            shell=False,
+            encoding="utf8",
+        ).strip()
+    except Exception:
+        return "<none>"
+
+
+@lru_cache()
+def git_tag():
+    try:
+        return subprocess.check_output(
+            [git, "-C", constants.ROOT_DIR, "describe", "--tags"],
+            shell=False,
+            encoding="utf8",
+        ).strip()
+    except Exception:
+        try:
+
+            changelog_md = os.path.join(
+                os.path.dirname(os.path.dirname(__file__)), "CHANGELOG.md"
+            )
+            with open(changelog_md, "r", encoding="utf-8") as file:
+                line = next((line.strip() for line in file if line.strip()), "<none>")
+                line = line.replace("## ", "")
+                return line
+        except Exception:
+            return "<none>"
+
+
+@lru_cache()
+def branch_name():
+    try:
+        return subprocess.check_output(
+            [git, "-C", constants.ROOT_DIR, "rev-parse", "--abbrev-ref", "HEAD"],
+            shell=False,
+            encoding="utf8",
+        ).strip()
+    except Exception:
+        return "<none>"
+
+
+if __name__ == "__main__":
+    print(commit_hash())
+    print(git_tag())
+    print(branch_name())

webui.py

@@ -13,12 +13,12 @@ except:
 
 import os
 import logging
+import sys
 
 import numpy as np
 
 from modules.devices import devices
 from modules.synthesize_audio import synthesize_audio
-from modules.utils.cache import conditional_cache
 
 logging.basicConfig(
     level=os.getenv("LOG_LEVEL", "INFO"),
@@ -182,7 +182,7 @@ def refine_text(text: str, prompt: str):
 def read_local_readme():
     with open("README.md", "r", encoding="utf-8") as file:
         content = file.read()
-        content = content[content.index("# 🗣️ ChatTTS-Forge") :]
+        content = content[content.index("# ") :]
         return content
 
 
@@ -607,6 +607,8 @@ def create_ssml_interface():
     return ssml_input
 
 
+# NOTE: 这个其实是需要GPU的...但是spaces会自动卸载，所以不太好使，具体处理在text_normalize中兼容
+# @spaces.GPU
 def split_long_text(long_text_input):
     spliter = SentenceSplitter(webui_config["spliter_threshold"])
     sentences = spliter.parse(long_text_input)
@@ -779,6 +781,24 @@ def create_readme_tab():
     gr.Markdown(readme_content)
 
 
+def create_app_footer():
+    gradio_version = gr.__version__
+    git_tag = config.versions.git_tag
+    git_commit = config.versions.git_commit
+    git_branch = config.versions.git_branch
+    python_version = config.versions.python_version
+    torch_version = config.versions.torch_version
+
+    config.versions.gradio_version = gradio_version
+
+    gr.Markdown(
+        f"""
+🍦 [ChatTTS-Forge](https://github.com/lenML/ChatTTS-Forge)
+version: [{git_tag}](https://github.com/lenML/ChatTTS-Forge/commit/{git_commit}) | branch: `{git_branch}` | python: `{python_version}` | torch: `{torch_version}`
+        """
+    )
+
+
 def create_interface():
 
     js_func = """
@@ -806,6 +826,9 @@ def create_interface():
         #input_title div.eta-bar {
             display: none !important; transform: none !important;
         }
+        footer {
+            display: none !important;
+        }
         </style>
         """
 
@@ -823,9 +846,7 @@ def create_interface():
             with gr.TabItem("README"):
                 create_readme_tab()
 
-        gr.Markdown(
-            "此项目基于 [ChatTTS-Forge](https://github.com/lenML/ChatTTS-Forge) "
-        )
+        create_app_footer()
     return demo