Update quizz_api.py

quizz_api.py

@@ -36,13 +36,15 @@ class QuizQuestion(BaseModel):
     question: str
     options: List[str]
     correct_answer: str
-    explanation: str   # ✅ ADDED
+    explanation: str
+
 
 class QuizRequest(BaseModel):
     topic: str
     num_questions: int = 5
     difficulty: Optional[str] = "medium"
 
+
 class QuizResponse(BaseModel):
     questions: List[QuizQuestion]
     topic: str
@@ -71,8 +73,8 @@ async def root():
 async def generate_quiz(request: QuizRequest):
     try:
         query = (
-            f"Create a {request.num_questions}-question {request.difficulty} "
-            f"difficulty quiz about {request.topic}"
+            f"Create a {request.num_questions}-question "
+            f"{request.difficulty} difficulty quiz about {request.topic}"
         )
 
         response = ""
@@ -93,7 +95,7 @@ async def generate_quiz(request: QuizRequest):
 
 
 # =========================
-# PARSER UPDATED
+# ROBUST PARSER
 # =========================
 
 def parse_quiz_response(response: str) -> List[dict]:
@@ -120,13 +122,20 @@ def parse_quiz_response(response: str) -> List[dict]:
             option_text = line.split(")", 1)[1].strip()
             current_question["options"].append(option_text)
 
-        # Detect Answer
+        # Detect Answer (robust extraction)
         elif line.lower().startswith("answer:"):
-            correct_letter = line.split(":")[1].strip()
-            index = ord(correct_letter.upper()) - ord("A")
-
-            if 0 <= index < len(current_question["options"]):
-                current_question["correct_answer"] = current_question["options"][index]
+            raw_answer = line.split(":", 1)[1].strip()
+
+            correct_letter = None
+            for char in raw_answer:
+                if char.upper() in ["A", "B", "C", "D"]:
+                    correct_letter = char.upper()
+                    break
+
+            if correct_letter:
+                index = ord(correct_letter) - ord("A")
+                if 0 <= index < len(current_question["options"]):
+                    current_question["correct_answer"] = current_question["options"][index]
 
         # Detect Explanation
         elif line.lower().startswith("explanation:"):