Update app.py

app.py

@@ -2,220 +2,304 @@ import gradio as gr
 from pdf2docx import Converter
 import os
 import zipfile
+import cv2
+import numpy as np
+from PIL import Image
+
+# Thử import PaddleOCR, nếu chưa cài thì báo lỗi nhẹ
+try:
+    from paddleocr import PaddleOCR
+    # Khởi tạo model OCR (tải model lần đầu sẽ hơi lâu)
+    # lang='vi' để hỗ trợ tiếng Việt
+    ocr_engine = PaddleOCR(use_angle_cls=True, lang='vi', show_log=False)
+    HAS_OCR = True
+except ImportError:
+    HAS_OCR = False
+    print("⚠️ CHƯA CÀI PADDLEOCR: Vui lòng chạy 'pip install paddlepaddle paddleocr'")
 
 # --- 1. CẤU HÌNH NGÔN NGỮ (TỪ ĐIỂN) ---
-# Tại đây chứa tất cả văn bản hiển thị
 TRANS = {
     "vi": {
-        "lang_btn": "🇬🇧 Switch to English",
-        "title": "# 📝 CHUYỂN ĐỔI PDF SANG WORD\n### Công cụ hỗ trợ Đề thi & Tài liệu (Hỗ trợ nhiều file)",
-        "input_group": "### 1. Tải tài liệu",
-        "input_label": "Chọn file PDF (Giữ Shift/Ctrl để chọn nhiều)",
-        "output_group": "### 2. Tải về kết quả",
-        "output_label": "File Word hoặc ZIP",
-        "status_label": "Trạng thái",
-        "convert_btn": "🚀 BẮT ĐẦU CHUYỂN ĐỔI",
-        "info_accordion": "ℹ️ Thông tin phát triển & Bản quyền",
+        "lang_btn": "🇬🇧 English",
+        "app_title": "DOCUMENTS TOOLKIT PRO",
+        # Sidebar items
+        "menu_label": "🛠️ Chọn Chức năng",
+        "menu_pdf": "📝 Chuyển PDF sang Word",
+        "menu_ocr": "👁️ OCR (Trích xuất chữ từ ảnh)",
+        "info_header": "ℹ️ Thông tin",
+        
+        # PDF Tool
+        "pdf_title": "### 📄 CHUYỂN ĐỔI PDF SANG WORD",
+        "pdf_input_label": "Tải file PDF (Chọn nhiều file)",
+        "pdf_btn": "🚀 CHUYỂN ĐỔI NGAY",
+        
+        # OCR Tool
+        "ocr_title": "### 👁️ TRÍCH XUẤT VĂN BẢN (OCR)",
+        "ocr_input_label": "Tải hình ảnh (PNG, JPG)",
+        "ocr_btn": "🔍 QUÉT VÀ TRÍCH XUẤT",
+        "ocr_output_text": "Nội dung văn bản",
+        "ocr_output_img": "Ảnh đã khoanh vùng chữ",
+        
+        # Messages
+        "msg_warning": "⚠️ Vui lòng tải file lên!",
+        "msg_processing": "Đang xử lý...",
+        "msg_success": "✅ Hoàn tất!",
+        "msg_error": "❌ Lỗi: ",
+        "ocr_not_installed": "⚠️ Lỗi: Chưa cài thư viện PaddleOCR.",
+        
+        # Info Content
         "info_content": """
-            ### 👨‍💻 Thông tin tác giả
-            * **Phát triển bởi:** Chu Viết Kiên
-            * **Liên hệ:** kiencv.3107@gmail.com
-            * **Phiên bản:** 1.2.0 (Multi-language)
-            
-            ### 📜 Bản quyền
-            * Công cụ sử dụng thư viện `pdf2docx`.
-            * File sẽ tự động xóa sau khi phiên làm việc kết thúc.
-            """,
-        # Các thông báo trong code xử lý
-        "msg_warning": "⚠️ Vui lòng tải ít nhất một file lên!",
-        "msg_process": "Đang xử lý file",
-        "msg_zip": "Đang đóng gói file...",
-        "msg_done": "Xong!",
-        "msg_success_single": "✅ Hoàn tất! Đã chuyển đổi:",
-        "msg_success_multi": "✅ Hoàn tất! Đã nén {n} file vào ZIP.",
-        "msg_error": "❌ Có lỗi xảy ra:"
+            **Phát triển bởi:** Chu Viết Kiên  
+            **Liên hệ:** kiencv.3107@gmail.com  
+            **Phiên bản:** 2.0 (OCR Update)
+        """
     },
     "en": {
-        "lang_btn": "🇻🇳 Chuyển sang Tiếng Việt",
-        "title": "# 📝 PDF TO WORD CONVERTER\n### Professional Tool for Documents (Batch Support)",
-        "input_group": "### 1. Upload Documents",
-        "input_label": "Select PDF Files (Hold Shift/Ctrl to select multiple)",
-        "output_group": "### 2. Download Result",
-        "output_label": "Word File or ZIP",
-        "status_label": "Status",
-        "convert_btn": "🚀 CONVERT NOW",
-        "info_accordion": "ℹ️ Developer Info & Copyright",
+        "lang_btn": "🇻🇳 Tiếng Việt",
+        "app_title": "DOCUMENTS TOOLKIT PRO",
+        # Sidebar items
+        "menu_label": "🛠️ Select Tool",
+        "menu_pdf": "📝 PDF to Word Converter",
+        "menu_ocr": "👁️ OCR (Image to Text)",
+        "info_header": "ℹ️ Information",
+        
+        # PDF Tool
+        "pdf_title": "### 📄 PDF TO WORD CONVERTER",
+        "pdf_input_label": "Upload PDF Files",
+        "pdf_btn": "🚀 CONVERT NOW",
+        
+        # OCR Tool
+        "ocr_title": "### 👁️ OPTICAL CHARACTER RECOGNITION (OCR)",
+        "ocr_input_label": "Upload Image (PNG, JPG)",
+        "ocr_btn": "🔍 SCAN & EXTRACT",
+        "ocr_output_text": "Extracted Text",
+        "ocr_output_img": "Annotated Image",
+        
+        # Messages
+        "msg_warning": "⚠️ Please upload a file!",
+        "msg_processing": "Processing...",
+        "msg_success": "✅ Done!",
+        "msg_error": "❌ Error: ",
+        "ocr_not_installed": "⚠️ Error: PaddleOCR library not installed.",
+        
+        # Info Content
         "info_content": """
-            ### 👨‍💻 Author Info
-            * **Developer:** Chu Viet Kien
-            * **Contact:** kiencv.3107@gmail.com
-            * **Version:** 1.2.0 (Multi-language)
-            
-            ### 📜 Disclaimer
-            * This tool uses `pdf2docx` library.
-            * Files are automatically deleted after the session.
-            """,
-        # Messages inside logic
-        "msg_warning": "⚠️ Please upload at least one file!",
-        "msg_process": "Processing file",
-        "msg_zip": "Zipping files...",
-        "msg_done": "Done!",
-        "msg_success_single": "✅ Success! Converted:",
-        "msg_success_multi": "✅ Success! Compressed {n} files into ZIP.",
-        "msg_error": "❌ Error occurred:"
+            **Developer:** Chu Viet Kien  
+            **Contact:** kiencv.3107@gmail.com  
+            **Version:** 2.0 (OCR Update)
+        """
     }
 }
 
 # --- 2. HÀM XỬ LÝ LOGIC ---
+
+# 2.1 Logic PDF -> Word
 def convert_pdfs_to_word(pdf_files, lang_code, progress=gr.Progress()):
-    # Lấy từ điển ngôn ngữ hiện tại
     T = TRANS[lang_code]
-
-    if not pdf_files:
-        return None, T["msg_warning"]
+    if not pdf_files: return None, T["msg_warning"]
     
-    if not isinstance(pdf_files, list):
-        pdf_files = [pdf_files]
-
+    if not isinstance(pdf_files, list): pdf_files = [pdf_files]
     converted_files = []
-    total_files = len(pdf_files)
     
     try:
-        for index, pdf_file in enumerate(pdf_files):
-            original_name = os.path.basename(pdf_file.name)
-            docx_filename = os.path.splitext(original_name)[0] + ".docx"
-            
-            # Thông báo tiến trình đa ngôn ngữ
-            msg = f"{T['msg_process']} {index + 1}/{total_files}: {original_name}"
-            progress((index / total_files), desc=msg)
-            
+        for idx, pdf_file in enumerate(pdf_files):
+            progress((idx / len(pdf_files)), desc=f"Processing {os.path.basename(pdf_file.name)}...")
+            docx_name = os.path.splitext(os.path.basename(pdf_file.name))[0] + ".docx"
             cv = Converter(pdf_file.name)
-            cv.convert(docx_filename, start=0, end=None)
+            cv.convert(docx_name, start=0, end=None)
             cv.close()
-            converted_files.append(docx_filename)
-
-        progress(0.95, desc=T["msg_zip"])
-        
+            converted_files.append(docx_name)
+            
         if len(converted_files) == 1:
-            final_output = converted_files[0]
-            status_msg = f"{T['msg_success_single']} {os.path.basename(final_output)}"
+            return converted_files[0], T["msg_success"]
         else:
-            zip_filename = "Converted_Documents.zip"
-            with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
-                for file in converted_files:
-                    zipf.write(file, arcname=os.path.basename(file))
-            final_output = zip_filename
-            status_msg = T["msg_success_multi"].format(n=len(converted_files))
+            zip_name = "Converted_Docs.zip"
+            with zipfile.ZipFile(zip_name, 'w') as zf:
+                for f in converted_files: zf.write(f)
+            return zip_name, T["msg_success"]
+    except Exception as e:
+        return None, f"{T['msg_error']} {str(e)}"
+
+# 2.2 Logic OCR
+def run_ocr_func(image, lang_code):
+    T = TRANS[lang_code]
+    if not HAS_OCR: return None, None, T["ocr_not_installed"]
+    if image is None: return None, None, T["msg_warning"]
 
-        progress(1.0, desc=T["msg_done"])
-        return final_output, status_msg
+    try:
+        # PaddleOCR nhận đường dẫn file hoặc numpy array
+        # Gradio Image input mặc định trả về numpy array (RGB)
+        
+        # Chạy OCR
+        result = ocr_engine.ocr(image, cls=True)
+        
+        # Xử lý kết quả
+        txts = []
+        boxes = []
+        scores = []
+        
+        # Result cấu trúc: [ [ [box], (text, score) ], ... ]
+        # Đôi khi result trả về list lồng nhau nếu có nhiều vùng
+        if result and result[0]: 
+            for line in result[0]:
+                boxes.append(line[0])
+                txts.append(line[1][0])
+                scores.append(line[1][1])
+        
+        # Tạo văn bản kết quả
+        full_text = "\n".join(txts)
+        
+        # Vẽ box lên ảnh để hiển thị (Dùng OpenCV)
+        # Convert RGB (Gradio) -> BGR (OpenCV)
+        img_cv = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+        for box in boxes:
+            box = np.array(box).astype(np.int32).reshape((-1, 1, 2))
+            cv2.polylines(img_cv, [box], True, (0, 0, 255), 2)
+            
+        # Convert ngược lại BGR -> RGB để hiển thị trên Gradio
+        final_img = cv2.cvtColor(img_cv, cv2.COLOR_BGR2RGB)
+        
+        # Lưu file text để tải về (tuỳ chọn)
+        with open("ocr_result.txt", "w", encoding="utf-8") as f:
+            f.write(full_text)
+            
+        return full_text, final_img, "ocr_result.txt"
 
     except Exception as e:
-        return None, f"{T['msg_error']} {str(e)}"
+        return str(e), None, None
 
-# --- 3. HÀM ĐỔI NGÔN NGỮ GIAO DIỆN ---
-def change_language(current_lang):
-    # Đảo ngược ngôn ngữ
-    new_lang = "en" if current_lang == "vi" else "vi"
+# --- 3. UI HELPER FUNCTIONS ---
+
+def change_ui_language(lang):
+    new_lang = "en" if lang == "vi" else "vi"
     T = TRANS[new_lang]
     
-    # Trả về updates cho từng thành phần UI
     return (
-        new_lang,                       # Cập nhật State
-        T["lang_btn"],                  # Cập nhật nút đổi ngôn ngữ
-        T["title"],                     # Cập nhật Tiêu đề
-        T["input_group"],               # Cập nhật Group Input
-        gr.update(label=T["input_label"]), # Cập nhật Label Input File
-        T["output_group"],              # Cập nhật Group Output
-        gr.update(label=T["output_label"]), # Cập nhật Label Output File
-        gr.update(label=T["status_label"]), # Cập nhật Label Status
-        T["convert_btn"],               # Cập nhật Nút Convert
-        gr.update(label=T["info_accordion"]), # Cập nhật Tiêu đề Accordion
-        T["info_content"]               # Cập nhật Nội dung Accordion
+        new_lang, # Update State
+        T["lang_btn"],
+        gr.update(label=T["menu_label"], choices=[T["menu_pdf"], T["menu_ocr"]], value=T["menu_pdf"]), # Radio
+        T["info_header"], # Accordion Label
+        T["info_content"], # Info Text
+        
+        # PDF UI
+        T["pdf_title"],
+        gr.update(label=T["pdf_input_label"]),
+        T["pdf_btn"],
+        
+        # OCR UI
+        T["ocr_title"],
+        gr.update(label=T["ocr_input_label"]),
+        T["ocr_btn"],
+        gr.update(label=T["ocr_output_text"]),
+        gr.update(label=T["ocr_output_img"])
     )
 
-# --- 4. GIAO DIỆN (UI) ---
+def toggle_tool(menu_choice, lang_code):
+    # Hàm này quyết định ẩn hiện Group nào dựa trên Radio Button
+    T = TRANS[lang_code]
+    if menu_choice == T["menu_pdf"] or menu_choice == TRANS["en"]["menu_pdf"] or menu_choice == TRANS["vi"]["menu_pdf"]:
+        return gr.update(visible=True), gr.update(visible=False)
+    else:
+        return gr.update(visible=False), gr.update(visible=True)
+
+# --- 4. GIAO DIỆN CHÍNH ---
+
 custom_css = """
-#convert_btn {
-    background: linear-gradient(90deg, #4b6cb7 0%, #182848 100%);
-    color: white; border: none; font-weight: bold; font-size: 16px;
-}
-#lang_btn {
-    border: 1px solid #ccc; color: #333;
-}
+.gradio-container {background-color: #f9f9f9}
+#btn_convert {background: linear-gradient(90deg, #4b6cb7, #182848); color: white;}
+#btn_ocr {background: linear-gradient(90deg, #d53369, #daae51); color: white;}
 footer {visibility: hidden}
 """
 
-theme = gr.themes.Soft(primary_hue="blue", secondary_hue="slate")
-
-with gr.Blocks(theme=theme, css=custom_css, title="PDF to Word Pro") as demo:
+with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="Toolkit Pro") as demo:
     
-    # Biến trạng thái lưu ngôn ngữ (Mặc định là 'vi')
     lang_state = gr.State(value="vi")
+    
+    # --- SIDEBAR (THANH BÊN) ---
+    with gr.Sidebar():
+        gr.Markdown("## 🛠️ CONTROL PANEL")
+        
+        # 1. Nút đổi ngôn ngữ
+        btn_lang = gr.Button(TRANS["vi"]["lang_btn"])
+        
+        # 2. Menu chọn công cụ
+        radio_menu = gr.Radio(
+            choices=[TRANS["vi"]["menu_pdf"], TRANS["vi"]["menu_ocr"]],
+            value=TRANS["vi"]["menu_pdf"],
+            label=TRANS["vi"]["menu_label"],
+            interactive=True
+        )
+        
+        gr.HTML("<hr>")
+        
+        # 3. Thông tin tác giả
+        with gr.Accordion(TRANS["vi"]["info_header"], open=True) as info_acc:
+            info_md = gr.Markdown(TRANS["vi"]["info_content"])
 
-    # --- Hàng nút đổi ngôn ngữ (Góc phải) ---
-    with gr.Row():
-        with gr.Column(scale=9): pass # Khoảng trống
-        with gr.Column(scale=1):
-            btn_change_lang = gr.Button(TRANS["vi"]["lang_btn"], elem_id="lang_btn", size="sm")
-
-    # --- Tiêu đề ---
-    header_md = gr.Markdown(TRANS["vi"]["title"])
-    gr.HTML("<hr>")
-
-    # --- Nội dung chính ---
-    with gr.Row():
-        # Input
-        with gr.Column(variant="panel"):
-            input_group_md = gr.Markdown(TRANS["vi"]["input_group"])
-            input_files = gr.File(
-                label=TRANS["vi"]["input_label"],
-                file_types=[".pdf"], file_count="multiple", height=250
-            )
+    # --- MAIN CONTENT AREA ---
+    
+    # === GROUP 1: PDF TO WORD ===
+    with gr.Group(visible=True) as group_pdf:
+        header_pdf = gr.Markdown(TRANS["vi"]["pdf_title"])
+        with gr.Row():
+            with gr.Column():
+                input_pdf = gr.File(label=TRANS["vi"]["pdf_input_label"], file_types=[".pdf"], file_count="multiple", height=200)
+            with gr.Column():
+                output_docx = gr.File(label="Download Result", height=100)
+                status_pdf = gr.Textbox(label="Status", interactive=False)
+        
+        btn_run_pdf = gr.Button(TRANS["vi"]["pdf_btn"], elem_id="btn_convert")
+
+    # === GROUP 2: OCR ===
+    with gr.Group(visible=False) as group_ocr:
+        header_ocr = gr.Markdown(TRANS["vi"]["ocr_title"])
+        with gr.Row():
+            # Cột trái: Input
+            with gr.Column(scale=1):
+                input_img = gr.Image(label=TRANS["vi"]["ocr_input_label"], type="numpy", height=300)
+                btn_run_ocr = gr.Button(TRANS["vi"]["ocr_btn"], elem_id="btn_ocr")
             
-        # Output
-        with gr.Column(variant="panel"):
-            output_group_md = gr.Markdown(TRANS["vi"]["output_group"])
-            output_file = gr.File(label=TRANS["vi"]["output_label"], height=150)
-            status_text = gr.Textbox(label=TRANS["vi"]["status_label"], interactive=False)
-
-    # Nút bấm
-    with gr.Row():
-        convert_btn = gr.Button(TRANS["vi"]["convert_btn"], elem_id="convert_btn", scale=1)
-
-    # --- Footer ---
-    gr.HTML("<br>")
-    with gr.Accordion(label=TRANS["vi"]["info_accordion"], open=False) as info_acc:
-        info_md = gr.Markdown(TRANS["vi"]["info_content"])
-
-    # --- CÁC SỰ KIỆN (EVENTS) ---
-
-    # 1. Sự kiện đổi ngôn ngữ
-    btn_change_lang.click(
-        fn=change_language,
+            # Cột phải: Output
+            with gr.Column(scale=1):
+                output_text = gr.Textbox(label=TRANS["vi"]["ocr_output_text"], lines=10, show_copy_button=True)
+                output_txt_file = gr.File(label="Download .txt")
+                
+        # Hiển thị ảnh kết quả bên dưới
+        output_img_viz = gr.Image(label=TRANS["vi"]["ocr_output_img"], interactive=False)
+
+    # --- EVENT HANDLERS ---
+    
+    # 1. Logic chạy PDF
+    btn_run_pdf.click(
+        fn=convert_pdfs_to_word,
+        inputs=[input_pdf, lang_state],
+        outputs=[output_docx, status_pdf]
+    )
+    
+    # 2. Logic chạy OCR
+    btn_run_ocr.click(
+        fn=run_ocr_func,
+        inputs=[input_img, lang_state],
+        outputs=[output_text, output_img_viz, output_txt_file]
+    )
+    
+    # 3. Logic chuyển đổi Tab (Ẩn hiện Group)
+    radio_menu.change(
+        fn=toggle_tool,
+        inputs=[radio_menu, lang_state],
+        outputs=[group_pdf, group_ocr]
+    )
+    
+    # 4. Logic đổi ngôn ngữ (Cập nhật toàn bộ Label)
+    btn_lang.click(
+        fn=change_ui_language,
         inputs=[lang_state],
         outputs=[
-            lang_state,         # 0. State mới
-            btn_change_lang,    # 1. Nút đổi ngôn ngữ
-            header_md,          # 2. Tiêu đề
-            input_group_md,     # 3. Tiêu đề input
-            input_files,        # 4. Label input file
-            output_group_md,    # 5. Tiêu đề output
-            output_file,        # 6. Label output file
-            status_text,        # 7. Label status
-            convert_btn,        # 8. Nút convert
-            info_acc,           # 9. Tiêu đề Accordion
-            info_md             # 10. Nội dung Accordion
+            lang_state, btn_lang, radio_menu, info_acc, info_md, # Sidebar updates
+            header_pdf, input_pdf, btn_run_pdf, # PDF updates
+            header_ocr, input_img, btn_run_ocr, output_text, output_img_viz # OCR updates
         ]
     )
 
-    # 2. Sự kiện Convert (Truyền thêm lang_state để biết hiện thông báo tiếng gì)
-    convert_btn.click(
-        fn=convert_pdfs_to_word,
-        inputs=[input_files, lang_state], # Input file và Ngôn ngữ hiện tại
-        outputs=[output_file, status_text],
-        show_progress="hidden"
-    )
-
 if __name__ == "__main__":
     demo.launch()