Removing models with passed tests for book

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 from dotenv import load_dotenv
 
 from constants import MODELS, MODELS_PATH, BOOKS, CLASSES, STUDENTS
-from utils import save_results
+from utils import save_results, get_test_by_student_class_book
 
 load_dotenv()
 
@@ -201,8 +201,48 @@ def prepare_questions_for_feedback(questions, answer_log):
 
 
 def update_students(class_name):
+    print('update_students')
     students = STUDENTS.get(class_name, [])
-    return gr.update(choices=students, value=None, visible=True)
+    return (
+        gr.update(choices=students, value=None, visible=True),  # student_name_input
+        gr.update(choices=[], value=None, visible=False),  # book_radio
+        gr.update(choices=[], value=None, visible=False),  # model_radio
+    )
+
+
+def handle_student_name_change(student_name):
+    print('handle_student_name_change')
+    if not student_name:
+        return (
+            gr.update(choices=[], value=None, visible=False),  # book_radio
+            gr.update(choices=[], value=None, visible=False),  # model_radio
+        )
+
+    return (
+        gr.update(choices=list(BOOKS.keys()), value=None, visible=True),  # book_radio
+        gr.update(choices=[], value=None, visible=False),  # model_radio
+    )
+
+
+def filter_models(student_name, class_name, book):
+    print('filter_models')
+    if not book:
+        return gr.update(choices=[], value=None, visible=False)
+
+    tests = get_test_by_student_class_book(student_name, class_name, book)
+    models_list = list(MODELS.values())
+
+    for test in tests:
+        if test.get("model") in models_list:
+            models_list.remove(test.get("model"))
+
+    print("Available models (before update):", models_list)
+
+    if not models_list:
+        return gr.update(choices=[], label="### Немає доступних моделей", value=None, visible=True)
+
+    return gr.update(choices=models_list, value=None, visible=True)
+
 
 with gr.Blocks() as demo:
     with gr.Column():
@@ -210,8 +250,8 @@ with gr.Blocks() as demo:
 
         class_name_input = gr.Dropdown(choices=CLASSES, label="Ваш клас", value=None)
         student_name_input = gr.Dropdown(label="Ваше ім'я", value=None, visible=False)
-        model_radio = gr.Radio(list(MODELS.keys()), label="Оберіть модель")
-        book_radio = gr.Radio(list(BOOKS.keys()), label="Оберіть книгу")
+        book_radio = gr.Radio([], label="Оберіть книгу", interactive=True, value=None, visible=False)
+        model_radio = gr.Radio([], label="Оберіть модель", interactive=True, value=None, visible=False)
         load_button = gr.Button("Завантажити питання")
 
         question_output = gr.Markdown(label="Питання")
@@ -234,10 +274,40 @@ with gr.Blocks() as demo:
 
         class_name_input.change(
             update_students,
-            inputs=[class_name_input],
-            outputs=[student_name_input]
+            inputs=[
+                class_name_input,
+            ],
+            outputs=[
+                student_name_input,
+                book_radio,
+                model_radio,
+            ]
+        )
+
+        student_name_input.change(
+            handle_student_name_change,
+            inputs=[
+                student_name_input,
+            ],
+            outputs=[
+                book_radio,
+                model_radio,
+            ]
         )
 
+        book_radio.change(
+            filter_models,
+            inputs=[
+                student_name_input,
+                class_name_input,
+                book_radio,
+            ],
+            outputs=[
+                model_radio,
+            ]
+        )
+
+
         def update_question(model, book, student_name, class_name):
             question, answers = load_questions(model, book, student_name, class_name)
             is_field_filled = len(answers) > 0

utils/db.py

@@ -51,3 +51,9 @@ def save_results(student_name,
         "created_at": db.command("serverStatus")["localTime"]
     })
     print("Saved results")
+
+
+def get_test_by_student_class_book(student_name, class_name, book):
+    print(f"Getting tests for student: {student_name}, class: {class_name}, book: {book}")
+    tests = collection.find({"student_name": student_name, "class": class_name, "book": book})
+    return tests