Books available only for appropriate classes

app.py

@@ -5,7 +5,7 @@ import random
 import gradio as gr
 from dotenv import load_dotenv
 
-from constants import MODELS, MODELS_PATH, BOOKS, CLASSES, STUDENTS, MODEL_NAME_TO_CHOICE
+from constants import MODELS, MODELS_PATH, BOOKS, CLASSES, STUDENTS, MODEL_NAME_TO_CHOICE, BOOKS_FOR_CLASSES
 from utils import save_results, get_test_by_student_class_book
 
 load_dotenv()
@@ -249,7 +249,7 @@ def update_students(class_name, available_models):
     students = STUDENTS.get(class_name, [])
     return (
         gr.update(choices=students, value=None, visible=True),  # student_name_input
-        gr.update(choices=[], value=None, visible=False),  # book_radio
+        gr.update(choices=list(BOOKS_FOR_CLASSES[class_name].keys()), value=None, visible=False),  # book_radio
         gr.update(choices=available_models, value=None, visible=False),  # model_radio
         gr.update(value=""),  # questions_output
         gr.update(choices=[], value=None, visible=False),  # answer_radio
@@ -270,7 +270,7 @@ def update_students(class_name, available_models):
     )
 
 
-def handle_student_name_change(student_name, available_models):
+def handle_student_name_change(student_name, available_models, class_name):
     print("handle_student_name_change", available_models)
     if student_name == "Вчитель":
         available_models = list(MODELS.keys())
@@ -278,7 +278,7 @@ def handle_student_name_change(student_name, available_models):
     print("handle_student_name_change available_models: ", available_models)
     if not student_name:
         return (
-            gr.update(choices=[], value=None, visible=False),  # book_radio
+            gr.update(choices=list(BOOKS_FOR_CLASSES[class_name].keys()), value=None, visible=False),  # book_radio
             gr.update(choices=available_models, value=None, visible=False),  # model_radio
             gr.update(value=""),  # questions_output
             gr.update(choices=[], value=None, visible=False),  # answer_radio
@@ -299,7 +299,7 @@ def handle_student_name_change(student_name, available_models):
         )
 
     return (
-        gr.update(choices=list(BOOKS.keys()), value=None, visible=True),  # book_radio
+        gr.update(choices=list(BOOKS_FOR_CLASSES[class_name].keys()), value=None, visible=True),  # book_radio
         gr.update(choices=available_models, value=None, visible=False),  # model_radio
         gr.update(value=""),  # questions_output
         gr.update(choices=[], value=None, visible=False),  # answer_radio
@@ -513,6 +513,7 @@ with gr.Blocks() as demo:
             inputs=[
                 student_name_input,
                 available_models_state,
+                class_name_input,
             ],
             outputs=[
                 book_radio,

constants.py

@@ -30,6 +30,16 @@ BOOKS = {
     "Іван Нечуйлевицький - Кайдашева сім'я": "kaydasheva-simia.json",
 }
 
+BOOKS_FOR_CLASSES = {
+    "9": {
+        "Тарас Шевченко - Катерина": "katerina.json",
+        "Тарас Шевченко - Сон (комедія)": "son.json",
+    },
+    "10": {
+        "Іван Нечуйлевицький - Кайдашева сім'я": "kaydasheva-simia.json",
+    },
+}
+
 CLASSES = [
     # "7-A", "7-B", "7-C", "7-D",
     "9", "10"