Update GlobalMovies.py

GlobalMovies.py

@@ -21,21 +21,20 @@ def get_foreign_movies(original_language="fr", page=1):
     response = requests.get(url)
     return response.json()["results"]
 
-def build_movie_dataset(language="fr", pages=2):
-    """
-    Builds a dataset of movie titles, descriptions, and original languages.
-    """
-    titles = []
-    descriptions = []
-    languages = []
-    for page in range(1, pages + 1):
-        movies = get_foreign_movies(language, page)
-        for movie in movies:
-            if movie.get("overview"):
-                titles.append(movie["title"])
-                descriptions.append(movie["overview"])
-                languages.append(movie["original_language"])
-    return titles, descriptions, languages
+def build_movie_dataset(languages=["fr", "es", "it"], pages=2):
+    titles, descriptions, langs = [], [], []
+    for lang in languages:
+        for page in range(1, pages + 1):
+            movies = get_foreign_movies(original_language=lang, page=page)
+            for movie in movies:
+                title = movie.get("title", "").strip()
+                overview = movie.get("overview", "").strip()
+                if title and overview:
+                    titles.append(title)
+                    descriptions.append(overview)
+                    langs.append(lang)
+    return titles, descriptions, langs
+
 
 # Build your dataset
 movie_titles, movie_descriptions, movie_languages = build_movie_dataset("fr")