From 5ea21433b1050042e6e00d9b181e81262a5fc42f Mon Sep 17 00:00:00 2001
From: Benjamim Lacerda <benjamim.lacerda16@gmail.com>
Date: Wed, 21 Aug 2024 14:31:35 -0300
Subject: [PATCH] Adiciona a model

---
 src/recommendation_model/utils/model.py | 54 +++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 src/recommendation_model/utils/model.py

diff --git a/src/recommendation_model/utils/model.py b/src/recommendation_model/utils/model.py
new file mode 100644
index 0000000..0763fd4
--- /dev/null
+++ b/src/recommendation_model/utils/model.py
@@ -0,0 +1,54 @@
+from typing import List, Optional
+
+class IVideo:
+    def __init__(self, id: int, title: str, description: Optional[str] = ''):
+        self.id = id
+        self.title = title
+        self.description = description
+        self.catalog = None  # Será preenchido após a categorização
+
+class Catalog:
+    def __init__(self):
+        self.journalism = {
+            "falaJovem": [],
+            "informeUnB": [],
+            "zapping": []
+        }
+        self.interviews = {
+            "brasilEmQuestao": [],
+            "dialogos": [],
+            "tirandoDeLetra": [],
+            "entrevistas": [],
+            "vastoMundo": [],
+            "vozesDiplomaticas": []
+        }
+        self.researchAndScience = {
+            "expliqueSuaTese": [],
+            "fazendoCiencia": [],
+            "radarDaExtencao": [],
+            "seLigaNoPAS": [],
+            "unbTvCiencia": [],
+            "universidadeParaQue": []
+        }
+        self.artAndCulture = {
+            "emCantos": [],
+            "casaDoSom": [],
+            "esbocos": [],
+            "exclusiva": []
+        }
+        self.specialSeries = {
+            "florestaDeGente": [],
+            "guiaDoCalouro": [],
+            "memoriasPauloFreire": [],
+            "desafiosDasEleicoes": [],
+            "vidaDeEstudante": [],
+            "arquiteturaICC": []
+        }
+        self.documentaries = {
+            "miniDoc": [],
+            "documentaries": []
+        }
+        self.varieties = {
+            "pitadasDoCerrado": []
+        }
+        self.unbtv = []
\ No newline at end of file