Update app.py

app.py

@@ -1,10 +1,8 @@
 import gradio as gr
 import difflib
-import nltk
+import re
 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)
@@ -34,31 +32,29 @@ def find_spelling_errors(s1, s2):
     
     return " ".join(highlighted_text)
 
-def process_paragraph(text):
-    sentences = nltk.sent_tokenize(text)  # Chia đoạn văn thành câu
+def process_text(text):
+    sentences = re.split(r'(?<=[.!?])\s+', text)
     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
+        if sentence.strip():
+            corrected_text = correct_spelling(sentence)
+            highlighted_text = find_spelling_errors(sentence, corrected_text)
+            corrected_sentences.append(corrected_text)
+            highlighted_sentences.append(highlighted_text)
+    
+    return "<br>".join(highlighted_sentences), "\n".join(corrected_sentences)
 
 checkpoint_path = "Diezu/Batpho_v2"
 model, tokenizer = load_model(checkpoint_path)
 
 demo = gr.Interface(
-    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)],
+    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")],
     title="Demo Sửa Lỗi Chính Tả",
-    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."
+    description="Sử dụng mô hình để sửa lỗi chính tả. Tách câu để xử lý nhanh hơ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()