Create download_models.py

download_models.py

@@ -0,0 +1,40 @@
+import os
+import urllib.request
+import zipfile
+import tarfile
+
+def download_file(url, dest):
+    if os.path.exists(dest):
+        print(f"File already exists: {dest}")
+        return
+    print(f"Downloading {url} to {dest}...")
+    urllib.request.urlretrieve(url, dest)
+    print("Download complete.")
+
+def setup_models():
+    # Vosk Model
+    vosk_dir = "models/vosk-model-en-us-0.22-lgraph"
+    if not os.path.exists(vosk_dir):
+        os.makedirs("models", exist_ok=True)
+        zip_path = "models/vosk-model.zip"
+        download_file("https://alphacephei.com/vosk/models/vosk-model-en-us-0.22-lgraph.zip", zip_path)
+        print("Extracting Vosk model...")
+        with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+            zip_ref.extractall("models")
+        os.remove(zip_path)
+        print("Vosk model setup complete.")
+    else:
+        print("Vosk model already exists.")
+
+    # Piper Model
+    piper_model_dir = "models/piper"
+    os.makedirs(piper_model_dir, exist_ok=True)
+    
+    piper_onnx = os.path.join(piper_model_dir, "en_US-lessac-medium.onnx")
+    piper_json = os.path.join(piper_model_dir, "en_US-lessac-medium.onnx.json")
+    
+    download_file("https://huggingface.co/rhasspy/piper-voices/resolve/main/en/en_US/lessac/medium/en_US-lessac-medium.onnx", piper_onnx)
+    download_file("https://huggingface.co/rhasspy/piper-voices/resolve/main/en/en_US/lessac/medium/en_US-lessac-medium.onnx.json", piper_json)
+
+if __name__ == "__main__":
+    setup_models()