Class and student name are dropdowns now

app.py

@@ -5,7 +5,7 @@ import random
 import gradio as gr
 from dotenv import load_dotenv
 
-from constants import MODELS, MODELS_PATH, BOOKS
+from constants import MODELS, MODELS_PATH, BOOKS, CLASSES, STUDENTS
 from utils import save_results
 
 load_dotenv()
@@ -35,7 +35,7 @@ def load_questions(model, book, student_name, class_name):
     current_question_index = 0
     answers_log.clear()
 
-    if not model or not book or not student_name or not class_name:
+    if not model or not book or student_name is None or class_name is None:
         return "# Будь ласка, заповніть усі поля!", []
 
     model_path = MODELS_PATH[model]
@@ -206,16 +206,20 @@ def prepare_questions_for_feedback(questions, answer_log):
     return "\n".join(feedback)
 
 
+def update_students(class_name):
+    students = STUDENTS.get(class_name, [])
+    return gr.update(choices=students, value=None, visible=True)
+
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("# Оберіть модель та книгу, щоб завантажити питання")
 
-        student_name_input = gr.Textbox(label="Ваше ім'я")
-        class_name_input = gr.Textbox(label="Ваш клас")
+        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="Оберіть книгу")
-
         load_button = gr.Button("Завантажити питання")
+
         question_output = gr.Markdown(label="Питання")
         answer_radio = gr.Radio([], label="Варіанти відповіді", interactive=True, visible=False)
         next_button = gr.Button("Наступне питання", visible=False)
@@ -235,6 +239,11 @@ with gr.Blocks() as demo:
         rating_buttons = gr.Radio([-2, -1, 0, 1, 2], label="Оцініть тест", visible=False, interactive=True)
         submit_feedback_button = gr.Button("Завершити тест", visible=False)
 
+        class_name_input.change(
+            update_students,
+            inputs=[class_name_input],
+            outputs=[student_name_input]
+        )
 
         def update_question(model, book, student_name, class_name):
             question, answers = load_questions(model, book, student_name, class_name)

constants.py

@@ -14,3 +14,66 @@ BOOKS = {
     # "Іван Нечуйлевицький - Кайдашева Сім'я": "nechuy-levytskyy-ivan-semenovych-kaydasheva-simia.json",
     "Іван Франко - Захар Беркут": "zahar-berkut.json",
 }
+
+CLASSES = ["7-A", "7-B", "7-C", "7-D"]
+
+STUDENTS = {
+    "7-A": [
+        "Гайворонський Владислав",
+        "Гордін Георгій",
+        "Гуменний Станіслав",
+        "Денчик Данило",
+        "Кичила Ілля",
+        "Климчкук Марія",
+        "Коньков Владислав",
+        "Молчанов Лев",
+        "Нємець Майя",
+        "Панасенко Микита",
+        "Паславська Марія",
+        "Рипаленко Артем",
+        "Савченко Роман",
+        "Смолкіна Соф'я",
+        "Страховецька Софія",
+    ],
+    "7-B": [
+        "Авдєєва Аліса",
+        "Агапов Олександр",
+        "Горюн Любава",
+        "Доротюк Варвара",
+        "Дробот Дмитро",
+        "Зіоменко Олег",
+        "Золотько Варвара",
+        "Ільєнко Софія",
+        "Книш Ксенія",
+        "Кувалдіна Катерина",
+        "Лемішев Кирило",
+        "Хоменко Валерія",
+        "Хорошавіна Софія",
+        "Шепеленко Віола",
+        "Щербина Герасим",
+    ],
+    "7-C": [
+        "Безрук Дар'я",
+        "Бець Вероніка",
+        "Головньова Владислава",
+        "Данченков Микита",
+        "Кадничанський Іосиф",
+        "Конопельцева Мілана",
+        "Муковоз Гліб",
+        "Положій Захар",
+        "Поручіков Владислав",
+        "Прокіпець Вероніка",
+        "Сирота Дмитро",
+        "Сомов Максим",
+        "Таран Вікторія",
+        "Чумачок Софія",
+        "Шигалевська Марія",
+    ],
+    "7-D": [
+        "Капустеринська Аліса",
+        "Леонідова Мілада",
+        "Олійник Давід",
+        "Шевченко Олександра",
+        "Щербакова Марія",
+    ]
+}