migrated to fastapi, which has a BackGroundTasks mechanism built in

beeldengeluid · Sep 25, 2024 · 40b80f2 · 40b80f2
1 parent dd0fab7
commit 40b80f2
Show file tree

Hide file tree

Showing 8 changed files with 227 additions and 466 deletions.
diff --git a/apis/__init__.py b/apis/__init__.py
diff --git a/apis/whisper_api.py b/apis/whisper_api.py
diff --git a/app/__init__.py b/app/__init__.py
diff --git a/download.py b/download.py
@@ -47,6 +47,9 @@ def http_download(url: str) -> Optional[DownloadResult]:
             os.makedirs(input_file_dir)
         with open(input_file, "wb") as file:
             response = requests.get(url)
+            if response.status_code >= 400:
+                logger.error(f"Could not download url: {response.status_code}")
+                return None
             file.write(response.content)
             file.close()
     download_time = (time.time() - start_time) * 1000  # time in ms