patch 2

api.py

@@ -1,5 +1,6 @@
 import requests
 import logging
+from urllib.parse import quote
 
 class InstancesAPI:
     def __init__(self, instances):
@@ -16,6 +17,7 @@ class InstancesAPI:
                 logging.error(f"Error contacting instance {instance_url}: {e}")
         return reports
 
+
     def download_music(self, instance_url, title):
         """
         Download a music file to an instance.
@@ -29,7 +31,10 @@ class InstancesAPI:
         """
         data = {}
         try:
-            response = requests.get(f"{instance_url}/api/get/music/{title}")
+            # URL-encode the title to handle special characters
+            encoded_title = quote(title)
+            
+            response = requests.get(f"{instance_url}/api/get/music/{encoded_title}")
             response.raise_for_status()
             data = response.json()
             

app.py

@@ -5,6 +5,7 @@ import os
 from typing import List
 from LoadBalancer import LoadBalancer
 from utils import is_valid_url
+from urllib.parse import unquote
 
 CACHE_DIR = os.getenv("CACHE_DIR")
 TOKEN = os.getenv("TOKEN")
@@ -93,24 +94,30 @@ async def get_instances_health():
 @app.get("/api/get/music/{file_name}")
 async def get_music_api(file_name: str):
     """Endpoint to get the music file by title."""
-    if not file_name:
+    
+    # Decode the file_name from URL encoding
+    decoded_file_name = unquote(file_name)
+    
+    if not decoded_file_name:
         raise HTTPException(status_code=400, detail="file_name parameter is required")
     
     # Check if the music file is already cached
-    if file_name in load_balancer.MUSIC_STORE:
-        url = load_balancer.MUSIC_STORE[file_name]
+    if decoded_file_name in load_balancer.MUSIC_STORE:
+        url = load_balancer.MUSIC_STORE[decoded_file_name]
         return JSONResponse(content={"url": url})
 
-    music_path = load_balancer.find_music_path(file_name)
+    music_path = load_balancer.find_music_path(decoded_file_name)
     
     if not music_path:
         raise HTTPException(status_code=404, detail="Music file not found")
     
     # Start the download in an instance
-    response = load_balancer.download_music_to_best_instance(file_name=file_name)
+    response = load_balancer.download_music_to_best_instance(file_name=decoded_file_name)
+    
     if response:
         return JSONResponse(content=response)
 
+
 @app.get('/api/get/category/all')
 async def get_categories():
     return load_balancer.get_all_categories()