Update app.py

app.py

@@ -111,11 +111,12 @@ async def match():
     title = track_info["entitiesByUniqueId"][entityUniqueId]["title"]
     artist = track_info["entitiesByUniqueId"][entityUniqueId]["artistName"]
     if youtube_url:
+        video_id = youtube_url.split("v=")[1] if "v=" in url else None
         if title and artist:
             filename = f"{title} - {artist}"
-            return {"url": youtube_url, "filename": filename}
+            return {"url": youtube_url, "filename": filename, "track_id": video_id}
         else:
-            return {"url": youtube_url, "filename": "Unknown Track - Unknown Artist"}
+            return {"url": youtube_url, "filename": "Unknown Track - Unknown Artist", "track_id": video_id}
     else:
                
         search_query = f'{title}+{artist}'
@@ -124,7 +125,7 @@ async def match():
         if 'videoId' in first_song:
             videoId = first_song["videoId"]
             ym_url = f'https://www.youtube.com/watch?v={videoId}'
-            return {"filename": title, "url": ym_url}
+            return {"filename": search_query, "url": ym_url, "track_id": videoId}
         else:
             raise HTTPException(status_code=404, detail="Video ID not found")