last update

projet_en_groupe/algorithme_netflix.py

@@ -150,13 +150,20 @@ def duration(data_1):
     sort_order = str(input("Entrez le numéro du type de tri : "))
 
     if not type_data.empty:
+        type_data['duration'] = type_data['duration'].str.extract('(\d+)').astype(int)
+
         if sort_order.lower() == '1':
-            type_data_sorted = type_data.sort_values(by='duration', ascending=True)
+            type_data_sorted = type_data.sort_values(by=['type', 'duration'], ascending=[True, True])
         elif sort_order.lower() == '2':
-            type_data_sorted = type_data.sort_values(by='duration', ascending=False)
+            type_data_sorted = type_data.sort_values(by=['type', 'duration'], ascending=[True, False])
         else:
             print("Invalid sort order. Defaulting to ascending order.")
-            type_data_sorted = type_data.sort_values(by='duration', ascending=True)
+            type_data_sorted = type_data.sort_values(by=['type', 'duration'], ascending=[True, True])
+        
+        type_data_sorted['duration'] = type_data_sorted.apply(
+            lambda row: f"{row['duration']} min" if row['type'].lower() == 'movie' else f"{row['duration']} Season", axis=1
+        )
+
 
         print(type_data_sorted)
         save_to_csv(type_data_sorted)
@@ -164,7 +171,6 @@ def duration(data_1):
         print(f"No movies or series found for the type {type_input}.")
 
 
-
 def director(data_1):
     filtered_data = filter_media_type(data_1)