Update app.py

app.py

@@ -252,30 +252,56 @@ async def get_track_download_url(video_url: str) -> str:
 
 def jio_search(query: str) -> str:
     try:
-        api_url = f"https://saavn.dev/api/search/songs?query={query}/"
+        # Construct the API URL
+        api_url = f"https://saavn.dev/api/search/songs?query={query}"
         session = cloudscraper.create_scraper()
-                
-        # Use the correct API endpoint
+        # Make the API request
         response = session.get(api_url)
-        return response
-
+        # Check if the request was successful
+        response.raise_for_status()
+        return response.text
+    except cloudscraper.exceptions.CloudflareChallengeError as cf_err:
+        logger.error(f"Cloudflare challenge error while searching for {query}: {cf_err}")
+        raise HTTPException(status_code=503, detail="Cloudflare challenge failed")
+    except Exception as e:
+        logger.error(f"Error while searching for {query}: {e}")
+        raise HTTPException(status_code=500, detail=f"An error occurred while searching: {str(e)}")
 
+        
 def jio_fetch(url: str) -> str:
     try:
-        api_url = f"https://saavn.dev/api/songs?link={url}/"
+        # Construct the API URL
+        api_url = f"https://saavn.dev/api/songs?link={url}"
         session = cloudscraper.create_scraper()
-                
-        # Use the correct API endpoint
+        # Make the API request
         response = session.get(api_url)
-        return response
-    
+        # Check if the request was successful
+        response.raise_for_status()
+        return response.text
+    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 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)}")
+
+        
 @app.post("/jio_dl")
 async def jio_download(request: Request):
-    data = await request.json()
-    url = data.get('url')
-    response = jio_fetch(url)
-    return response
-
+    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")
+        response = jio_fetch(url)
+        return response
+    except HTTPException:
+        # Re - raise the HTTPException if it's already raised in jio_fetch
+        raise
+    except Exception as e:
+        logger.error(f"Error in jio_download: {e}")
+        raise HTTPException(status_code=500, detail=f"An error occurred during download: {str(e)}")