Upload folder using huggingface_hub

modules/dropbox/audio.py

@@ -69,41 +69,53 @@ CACHE_TTL = timedelta(hours=3, minutes=30)  # refresh before 4h expiry
 
 async def get_audio_urls(req: AudioRequest):
     base_path = f"/{req.scripture_name}/audio"
-    files_to_check = {
-        "recitation": f"{req.global_index}-recitation.mp3",
-        "santhai": f"{req.global_index}-santhai.mp3",
-    }
+    prefix = f"{req.global_index}-"
 
     urls = {}
-    now = datetime.now(timezone.utc)  # timezone-aware UTC datetime
+    now = datetime.now(timezone.utc)
 
-    for key, filename in files_to_check.items():
-        cache_key = (req.scripture_name, req.global_index, key)
+    try:
+        # List all files under the scripture's audio directory
+        entries = dbx.files_list_folder(base_path).entries
+    except dropbox.exceptions.ApiError:
+        raise HTTPException(status_code=404, detail="Audio directory not found")
+
+    # Filter files that match the prefix
+    matching_files = [
+        entry for entry in entries
+        if isinstance(entry, FileMetadata) and entry.name.startswith(prefix)
+    ]
+
+    if not matching_files:
+        raise HTTPException(status_code=404, detail="No audio files found")
+
+    for entry in matching_files:
+        # Extract file type from "<global_index>-<type>.mp3"
+        # Example: "123-recitation.mp3" -> "recitation"
+        filename = entry.name
+        file_type = filename[len(prefix):].rsplit(".", 1)[0]
+
+        cache_key = (req.scripture_name, req.global_index, file_type)
 
-        # Check cache first
+        # Check cache
         cached = audio_cache.get(cache_key)
         if cached and cached["expiry"] > now:
-            urls[key] = cached["url"]
+            urls[file_type] = cached["url"]
             continue
 
-        # Generate new temporary link
+        # Generate temporary link
         file_path = f"{base_path}/{filename}"
         try:
-            metadata = dbx.files_get_metadata(file_path)
-            if isinstance(metadata, FileMetadata):
-                temp_link = dbx.files_get_temporary_link(file_path).link
-                urls[key] = temp_link
-                # store in cache with expiry
-                audio_cache[cache_key] = {"url": temp_link, "expiry": now + CACHE_TTL}
+            temp_link = dbx.files_get_temporary_link(file_path).link
+            urls[file_type] = temp_link
+            audio_cache[cache_key] = {"url": temp_link, "expiry": now + CACHE_TTL}
         except dropbox.exceptions.ApiError:
-            urls[key] = None
-
-    if not any(urls.values()):
-        raise HTTPException(status_code=404, detail="No audio files found")
+            urls[file_type] = None
 
     return urls
 
 
+
 async def cleanup_audio_url_cache(interval_seconds: int = 600):
     """Periodically remove expired entries from audio_cache."""
     while True: