Update app.py

app.py

@@ -287,15 +287,27 @@ def jio_fetch(url: str) -> str:
 
         
 @app.post("/jio_dl")
-async def jio_download(request: Request):
+async def jio_download(request: JioDownloadRequest):
     try:
-        data = await request.json()
-        url = data.get('url')
-        if not url:
-            logger.error("No 'url' provided in the request data.")
-            raise HTTPException(status_code=400, detail="Missing 'url' in request data")
+        url = request.url
+        quality = request.quality
+        if not url or not quality:
+            logger.error("Missing 'url' or 'quality' in request data.")
+            raise HTTPException(status_code=400, detail="Missing 'url' or 'quality' in request data")
         response = jio_fetch(url)
-        return response
+        data = response.get("data")
+        if not data or len(data) == 0:
+            logger.error("No data found in the response.")
+            raise HTTPException(status_code=404, detail="No data found for the given URL.")
+        download_urls = data[0].get("downloadUrl")
+        if not download_urls:
+            logger.error("No download URLs found in the response.")
+            raise HTTPException(status_code=404, detail="No download URLs found for the given song.")
+        for download_url in download_urls:
+            if download_url.get("quality") == quality:
+                return {"download_url": download_url.get("url")}
+        logger.error(f"No download URL found for quality {quality}.")
+        raise HTTPException(status_code=404, detail=f"No download URL found for quality {quality}.")
     except HTTPException:
         # Re - raise the HTTPException if it's already raised in jio_fetch
         raise