Update app.py

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
 import difflib
+import nltk
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
+nltk.download('punkt')
+
 def load_model(checkpoint_path):
     tokenizer = AutoTokenizer.from_pretrained(checkpoint_path)
     model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint_path)
@@ -31,20 +34,31 @@ def find_spelling_errors(s1, s2):
     
     return " ".join(highlighted_text)
 
-def process_text(text):
-    corrected_text = correct_spelling(text)
-    highlighted_text = find_spelling_errors(text, corrected_text)
+def process_paragraph(text):
+    sentences = nltk.sent_tokenize(text)  # Chia đoạn văn thành câu
+    corrected_sentences = []
+    highlighted_sentences = []
+
+    for sentence in sentences:
+        corrected = correct_spelling(sentence)
+        highlighted = find_spelling_errors(sentence, corrected)
+        corrected_sentences.append(corrected)
+        highlighted_sentences.append(highlighted)
+
+    highlighted_text = " ".join(highlighted_sentences)
+    corrected_text = " ".join(corrected_sentences)
+
     return highlighted_text, corrected_text
 
 checkpoint_path = "Diezu/Batpho_v2"
 model, tokenizer = load_model(checkpoint_path)
 
 demo = gr.Interface(
-    fn=process_text,
-    inputs=gr.Textbox(placeholder="Nhập văn bản có lỗi chính tả..."),
-    outputs=[gr.HTML(label="Phát hiện lỗi"), gr.Textbox(label="Văn bản đã sửa")],
+    fn=process_paragraph,
+    inputs=gr.Textbox(placeholder="Nhập đoạn văn có lỗi chính tả...", lines=5),
+    outputs=[gr.HTML(label="Phát hiện lỗi"), gr.Textbox(label="Văn bản đã sửa", lines=5)],
     title="Demo Sửa Lỗi Chính Tả",
-    description="Sử dụng mô hình để sửa lỗi chính tả. Từ bị thay thế sẽ được gạch chân đỏ, từ bị xóa có màu cam gạch ngang, từ được thêm có màu xanh đậm."
+    description="Sử dụng mô hình để sửa lỗi chính tả trong đoạn văn. Từ bị thay thế sẽ được gạch chân đỏ, từ bị xóa có màu cam gạch ngang, từ được thêm có màu xanh đậm."
 )
 
 demo.launch()