Update app.py

app.py

@@ -35,6 +35,7 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 class ExaminerRequest(BaseModel):
     session_id: str
     num_questions: int = 5
+    comment: Optional[str] = None
 
 
 class SupervisorRequest(BaseModel):
@@ -96,6 +97,9 @@ async def generate_quiz_endpoint(request: ExaminerRequest):
     """
     Generate a quiz based on a previously summarized document.
     Requires a valid session_id from /summarizer.
+    
+    Optional comment parameter can be used to focus the quiz on specific areas,
+    such as supervisor feedback from previous quizzes.
     """
     session = session_manager.get_session(request.session_id)
     if not session:
@@ -111,7 +115,8 @@ async def generate_quiz_endpoint(request: ExaminerRequest):
         quiz = generate_quiz(
             docs=session.docs,
             summary=session.summary,
-            num_questions=request.num_questions
+            num_questions=request.num_questions,
+            comment=request.comment
         )
         session_manager.update_quiz(request.session_id, quiz)