Update app.py

app.py

@@ -3,6 +3,7 @@ import requests, os, random
 from imdb import Cinemagoer
 IMGBB_API_KEY = os.getenv("IMGBB_API_KEY")
 GIST_URL = os.getenv("GIST_URL")
+TMDB_API_KEY = os.getenv("TMDB_API_KEY")
 DATA_DIR = "/data"
 
 app = FastAPI()
@@ -149,6 +150,14 @@ async def bing_image(request: Request):
 async def parentalguide(request: Request):
     data = await request.json()
     imdb_id = data.get("imdb_id")
+    tmdb_id = data.get("tmdb_id",None)
+    if tmdb_id is not None:
+        url = f"https://api.themoviedb.org/3/{media_type}/{tmdb_id}/external_ids"
+        params = {"api_key": TMDB_API_KEY}
+        response = requests.get(url, params=params)
+        if response.status_code == 200:
+            data = response.json()
+            imdb_id = data.get("imdb_id")
     print(imdb_id)
     ia = Cinemagoer()
     movie = ia.get_movie(imdb_id, info=['parents guide'])