Update app.py

app.py

@@ -11,35 +11,64 @@ def spellcheck(text):
     end_time = time.time()
 
     if response.status_code != 200:
-        return "Lỗi kết nối API", "", "", ""
+        return "Lỗi kết nối API", [], text, f"{end_time - start_time:.4f} giây"
 
     result = response.json()
 
-    corrected_text = result.get("corrected_text", text)
-    processing_time = f"{end_time - start_time:.4f} giây"
-
     if not result.get("has_errors", False):
-        return "Không có lỗi chính tả", corrected_text, "", processing_time
+        return "Không có lỗi chính tả", [], text, f"{end_time - start_time:.4f} giây"
 
     errors = result.get("errors", [])
-    error_info = "\n".join([f"Từ sai: `{e['word']}` - Gợi ý: {', '.join(e['suggestions']) if e['suggestions'] else 'Xóa bỏ'}" for e in errors])
 
-    return "Có lỗi chính tả", corrected_text, error_info, processing_time
+    error_list = [(e["word"], e["suggestions"]) for e in errors]
+
+    return "Có lỗi chính tả", error_list, text, f"{end_time - start_time:.4f} giây"
+
+def apply_correction(original_text, corrections):
+    for error_word, chosen_word in corrections:
+        if chosen_word and chosen_word != error_word:
+            original_text = original_text.replace(error_word, chosen_word, 1)
+    return original_text
+
+def update_text(text, corrections):
+    corrected_text = apply_correction(text, corrections)
+    return corrected_text
 
 with gr.Blocks() as demo:
     gr.Markdown("# Demo Kiểm tra lỗi chính tả tiếng Việt")
 
     input_text = gr.Textbox(label="Nhập văn bản")
     output_status = gr.Textbox(label="Trạng thái", interactive=False)
-    output_corrected = gr.Textbox(label="Văn bản sau khi sửa", interactive=False)
-    output_errors = gr.Textbox(label="Chi tiết lỗi", lines=4, interactive=False)
     output_time = gr.Textbox(label="Thời gian xử lý", interactive=False)
+    corrected_text = gr.Textbox(label="Văn bản sau khi sửa", interactive=False)
 
-    submit_btn = gr.Button("Kiểm tra lỗi")
+    correction_inputs = []
+
+    def show_errors(text):
+        status, errors, original_text, time_taken = spellcheck(text)
 
-    submit_btn.click(spellcheck, inputs=input_text, outputs=[output_status, output_corrected, output_errors, output_time])
+        for ci in correction_inputs:
+            demo.remove(ci)
+        correction_inputs.clear()
+
+        choices = []
+        for error_word, suggestions in errors:
+            drop = gr.Dropdown(choices=[error_word] + suggestions, label=f"Từ sai: {error_word}")
+            correction_inputs.append(drop)
+            choices.append((error_word, drop))
+
+        return status, original_text, time_taken, choices
+
+    def apply_changes(text, *chosen_corrections):
+        corrections = [(word, chosen_corrections[i] or word) for i, (word, _) in enumerate(chosen_corrections)]
+        updated_text = update_text(text, corrections)
+        return updated_text
+
+    submit_btn = gr.Button("Kiểm tra lỗi")
+    confirm_btn = gr.Button("Xác nhận sửa")
 
-    gr.Markdown("### Ví dụ: Nhập văn bản có lỗi chính tả để kiểm tra")
+    submit_btn.click(show_errors, inputs=input_text, outputs=[output_status, corrected_text, output_time, correction_inputs])
+    confirm_btn.click(apply_changes, inputs=[input_text] + correction_inputs, outputs=corrected_text)
 
 if __name__ == "__main__":
     demo.launch()