last update

projet_en_groupe/algorithme_netflix.py

@@ -399,9 +399,10 @@ def directors_nationality(data_1):
                 directors_nationality_dict[director]['nationalities'].add(str(nationality))
 
     # Remove duplicates from the nationalities list for each director
+    # Remove consecutive duplicates from the nationalities list for each director
     for director_info in directors_nationality_dict.values():
-        director_info['nationalities'] = list(set(director_info['nationalities']))
-
+        nationalities_set = set(director_info['nationalities'])
+        director_info['nationalities'] = list(nationalities_set)
     # Sort the directors by the number of movies and series produced
     sorted_directors = sorted(directors_nationality_dict.items(), key=lambda x: x[1]['number of movies or series'], reverse=True)
 
@@ -421,8 +422,6 @@ def directors_nationality(data_1):
     return directors_df
 
 
-
-
 # Allow to filter if we want movie, tv show or both 
 def filter_media_type(data):
     media_type = input("What type of media do you want to display? (Movie/TV Show/Both): ").lower()
@@ -491,11 +490,6 @@ def save_to_csv(data, default_filename='output.csv'):
                 data.to_csv(file_name, index=False)
                 print(f"Data saved to {file_name}")
 
-                # Ask if the user wants to open the file
-                open_choice = input("Do you want to open the saved file? (YES/NO): ").upper()
-                if open_choice == 'YES':
-                    os.system(file_name)
-
             else:
                 # Prompt for a new file name
                 new_filename = input("Enter a new file name (DO NOT include .csv extension): ")