Update app.py

app.py

@@ -19,10 +19,14 @@ def find_spelling_errors(s1, s2):
     b = s2.split()
     matcher = difflib.SequenceMatcher(None, a, b)
     highlighted_text = []
+    corrected_words = {}
     
     for tag, i1, i2, j1, j2 in matcher.get_opcodes():
         if tag == 'replace':
-            highlighted_text.append(f'<span style="color: red; text-decoration: underline;">{" ".join(a[i1:i2])}</span>')
+            incorrect_word = " ".join(a[i1:i2])
+            correct_word = " ".join(b[j1:j2])
+            highlighted_text.append(f'<button onclick="replaceWord(this, \"{correct_word}\")" style="color: red; text-decoration: underline; border: none; background: none;">{incorrect_word}</button>')
+            corrected_words[incorrect_word] = correct_word
         elif tag == 'delete':
             highlighted_text.append(f'<span style="color: orange; text-decoration: line-through;">{" ".join(a[i1:i2])}</span>')
         elif tag == 'insert':
@@ -30,31 +34,44 @@ def find_spelling_errors(s1, s2):
         else:
             highlighted_text.append(" ".join(a[i1:i2]))
     
-    return " ".join(highlighted_text)
+    return " ".join(highlighted_text), corrected_words
 
 def process_text(text):
     sentences = re.split(r'(?<=[.!?])\s+', text)
     corrected_sentences = []
     highlighted_sentences = []
+    all_corrected_words = {}
     
     for sentence in sentences:
         if sentence.strip():
             corrected_text = correct_spelling(sentence)
-            highlighted_text = find_spelling_errors(sentence, corrected_text)
+            highlighted_text, corrected_words = find_spelling_errors(sentence, corrected_text)
             corrected_sentences.append(corrected_text)
             highlighted_sentences.append(highlighted_text)
+            all_corrected_words.update(corrected_words)
     
-    return "<br>".join(highlighted_sentences), "\n".join(corrected_sentences)
+    return "<br>".join(highlighted_sentences), "\n".join(corrected_sentences), all_corrected_words
 
-checkpoint_path = "Diezu/Batpho_v2"
-model, tokenizer = load_model(checkpoint_path)
+def apply_full_correction(text, corrected_words):
+    for incorrect, correct in corrected_words.items():
+        text = text.replace(incorrect, f'<span style="color: blue; font-weight: bold;">{correct}</span>')
+    return text
 
-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")],
-    title="Demo Sửa Lỗi Chính Tả",
-    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 = gr.Blocks()
+with demo:
+    input_text = gr.Textbox(placeholder="Nhập văn bản có lỗi chính tả...")
+    output_html = gr.HTML()
+    corrected_textbox = gr.Textbox()
+    corrected_words_state = gr.State({})
+    
+    btn_check = gr.Button("Phát hiện lỗi")
+    btn_correct_all = gr.Button("Sửa toàn bộ lỗi")
+    
+    def update_text(text):
+        highlighted, corrected, corrected_words = process_text(text)
+        return highlighted, corrected, corrected_words
+    
+    btn_check.click(update_text, inputs=input_text, outputs=[output_html, corrected_textbox, corrected_words_state])
+    btn_correct_all.click(apply_full_correction, inputs=[input_text, corrected_words_state], outputs=output_html)
 
 demo.launch()