Update main.py

main.py

@@ -18,6 +18,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 import asyncio
+import aiohttp
 import base64
 import json
 import logging
@@ -341,8 +342,39 @@ def sibyl_system(
     else:
         return {"status": "false", "message": "Not Found User"}
 
+
+async def get_torrent_info(url):
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url) as response:
+                html = await response.text()
+                soup = BeautifulSoup(html, "html.parser")
+                torrents = []
+                for div in soup.find_all("div", class_="modal-torrent"):
+                    quality = div.find("div", class_="modal-quality").find("span").text
+                    all_p = div.find_all("p", class_="quality-size")
+                    quality_type = all_p[0].text if all_p else "N/A"
+                    size = all_p[1].text if len(all_p) > 1 else "N/A"
+                    torrent_link = div.find("a", class_="download-torrent")["href"]
+                    magnet = div.find("a", class_="magnet-download")["href"]
+                    hash = re.search(r"([{a-f\d,A-F\d}]{32,40})\b", magnet).group(0)
+                    torrents.append(
+                        {
+                            "quality": quality,
+                            "type": quality_type,
+                            "size": size,
+                            "torrent": torrent_link,
+                            "magnet": magnet,
+                            "hash": hash,
+                        }
+                    )
+                return torrents
+    except Exception as e:
+        print("Error fetching torrent info:", e)
+        return []
+
 @app.get("/UFoP/yts", response_model=List[MovieInfo])
-def get_movie_info(name: str, api_key: str = Header(...)):
+async def get_movie_info(name: str, api_key: str = Header(...)):
     validate_api_key(api_key)  # Validate API key here
     results = []
     try:
@@ -390,6 +422,11 @@ def get_movie_info(name: str, api_key: str = Header(...)):
                     for torrent in torrent_info.findAll("a"):
                         if "magnet" in torrent["href"]:
                             magnet_links.append(torrent["href"])
+                        elif torrent.text[:3] == "720":
+                            torrent_720 = torrent["href"]
+                        elif torrent.text[:4] == "1080":
+                            torrent_1080 = torrent["href"]
+                    torrents = await get_torrent_info(movie_url)  # Get torrent info asynchronously
                     entry = {
                         "yts_link": movie_url,
                         "name": movie_name,
@@ -398,7 +435,7 @@ def get_movie_info(name: str, api_key: str = Header(...)):
                         "genre": genre,
                         "imdb_ratings": rating,
                         "likes": likes,
-                        "magnet_links": magnet_links
+                        "torrents": torrents,
                     }
                     results.append(entry)
                 except Exception as e: