Update app.py

app.py

@@ -11,7 +11,7 @@ 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, "", "", ""
 
     result = response.json()
 
@@ -19,12 +19,13 @@ def spellcheck(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ả", 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_words = "\n".join([e['word'] for e in errors])
+    suggestions = "\n".join([', '.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
+    return "Có lỗi chính tả", corrected_text, error_words, suggestions, processing_time
 
 with gr.Blocks() as demo:
     gr.Markdown("# Demo Kiểm tra lỗi chính tả tiếng Việt")
@@ -32,14 +33,18 @@ with gr.Blocks() as demo:
     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)
+
+    with gr.Row():
+        output_errors = gr.Textbox(label="Từ sai", lines=4, interactive=False)
+        output_suggestions = gr.Textbox(label="Gợi ý sửa", lines=4, interactive=False)
+
     output_time = gr.Textbox(label="Thời gian xử lý", interactive=False)
 
     submit_btn = gr.Button("Kiểm tra lỗi")
 
-    submit_btn.click(spellcheck, inputs=input_text, outputs=[output_status, output_corrected, output_errors, output_time])
+    submit_btn.click(spellcheck, inputs=input_text, outputs=[output_status, output_corrected, output_errors, output_suggestions, output_time])
 
     gr.Markdown("### Ví dụ: Nhập văn bản có lỗi chính tả để kiểm tra")
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()