Update app.py

app.py

@@ -38,27 +38,48 @@ def summarize_and_quiz(text, num_questions):
             do_sample=False
         )[0]['summary_text']
     except:
-        summary = "Có lỗi xảy ra trong quá trình tóm tắt."
+        summary = "❌ Có lỗi xảy ra trong quá trình tóm tắt."
 
     # --- TẠO QUIZ ---
     prompt = f"""
-    Hãy tạo {num_questions} câu hỏi trắc nghiệm dựa trên nội dung sau:
+Dựa trên nội dung sau, hãy tạo {num_questions} câu hỏi trắc nghiệm.
 
-    {summary}
+NỘI DUNG:
+{summary}
 
-    Yêu cầu:
-    - Mỗi câu có 4 lựa chọn: A, B, C, D
-    - Có đáp án đúng
-    - Ngắn gọn, rõ ràng
-    """
+YÊU CẦU:
+1. Mỗi câu hỏi nằm trên 1 dòng riêng
+2. Có 4 đáp án A, B, C, D
+3. Phải ghi đáp án đúng ở cuối câu
+4. Không sử dụng ký hiệu như <extra_id_0> hoặc <extra_id_1>
+5. Viết cùng ngôn ngữ với nội dung (Việt/Anh)
+
+Ví dụ:
+1. Câu hỏi ...?
+A. ...
+B. ...
+C. ...
+D. ...
+Đáp án: B
+
+BẮT ĐẦU TẠO CÂU HỎI:
+"""
 
     try:
         quiz = quiz_generator(
             prompt,
-            max_length=512
+            max_length=512,
+            do_sample=False
         )[0]["generated_text"]
+
+        # Xoá hoàn toàn rác của T5
+        quiz = quiz.replace("<extra_id_0>", "").replace("<extra_id_1>", "").strip()
+
+        if len(quiz) < 30:
+            quiz = "❌ Không tạo được câu hỏi phù hợp. Hãy thử lại với nội dung dài hơn."
+
     except:
-        quiz = "Có lỗi xảy ra trong quá trình tạo câu hỏi."
+        quiz = "❌ Có lỗi xảy ra trong quá trình tạo câu hỏi."
 
     return summary, quiz
 
@@ -87,10 +108,12 @@ interface = gr.Interface(
     title="📚 AI TÓM TẮT & TẠO QUIZ CHO SINH VIÊN",
     description="""
     ✅ Hỗ trợ: Tiếng Việt + Tiếng Anh  
-    ✅ Tóm tắt không làm mất ý chính  
-    ✅ Tạo 5–10 câu hỏi ôn tập ngay lập tức  
-    
-    Chỉ cần dán nội dung vào và nhấn Submit.
+    ✅ Đã fix lỗi: <extra_id_0>  
+    ✅ Tóm tắt không mất ý chính  
+    ✅ Tạo 5–10 câu hỏi trắc nghiệm tự động  
+    ✅ Tối ưu chạy trên HuggingFace Free  
+
+    👉 Chỉ cần dán nội dung vào và nhấn Submit.
     """
 )