Update generate from pdf functionality

app.py

@@ -282,7 +282,7 @@ async def generate_endpoint(
                 counter += 1
 
         except Exception as e:
-            raise HTTPException(status_code=500, detail=f"Generation from saved file failed: {e}")
+            raise HTTPException(status_code=500, detail=f"Generation from file failed: {e}")
 
     validation_report = None
 

generator.py

@@ -279,6 +279,7 @@ class RAGMCQ:
                     if isinstance(options, list):
                         options = {str(i+1): o for i, o in enumerate(options)}
                     correct_key = payload.get("đáp án") or payload.get("answer") or payload.get("correct") or None
+                    concepts = payload.get("khái niệm sử dụng") or payload.get("concepts") or payload.get("concepts used") or None
                     correct_text = ""
                     if isinstance(correct_key, str) and correct_key.strip() in options:
                         correct_text = options[correct_key.strip()]
@@ -290,6 +291,12 @@ class RAGMCQ:
                     )
                     payload["difficulty"] = {"score": diff_score, "label": diff_label}
 
+                    diff_score, diff_label, components = self._estimate_difficulty_for_generation( # type: ignore
+                        q_text=q_text, options={k: str(v) for k,v in options.items()}, correct_text=str(correct_text), context_text=structured_context, concepts_used=concepts 
+                    )
+
+                    payload["độ khó"] = {"điểm": diff_score, "mức độ": diff_label}
+
                     qcount += 1
                     output[str(qcount)] = mcq_block[item]
                     if qcount >= n_questions:
@@ -969,7 +976,7 @@ class RAGMCQ:
 
                     #? change estimate
                     diff_score, diff_label, components = self._estimate_difficulty_for_generation( # type: ignore
-                        q_text=q_text, options={k: str(v) for k,v in options.items()}, correct_text=str(correct_text), context_text=structured_context, concepts_used = concepts 
+                        q_text=q_text, options={k: str(v) for k,v in options.items()}, correct_text=str(correct_text), context_text=structured_context, concepts_used=concepts 
                     )
 
                     payload["độ khó"] = {"điểm": diff_score, "mức độ": diff_label}