Adicao da funcao para transformar os videos em dataframe

src/recommendation_model/utils/get_videos.py

@@ -55,4 +55,23 @@ def clean_text(videos: List[IVideo]):
         video.description = video.description.translate(str.maketrans('', '', string.punctuation))
         video.title = video.title.translate(str.maketrans('', '', string.punctuation))
 
-    return videos
+    return videos
+
+# Função para transformar a lista de vídeos em um dataframe
+def videos_to_dataframe(videos: List[IVideo]) -> pd.DataFrame:
+    videos = clean_text(videos)
+
+    # Converte a lista de objetos IVideo para uma lista de dicionários
+    video_dicts = [
+        {
+            'ID': video.id,
+            'Título': video.title,
+            'Descrição': video.description,
+            'Categoria': video.catalog
+        }
+        for video in videos
+    ]
+
+    df = pd.DataFrame(video_dicts)
+
+    return df