| 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_small_padding="0px",
|
| button_large_padding="3px",
|
| )
|
|
|
