Update app.py

app.py

@@ -20,14 +20,16 @@ quiz_generator = pipeline(
 
 def extract_text(file):
     if file.name.endswith(".pdf"):
-        reader = PdfReader(file)
+        reader = PdfReader(file.name)
         text = ""
         for page in reader.pages:
-            text += page.extract_text() + "\n"
+            page_text = page.extract_text()
+            if page_text:
+                text += page_text + "\n"
         return text
 
     elif file.name.endswith(".docx"):
-        doc = docx.Document(file)
+        doc = docx.Document(file.name)
         return "\n".join([para.text for para in doc.paragraphs])
 
     elif file.name.endswith(".txt"):
@@ -36,27 +38,33 @@ def extract_text(file):
     else:
         return "Định dạng file chưa hỗ trợ."
 
+
 # ========================
 # AI Functions
 # ========================
 
 def summarize_and_quiz(file, num_questions):
+    num_questions = int(num_questions)
 
     text = extract_text(file)
+    text = text[:3000]
 
     if len(text) < 100:
-        return "Văn bản quá ngắn.", ""
+        return "Văn bản quá ngắn hoặc không đọc được.", ""
 
-    # --- TÓM TẮT ---
-    summary = summarizer(text, max_length=200, min_length=80, do_sample=False)[0]['summary_text']
+    summary = summarizer(
+        text,
+        max_length=200,
+        min_length=80,
+        do_sample=False
+    )[0]['summary_text']
 
-    # --- TẠO QUIZ ---
     prompt = f"""
     Tạo {num_questions} câu hỏi trắc nghiệm dựa trên nội dung sau:
 
     {summary}
 
-    Câu hỏi ngắn gọn, rõ ràng, có đáp án.
+    Mỗi câu có 4 lựa chọn A,B,C,D và ghi rõ đáp án đúng.
     """
 
     quiz = quiz_generator(prompt, max_length=512)[0]["generated_text"]
@@ -64,6 +72,7 @@ def summarize_and_quiz(file, num_questions):
     return summary, quiz
 
 
+
 # ========================
 # Gradio Interface
 # ========================