v3 download model added

app/main.py

@@ -5,6 +5,11 @@ from app.core.errors import error_handler
 from fastapi.middleware.cors import CORSMiddleware
 
 from app.utils import print_routes
+from app.utils.checks import run_checks
+
+if not run_checks():
+    raise Exception("Failed to pass all checks")
+
 
 app = FastAPI(
     title=settings.PROJECT_NAME, openapi_url=f"{settings.API_V1_STR}/openapi.json"

app/utils/checks.py

@@ -0,0 +1,39 @@
+import os
+from app.utils.constant import model_names, model_urls
+from app.utils.utils import download_file
+
+
+def run_checks():
+    try:
+        if not check_models_exist():
+            return False
+        return True
+    except Exception as e:
+        print("Error in run_checks: {}".format(str(e)))
+        return False
+
+
+def check_models_exist():
+    try:
+        for key, value in model_names.items():
+            if os.path.exists(os.path.join(os.getcwd(), "models", value)):
+                print("Model {} exists".format(key))
+            else:
+                print("Model {} does not exist".format(key))
+                download_model(key)
+        return True
+    except Exception as e:
+        print("Error in check_models_exist: {}".format(str(e)))
+        return False
+
+
+def download_model(model_key: str):
+    try:
+        print("Downloading model {} from {}".format(model_key, model_urls[model_key]))
+        download_file(
+            model_urls[model_key],
+            os.path.join(os.getcwd(), "models", model_names[model_key]),
+        )
+        print("Downloaded model {} from {}".format(model_key, model_urls[model_key]))
+    except Exception as e:
+        print("Error in download_models: {}".format(str(e)))

app/utils/constant.py

@@ -3,3 +3,9 @@ model_names = {
     "tiny.en.q5": "ggml-model-whisper-tiny.en-q5_1.bin",
     "base.en.q5": "ggml-model-whisper-base.en-q5_1.bin",
 }
+
+model_urls = {
+    "tiny.en": "https://firebasestorage.googleapis.com/v0/b/model-innovatorved.appspot.com/o/ggml-model-whisper-base.en-q5_1.bin?alt=media",
+    "tiny.en.q5": "https://firebasestorage.googleapis.com/v0/b/model-innovatorved.appspot.com/o/ggml-model-whisper-tiny.en-q5_1.bin?alt=media",
+    "base.en.q5": "https://firebasestorage.googleapis.com/v0/b/model-innovatorved.appspot.com/o/ggml-model-whisper-base.en-q5_1.bin?alt=media",
+}

app/utils/utils.py

@@ -1,8 +1,12 @@
+import os
+import urllib
 import subprocess
 import uuid
 import logging
 import wave
 import gdown
+from tqdm import tqdm
+
 
 from .constant import model_names
 
@@ -102,10 +106,30 @@ def get_model_name(model: str = None):
     return model_names["tiny.en.q5"]
 
 
-def download_from_drive(url , output):
+def download_from_drive(url, output):
     try:
         gdown.download(url, output, quiet=False)
         return True
     except:
         print("Error Occured in Downloading model from Gdrive")
         return False
+
+
+def download_file(url, filepath):
+    try:
+        filename = os.path.basename(url)
+
+        with tqdm(
+            unit="B", unit_scale=True, unit_divisor=1024, miniters=1, desc=filename
+        ) as progress_bar:
+            urllib.request.urlretrieve(
+                url,
+                filepath,
+                reporthook=lambda block_num, block_size, total_size: progress_bar.update(
+                    block_size
+                ),
+            )
+
+        print("File downloaded successfully!")
+    except Exception as e:
+        print(f"An error occurred: {e}")