Update app.py

app.py

@@ -1,78 +1,68 @@
 import gradio as gr
 from transformers import pipeline
-import docx
-from pypdf import PdfReader
 
 # ========================
-# Load models
+# Load models (tối ưu cho HF Free + đa ngôn ngữ)
 # ========================
 
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+summarizer = pipeline(
+    "summarization",
+    model="csebuetnlp/mT5_multilingual_XLSum"
+)
 
 quiz_generator = pipeline(
     "text2text-generation",
-    model="google/flan-t5-base"
+    model="google/mt5-small"
 )
 
-# ========================
-# Read file
-# ========================
-
-def extract_text(file):
-    if file.name.endswith(".pdf"):
-        reader = PdfReader(file.name)
-        text = ""
-        for page in reader.pages:
-            page_text = page.extract_text()
-            if page_text:
-                text += page_text + "\n"
-        return text
-
-    elif file.name.endswith(".docx"):
-        doc = docx.Document(file.name)
-        return "\n".join([para.text for para in doc.paragraphs])
-
-    elif file.name.endswith(".txt"):
-        return file.read().decode("utf-8")
-
-    else:
-        return "Định dạng file chưa hỗ trợ."
-
-
 # ========================
 # AI Functions
 # ========================
 
-def summarize_and_quiz(file, num_questions):
-    num_questions = int(num_questions)
+def summarize_and_quiz(text, num_questions):
 
-    text = extract_text(file)
-    text = text[:3000]
+    num_questions = int(num_questions)
 
-    if len(text) < 100:
-        return "Văn bản quá ngắn hoặc không đọc được.", ""
+    if not text or len(text) < 100:
+        return "Văn bản quá ngắn. Hãy nhập ít nhất 100 ký tự.", ""
 
-    summary = summarizer(
-        text,
-        max_length=200,
-        min_length=80,
-        do_sample=False
-    )[0]['summary_text']
+    # Cắt bớt để tránh quá giới hạn model
+    text = text[:3000]
 
+    # --- TÓM TẮT ---
+    try:
+        summary = summarizer(
+            text,
+            max_length=200,
+            min_length=80,
+            do_sample=False
+        )[0]['summary_text']
+    except:
+        summary = "Có lỗi xảy ra trong quá trình tóm tắt."
+
+    # --- 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:
+    Hãy tạo {num_questions} câu hỏi trắc nghiệm dựa trên nội dung sau:
 
     {summary}
 
-    Mỗi câu có 4 lựa chọn A,B,C,D và ghi rõ đáp án đúng.
+    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
     """
 
-    quiz = quiz_generator(prompt, max_length=512)[0]["generated_text"]
+    try:
+        quiz = quiz_generator(
+            prompt,
+            max_length=512
+        )[0]["generated_text"]
+    except:
+        quiz = "Có lỗi xảy ra trong quá trình tạo câu hỏi."
 
     return summary, quiz
 
 
-
 # ========================
 # Gradio Interface
 # ========================
@@ -81,19 +71,26 @@ interface = gr.Interface(
     fn=summarize_and_quiz,
 
     inputs=[
-        gr.File(label="Upload PDF / Word / Text"),
+        gr.Textbox(
+            label="📄 Nhập văn bản (Tiếng Việt hoặc Tiếng Anh)",
+            placeholder="Dán nội dung bài học / tài liệu / văn bản cần tóm tắt vào đây...",
+            lines=15
+        ),
         gr.Slider(5, 10, value=5, step=1, label="Số câu hỏi (Quiz)")
     ],
 
     outputs=[
-        gr.Textbox(label="📌 Nội dung tóm tắt"),
-        gr.Textbox(label="📝 Câu hỏi ôn tập (Quiz)", lines=10)
+        gr.Textbox(label="📌 Nội dung tóm tắt", lines=8),
+        gr.Textbox(label="📝 Câu hỏi ôn tập (Quiz)", lines=12)
     ],
 
     title="📚 AI TÓM TẮT & TẠO QUIZ CHO SINH VIÊN",
     description="""
-    Ứng dụng AI tóm tắt văn bản học thuật và tự động tạo câu hỏi ôn tập (5–10 câu)
-    mà không làm mất nội dung chính.
+    ✅ 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.
     """
 )