|
import importlib.util |
|
import os |
|
import subprocess |
|
import sys |
|
from functools import lru_cache |
|
from pathlib import Path |
|
from typing import Iterable |
|
|
|
import gradio as gr |
|
from gradio.themes.base import Base |
|
from gradio.themes.utils import colors, fonts, sizes |
|
|
|
GIT = ( |
|
(Path(os.environ.get("GIT_HOME", "")) / "git").resolve() |
|
if sys.platform == "win32" |
|
else "git" |
|
) |
|
GIT = str(GIT) |
|
|
|
|
|
def is_module_installed(module_name: str) -> bool: |
|
spec = importlib.util.find_spec(module_name) |
|
return spec is not None |
|
|
|
|
|
@lru_cache() |
|
def commit_hash(): |
|
try: |
|
return subprocess.check_output( |
|
[GIT, "log", "-1", "--format='%h %s'"], shell=False, encoding="utf8" |
|
).strip() |
|
except Exception: |
|
return "<none>" |
|
|
|
|
|
def versions_html(): |
|
import torch |
|
|
|
python_version = ".".join([str(x) for x in sys.version_info[0:3]]) |
|
commit = commit_hash() |
|
hash = commit.strip("'").split(" ")[0] |
|
|
|
return f""" |
|
version: <a href="https://github.com/fishaudio/fish-speech/commit/{hash}">{hash}</a> |
|
 •  |
|
python: <span title="{sys.version}">{python_version}</span> |
|
 •  |
|
torch: {getattr(torch, '__long_version__',torch.__version__)} |
|
 •  |
|
gradio: {gr.__version__} |
|
 •  |
|
author: <a href="https://github.com/fishaudio">fishaudio</a> |
|
""" |
|
|
|
|
|
def version_check(commit): |
|
try: |
|
import requests |
|
|
|
commits = requests.get( |
|
"https://api.github.com/repos/fishaudio/fish-speech/branches/main" |
|
).json() |
|
if commit != "<none>" and commits["commit"]["sha"] != commit: |
|
print("--------------------------------------------------------") |
|
print("| You are not up to date with the most recent release. |") |
|
print("| Consider running `git pull` to update. |") |
|
print("--------------------------------------------------------") |
|
elif commits["commit"]["sha"] == commit: |
|
print("You are up to date with the most recent release.") |
|
else: |
|
print("Not a git clone, can't perform version check.") |
|
except Exception as e: |
|
print("version check failed", e) |
|
|
|
|
|
class Seafoam(Base): |
|
def __init__( |
|
self, |
|
*, |
|
primary_hue: colors.Color | str = colors.emerald, |
|
secondary_hue: colors.Color | str = colors.blue, |
|
neutral_hue: colors.Color | str = colors.blue, |
|
spacing_size: sizes.Size | str = sizes.spacing_md, |
|
radius_size: sizes.Size | str = sizes.radius_md, |
|
text_size: sizes.Size | str = sizes.text_lg, |
|
font: fonts.Font | str | Iterable[fonts.Font | str] = ( |
|
fonts.GoogleFont("Quicksand"), |
|
"ui-sans-serif", |
|
"sans-serif", |
|
), |
|
font_mono: fonts.Font | str | Iterable[fonts.Font | str] = ( |
|
fonts.GoogleFont("IBM Plex Mono"), |
|
"ui-monospace", |
|
"monospace", |
|
), |
|
): |
|
super().__init__( |
|
primary_hue=primary_hue, |
|
secondary_hue=secondary_hue, |
|
neutral_hue=neutral_hue, |
|
spacing_size=spacing_size, |
|
radius_size=radius_size, |
|
text_size=text_size, |
|
font=font, |
|
font_mono=font_mono, |
|
) |
|
super().set( |
|
button_primary_background_fill="linear-gradient(90deg, *primary_300, *secondary_400)", |
|
button_primary_background_fill_hover="linear-gradient(90deg, *primary_200, *secondary_300)", |
|
button_primary_text_color="white", |
|
button_primary_background_fill_dark="linear-gradient(90deg, *primary_600, *secondary_800)", |
|
slider_color="*secondary_300", |
|
slider_color_dark="*secondary_600", |
|
block_title_text_weight="600", |
|
block_border_width="3px", |
|
block_shadow="*shadow_drop_lg", |
|
button_shadow="*shadow_drop_lg", |
|
button_small_padding="0px", |
|
button_large_padding="3px", |
|
) |
|
|
