Moved constants to separated file

app.py

@@ -4,26 +4,11 @@ import os
 import gradio as gr
 from dotenv import load_dotenv
 
+from constants import MODELS, MODELS_PATH, BOOKS
 from utils import save_results
 
 load_dotenv()
 
-MODELS = {
-    "GPT-4o": "gpt-4o",
-    "Gemini 1.5 Pro": "gemini-1.5-pro",
-    "Claude 3.5 Sonnet": "claude-3-5-sonnet-20241022",
-}
-
-MODELS_PATH = {
-    "GPT-4o": "gpt_4o",
-    "Gemini 1.5 Pro": "gemini_1_5_pro",
-    "Claude 3.5 Sonnet": "claude_3_5_sonnet",
-}
-
-BOOKS = {
-    "Іван Нечуйлевицький - Кайдашева Сім'я": "nechuy-levytskyy-ivan-semenovych-kaydasheva-simia.json",
-}
-
 if os.getenv("ENV_TYPE") == "dev":
     MODELS["Test"] = "test"
     MODELS_PATH["Test"] = "test"
@@ -95,7 +80,7 @@ def summarize_results(student_name, class_name, model, book, feedback, rating):
             "answer": answer
         })
 
-    save_results(student_name, class_name, model, book, questions, feedback, rating)
+    save_results(student_name, class_name, MODELS[model], book, questions, feedback, rating)
 
     correct_answers_count = sum(1 for answer in answers_log if answer['isCorrect'])
     total_questions = len(questions_data)

constants.py

@@ -0,0 +1,15 @@
+MODELS = {
+    "GPT-4o": "gpt-4o",
+    "Gemini 1.5 Pro": "gemini-1.5-pro",
+    "Claude 3.5 Sonnet": "claude-3-5-sonnet-20241022",
+}
+
+MODELS_PATH = {
+    "GPT-4o": "gpt_4o",
+    "Gemini 1.5 Pro": "gemini_1_5_pro",
+    "Claude 3.5 Sonnet": "claude_3_5_sonnet",
+}
+
+BOOKS = {
+    "Іван Нечуйлевицький - Кайдашева Сім'я": "nechuy-levytskyy-ivan-semenovych-kaydasheva-simia.json",
+}