sort apks

generate-tts.py

@@ -6,6 +6,36 @@ from typing import List
 
 BASE_URL = "https://huggingface.co/csukuangfj/sherpa-onnx-apk/resolve/main/"
 
+from dataclasses import dataclass
+@dataclass
+class APK:
+    major: int
+    minor: int
+    patch: int
+    arch: str
+    lang: str
+    src: str # piper, coqui
+    def __init__(self, s):
+        s = str(s)
+        split = s.split('-')
+        self.major, self.minor, self.patch = list(map(int, split[2].split('.')))
+        self.arch = split[3]
+        self.lang = split[4]
+        self.src = split[7]
+        if 'arm' in s:
+            self.arch += '-' + split[4]
+            self.lang = split[5]
+            self.src = split[8]
+
+        if 'armeabi' in self.arch:
+            self.arch = 'y' + self.arch
+
+        if 'arm64' in self.arch:
+            self.arch = 'z' + self.arch
+
+def sort_by_apk(x):
+    x = APK(x)
+    return (x.major, x.minor, x.patch, x.arch, -ord(x.src[0]),-ord(x.lang[0]))
 
 def generate_url(files: List[str]) -> List[str]:
     ans = []
@@ -16,7 +46,7 @@ def generate_url(files: List[str]) -> List[str]:
 
 
 def get_all_files(d: str, suffix: str) -> List[str]:
-    ans = sorted(Path(d).glob(suffix), reverse=True)
+    ans = sorted(Path(d).glob(suffix), key=sort_by_apk, reverse=True)
     return list(map(lambda x: BASE_URL + str(x), ans))
 
 
@@ -48,6 +78,11 @@ in <a href="https://github.com/rhasspy/piper">piper</a> is
 https://huggingface.co/rhasspy/piper-voices/blob/main/en/en_US/lessac/medium/en_US-lessac-medium.onnx
 </a><br/><br/>
 
+<span style="color:red;">Note:</span> Models from
+<a href="https://github.com/coqui-ai/TTS">coqui-ai/TTS</a> have their names prefixed
+with <strong>coqui-</strong>.
+<br/><br/>
+
 You can find many more models that have not been converted to <strong>sherpa-onnx</strong>
 at
 <a href="https://huggingface.co/rhasspy/piper-voices">https://huggingface.co/rhasspy/piper-voices</a>