danielraynaud commited on
Commit
752eafd
·
verified ·
1 Parent(s): 8328a9f

Update study_generators.py

Browse files
Files changed (1) hide show
  1. study_generators.py +3 -3
study_generators.py CHANGED
@@ -7,8 +7,8 @@ logging.basicConfig(level=logging.INFO)
7
  logger = logging.getLogger(__name__)
8
 
9
  class StudyMaterialGenerator:
10
- def __init__(self, db):
11
- self.db = db
12
 
13
  def generate_daily_questions(self, user_id: str, num_questions: int = 10) -> List[Dict]:
14
  """Gera questões diárias baseadas no perfil do usuário"""
@@ -37,7 +37,7 @@ class StudyMaterialGenerator:
37
  logger.error(f"Erro ao gerar questões: {e}")
38
  return []
39
 
40
- def generate_study_material(self, topic: str) -> Dict:
41
  """Gera material de estudo para um tópico específico"""
42
  try:
43
  conn = self.db.get_connection()
 
7
  logger = logging.getLogger(__name__)
8
 
9
  class StudyMaterialGenerator:
10
+ def __init__(self, db_connection):
11
+ self.conn = db_connection
12
 
13
  def generate_daily_questions(self, user_id: str, num_questions: int = 10) -> List[Dict]:
14
  """Gera questões diárias baseadas no perfil do usuário"""
 
37
  logger.error(f"Erro ao gerar questões: {e}")
38
  return []
39
 
40
+ def generate_materials(self, user_id, topic, hours): -> Dict:
41
  """Gera material de estudo para um tópico específico"""
42
  try:
43
  conn = self.db.get_connection()