Update app.py

app.py

@@ -1,8 +1,8 @@
+import base64
 from fastapi import FastAPI, HTTPException, Request
 import requests
 from bs4 import BeautifulSoup
 import os
-import base64
 
 app = FastAPI()
 
@@ -35,18 +35,27 @@ def get_track(request: Request, track_id: str = None, track_url: str = None):
     
     if track_url:
         track_id = track_url.split("/")[-1]
-
+        
+    # First request to get download URL
     response = requests.get(f"{SPOTIFY_API_URL}/tracks/{track_id}", headers=headers)
     track_data = response.json()
-    respone = requests.get(f"https://downspotifyapis.vercel.app/api/v2/track/${track_id}")
+    
+    respone = requests.get(f"https://downspotifyapis.vercel.app/api/v2/track/{track_id}")
     download_res = respone.json()
-
+    
+    # If URL is not valid, send another request to second URL
+    url = download_res.get("url")
+    if not url:
+        respone = requests.get(f"https://downspotifyapis.vercel.app/api/v1/track/{track_id}")
+        download_res = respone.json()
+        url = download_res.get("url")
+    
     return {
+        "track_id": track_id,
         "name": track_data["name"],
         "album": track_data["album"]["name"],
         "artist": track_data["artists"][0]["name"],
         "release_date": track_data["album"]["release_date"],
         "duration_ms": track_data["duration_ms"],
-        "url": download_res["url"]
-        
+        "url": url
     }