Update main.py

main.py

@@ -386,10 +386,13 @@ def get_movie_info(name: str, api_key: str = Header(...)):
                     genre = n_soup.findAll("h2")[1].text
                     likes = info.find("span", id="movie-likes").text
                     imdb_link = info.find("a", title="IMDb Rating")["href"]
+                    magnet_links = []
                     for torrent in torrent_info.findAll("a"):
-                        if torrent.text[:3] == "720":
+                        if "magnet" in torrent["href"]:
+                            magnet_links.append(torrent["href"])
+                        elif torrent.text[:3] == "720":
                             torrent_720 = torrent["href"]
-                        if torrent.text[:4] == "1080":
+                        elif torrent.text[:4] == "1080":
                             torrent_1080 = torrent["href"]
                     entry = {
                         "yts_link": movie_url,
@@ -400,7 +403,8 @@ def get_movie_info(name: str, api_key: str = Header(...)):
                         "imdb_ratings": rating,
                         "likes": likes,
                         "torrent_720p": torrent_720,
-                        "torrent_1080p": torrent_1080
+                        "torrent_1080p": torrent_1080,
+                        "magnet_links": magnet_links
                     }
                     results.append(entry)
                 except Exception as e: