Fixed selecting available models

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
+from constants import MODELS, MODELS_PATH, BOOKS, CLASSES, STUDENTS, MODEL_NAME_TO_CHOICE
 from utils import save_results, get_test_by_student_class_book
 
 load_dotenv()
@@ -230,17 +230,22 @@ def filter_models(student_name, class_name, 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())
+    available_models = list(MODELS.values())
 
     for test in tests:
-        if test.get("model") in models_list:
-            models_list.remove(test.get("model"))
+        if test.get("model") in available_models:
+            available_models.remove(test.get("model"))
 
-    print("Available models (before update):", models_list)
+    print("Available models (before update):", available_models)
 
-    if not models_list:
+    if not available_models:
         return gr.update(choices=[], label="Немає доступних моделей", value=None, visible=True)
 
+    models_list = []
+
+    for model in list(available_models):
+        models_list.append(MODEL_NAME_TO_CHOICE.get(model))
+
     return gr.update(choices=models_list, value=None, visible=True)
 
 

constants.py

@@ -4,6 +4,12 @@ MODELS = {
     "Claude 3.5 Sonnet": "claude-3-5-sonnet-20241022",
 }
 
+MODEL_NAME_TO_CHOICE = {
+    "gpt-4o": "GPT-4o",
+    "gemini-1.5-pro": "Gemini 1.5 Pro",
+    "claude-3-5-sonnet-20241022": "Claude 3.5 Sonnet",
+}
+
 MODELS_PATH = {
     "GPT-4o": "gpt_4o",
     "Gemini 1.5 Pro": "gemini_1_5_pro",