Update app.py

app.py

@@ -260,29 +260,34 @@ def jio_search(query: str, quality: str) -> str:
         response = session.get(api_url)
         # Check if the request was successful
         response.raise_for_status()
+        # Get the data from the response
         data = response.json().get("data")
-        song_data = data.get("results")
-        if not song_data or len(song_data) == 0:
+        if not data:
             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}.")
+            raise HTTPException(status_code=404, detail="No data found for the given query.")
+        # Get the song results
+        song_results = data.get("results")
+        if not song_results or len(song_results) == 0:
+            logger.error("No song results found in the response.")
+            raise HTTPException(status_code=404, detail="No song results found for the given query.")
+        # Iterate through each song result
+        for song in song_results:
+            download_urls = song.get("downloadUrl")
+            if download_urls:
+                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} in the search results for query {query}.")
+        raise HTTPException(status_code=404, detail=f"No download URL found for quality {quality} in the search results for query {query}.")
     except cloudscraper.exceptions.CloudflareChallengeError as cf_err:
-        logger.error(f"Cloudflare challenge error while fetching {url}: {cf_err}")
+        logger.error(f"Cloudflare challenge error while searching for {query}: {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)}")
+        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, quality: str) -> str: