Update app.py

app.py

@@ -269,7 +269,7 @@ def jio_search(query: str) -> str:
         raise HTTPException(status_code=500, detail=f"An error occurred while searching: {str(e)}")
 
         
-def jio_fetch(url: str) -> dict:
+def jio_fetch(url: str, quality: str) -> str:
     try:
         # Construct the API URL
         api_url = f"https://saavn.dev/api/songs?link={url}"
@@ -278,20 +278,35 @@ def jio_fetch(url: str) -> dict:
         response = session.get(api_url)
         # Check if the request was successful
         response.raise_for_status()
-        return response.json()
+        data = response.json()
+        song_data = data.get("data")
+        if not song_data or len(song_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 = song_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.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 cloudscraper.exceptions.CloudflareChallengeError as cf_err:
         logger.error(f"Cloudflare challenge error while fetching {url}: {cf_err}")
         raise HTTPException(status_code=503, detail="Cloudflare challenge failed")
+    except HTTPException:
+        # Re - raise the HTTPException if it's already raised
+        raise
     except Exception as e:
         logger.error(f"Error while fetching {url}: {e}")
         raise HTTPException(status_code=500, detail=f"An error occurred while fetching: {str(e)}")
-
-
+        
 # Define the request model
 class JioDownloadRequest(BaseModel):
     url: str
     quality: str
-        
+    
 @app.post("/jio_dl")
 async def jio_download(request: JioDownloadRequest):
     try:
@@ -300,20 +315,8 @@ async def jio_download(request: JioDownloadRequest):
         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)
-        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}.")
+        download_url = jio_fetch(url, quality)
+        return {"download_url": download_url}
     except HTTPException:
         # Re - raise the HTTPException if it's already raised in jio_fetch
         raise