Update app.py

app.py

@@ -161,34 +161,46 @@ def download_track(request: DownloadRequest):
         raise HTTPException(status_code=500, detail=str(e))
 
 
+# Pydantic model for album request
+class AlbumRequest(BaseModel):
+    album_id: str
+
+
 # Fetch album data
 @app.post("/album")
-def fetch_album(album_id: int = Query(..., description="ID of the album to fetch")):
+def fetch_album(request: AlbumRequest):
+    album_id = request.album_id
     try:
         response = requests.get(f"{DEEZER_API_URL}/album/{album_id}")
         response.raise_for_status()
         return response.json()
     except requests.exceptions.RequestException as e:
         logger.error(f"Network error fetching album: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code = 500, detail=str(e))
     except Exception as e:
         logger.error(f"Error fetching album: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code = 500, detail=str(e))
+
+
+# Pydantic model for playlist request
+class PlaylistRequest(BaseModel):
+    playlist_id: str
 
 
 # Fetch playlist data
 @app.post("/playlist")
-def fetch_playlist(playlist_id: int = Query(..., description="ID of the playlist to fetch")):
+def fetch_playlist(request: PlaylistRequest):
+    playlist_id = request.playlist_id
     try:
         response = requests.get(f"{DEEZER_API_URL}/playlist/{playlist_id}")
         response.raise_for_status()
         return response.json()
     except requests.exceptions.RequestException as e:
         logger.error(f"Network error fetching playlist: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code = 500, detail=str(e))
     except Exception as e:
         logger.error(f"Error fetching playlist: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code = 500, detail=str(e))