2024-10-08T20-16-02Z

src/quizzes/models.py

@@ -2,6 +2,7 @@ import os
 import typing
 
 import openai
+
 from django.contrib.postgres import fields
 from django.db import models
 
@@ -69,6 +70,10 @@ class LLMGradedAnswer(Answer):
             dict: The result of the grading.
         """
         try:
+            # use dotenv to load env
+            from dotenv import load_dotenv
+
+            load_dotenv()
             openai.api_key = os.getenv("OPENAI_API_KEY")
             prompt = f"Grade the following answer based on the rubric:\nRubric: {self.question.rubrics}\nAnswer: {user_answer}"
 
@@ -82,7 +87,7 @@ class LLMGradedAnswer(Answer):
 
             return {"result": "success", "message": response.choices[0].message["content"]}
 
-        except openai.error.OpenAIError as e:
+        except Exception as e:
             print(f"An error occurred: {e}")
             return {"result": "error", "message": str(e)}
 

src/quizzes/views.py

@@ -45,3 +45,25 @@ def grade_question(request, question_id):
         "quizzes/partial.html",
         {"is_correct": is_correct, "correct_answer": answer.correct_answer},
     )
+
+
+def grade_question(request, question_id):
+    question = get_object_or_404(Question, pk=question_id)
+    answer = question.get_answer()
+    if answer is None:
+        return render(request, "quizzes/partial.html", {"error": "Question must have an answer"}, status=422)
+
+    user_answer = request.POST.get("answer")
+    if hasattr(answer, "grade"):
+        result = answer.grade(user_answer)
+        is_correct = result.get("result") == "success"
+        correct_answer = result.get("message")
+    else:
+        is_correct = answer.is_correct(user_answer)
+        correct_answer = answer.correct_answer
+
+    return render(
+        request,
+        "quizzes/partial.html",
+        {"is_correct": is_correct, "correct_answer": correct_answer},
+    )