Update app.py

app.py

@@ -11,59 +11,35 @@ def spellcheck(text):
     end_time = time.time()
 
     if response.status_code != 200:
-        return "Lỗi kết nối API", [], text, f"{end_time - start_time:.4f} giây"
+        return "Lỗi kết nối API", "", "", ""
 
     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ả", [], text, f"{end_time - start_time:.4f} giây"
+        return "Không có lỗi chính tả", corrected_text, "", processing_time
 
     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])
 
-    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
+    return "Có lỗi chính tả", corrected_text, error_info, processing_time
 
 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)
-
-    correction_inputs = gr.State([])
-
-    def show_errors(text):
-        status, errors, original_text, time_taken = spellcheck(text)
-
-        choices = [(error_word, gr.Dropdown(choices=[error_word] + suggestions, label=f"Từ sai: {error_word}")) for error_word, suggestions in errors]
-
-        return status, original_text, time_taken, choices
-
-    def apply_changes(text, corrections):
-        corrections = [(word, chosen_word) for word, chosen_word in 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")
 
-    submit_btn.click(show_errors, inputs=input_text, outputs=[output_status, corrected_text, output_time, correction_inputs])
+    submit_btn.click(spellcheck, inputs=input_text, outputs=[output_status, output_corrected, output_errors, output_time])
 
-    confirm_btn.click(lambda text, choices: apply_changes(text, [(word, drop.value) for word, drop in choices]), 
-                      inputs=[input_text, correction_inputs], 
-                      outputs=corrected_text)
+    gr.Markdown("### Ví dụ: Nhập văn bản có lỗi chính tả để kiểm tra")
 
 if __name__ == "__main__":
     demo.launch()