Saving test results and feedback in MongoDB. utils refactoring

app.py

@@ -1,8 +1,11 @@
 import json
 import os
+
 import gradio as gr
 from dotenv import load_dotenv
 
+from utils import save_results
+
 load_dotenv()
 
 MODELS = {
@@ -64,7 +67,8 @@ def get_next_question(selected_answer):
     # Writing answer in log
     answers_log.append({
         "selected": selected_answer,
-        "isCorrect": any(answer['answer'] == selected_answer and answer['isCorrect'] for answer in questions_data[current_question_index]['answers'])
+        "isCorrect": any(answer['answer'] == selected_answer and answer['isCorrect'] for answer in
+                         questions_data[current_question_index]['answers'])
     })
 
     # Move to the next question
@@ -73,14 +77,26 @@ def get_next_question(selected_answer):
         next_question = questions_data[current_question_index]
         question_text = f"# Питання:\n## {next_question['question']}"
         answers = [answer['answer'] for answer in next_question['answers']]
-        return question_text, gr.update(choices=answers, interactive=True), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        return question_text, gr.update(choices=answers, interactive=True), gr.update(visible=True), gr.update(
+            visible=False), gr.update(visible=False), gr.update(visible=False)
     else:
         # All questions are completed — ask for feedback
         question_text = "Дякуємо за участь у тесті! Залиште, будь ласка, свій відгук і оцініть тест."
-        return question_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+        return question_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(
+            visible=True), gr.update(visible=True)
+
+
+def summarize_results(student_name, class_name, model, book, feedback, rating):
+    global questions_data, answers_log
+    questions = []
+    for question, answer in zip(questions_data, answers_log):
+        questions.append({
+            "question": question,
+            "answer": answer
+        })
 
+    save_results(student_name, class_name, model, book, questions, feedback, rating)
 
-def summarize_results(feedback, rating):
     correct_answers_count = sum(1 for answer in answers_log if answer['isCorrect'])
     total_questions = len(questions_data)
     summary = f"# Усі питання пройдено.\n\n## Ви відповіли правильно на {correct_answers_count} з {total_questions} питань.\n\n### Ваш відгук: {feedback}\n### Ваша оцінка: {rating}/5"
@@ -91,6 +107,8 @@ with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("# Оберіть модель та книгу, щоб завантажити питання")
 
+        student_name_input = gr.Textbox(label="Ваше ім'я")
+        class_name_input = gr.Textbox(label="Ваш клас")
         model_radio = gr.Radio(list(MODELS.keys()), label="Оберіть модель")
         book_radio = gr.Radio(list(BOOKS.keys()), label="Оберіть книгу")
 
@@ -103,9 +121,12 @@ with gr.Blocks() as demo:
         rating_slider = gr.Slider(1, 5, step=1, label="Оцініть тест", visible=False)
         submit_feedback_button = gr.Button("Завершити тест", visible=False)
 
+
         def update_question(model, book):
             question, answers = load_questions(model, book)
-            return question, gr.update(choices=answers, interactive=True, visible=True), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+            return question, gr.update(choices=answers, interactive=True, visible=True), gr.update(
+                visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+
 
         load_button.click(
             update_question,
@@ -121,10 +142,9 @@ with gr.Blocks() as demo:
 
         submit_feedback_button.click(
             summarize_results,
-            inputs=[feedback_input, rating_slider],
+            inputs=[student_name_input, class_name_input, model_radio, book_radio, feedback_input, rating_slider],
             outputs=[question_output, feedback_input, rating_slider, submit_feedback_button]
         )
 
-
 if __name__ == "__main__":
     demo.launch()

utils/__init__.py

@@ -0,0 +1,2 @@
+from .utils import *
+from .db import *

utils/db.py

@@ -0,0 +1,26 @@
+import os
+
+from dotenv import load_dotenv
+from pymongo import MongoClient
+
+load_dotenv()
+
+client = MongoClient(os.getenv("MONGO_CONNECTION_STRING"))
+
+db = client[os.getenv("MONGO_DB_NAME")]
+
+collection_name = os.getenv("DEV_MONGO_COLLECTION") if os.getenv("ENV_TYPE") == "dev" else os.getenv(
+    "PROD_MONGO_COLLECTION")
+collection = db[collection_name]
+
+
+def save_results(student_name, class_name, model, book, questions, feedback, rating):
+    collection.insert_one({
+        "student_name": student_name,
+        "class": class_name,
+        "model": model,
+        "book": book,
+        "questions": questions,
+        "feedback": feedback,
+        "rating": rating
+    })