Update main.py

main.py

@@ -341,6 +341,75 @@ def sibyl_system(
     else:
         return {"status": "false", "message": "Not Found User"}
 
+@app.get("/UFoP/yts", response_model=List[MovieInfo])
+def get_movie_info(name: str, api_key: str = Header(...)):
+    validate_api_key(api_key)  # Validate API key here
+    results = []
+    try:
+        for page in range(1, 2):
+            url = f"https://yts.mx/browse-movies/{name}/all/all/0/seeds/0/all"
+            r = requests.get(url).text
+            soup = BeautifulSoup(r, "lxml")
+            for movie in soup.findAll("div", class_="browse-movie-wrap col-xs-10 col-sm-4 col-md-5 col-lg-4"):
+                mov_name = movie.find("div", class_="browse-movie-bottom")
+                movie_name = mov_name.a.text
+                movie_year = mov_name.div.text
+                movie_name = movie_name + " " + movie_year
+                rating = movie.find("h4", class_="rating", text=True)
+                if rating is not None:
+                    rating = rating.text
+                    rating = rating[:3]
+                else:
+                    rating = "0.0"
+                if rating[2] == "/":
+                    rating = rating[0:2]
+                try:
+                    if movie_name[0] == "[" and movie_name[3] == "]":
+                        movie_name = movie_name[5:]
+                    movie_name = movie_name.replace(" ", "-")
+                    index = 0
+                    for char in movie_name:
+                        if not char.isalnum() and char != "-":
+                            movie_name = movie_name.replace(char, "")
+                    for char in movie_name:
+                        if char == "-" and movie_name[index + 1] == "-":
+                            movie_name = movie_name[:index] + movie_name[index + 1:]
+                        if index < len(movie_name) - 1:
+                            index = index + 1
+                    if "--" in movie_name:
+                        movie_name = movie_name.replace("--", "-")
+                    movie_url = f"https://yts.mx/movie/{movie_name.lower()}"
+                    request = requests.get(movie_url).text
+                    n_soup = BeautifulSoup(request, "lxml")
+                    info = n_soup.find("div", class_="bottom-info")
+                    torrent_info = n_soup.find("p", class_="hidden-xs hidden-sm")
+                    genre = n_soup.findAll("h2")[1].text
+                    likes = info.find("span", id="movie-likes").text
+                    imdb_link = info.find("a", title="IMDb Rating")["href"]
+                    for torrent in torrent_info.findAll("a"):
+                        if torrent.text[:3] == "720":
+                            torrent_720 = torrent["href"]
+                        if torrent.text[:4] == "1080":
+                            torrent_1080 = torrent["href"]
+                    entry = {
+                        "yts_link": movie_url,
+                        "name": movie_name,
+                        "year": movie_year,
+                        "imdb_links": imdb_link,
+                        "genre": genre,
+                        "imdb_ratings": rating,
+                        "likes": likes,
+                        "torrent_720p": torrent_720,
+                        "torrent_1080p": torrent_1080
+                    }
+                    results.append(entry)
+                except Exception as e:
+                    print("Error:", e)
+                    continue
+    except Exception as e:
+        print("Error:", e)
+    return results
+
 
 @app.post(
     "/UFoP/gemini-the-oracle",