Update app.py

app.py

@@ -14,13 +14,13 @@ ocr_engine = None
 
 try:
     from paddleocr import PaddleOCR
-    # Khởi tạo OCR engine (ẩn log để sạch console)
-    ocr_engine = PaddleOCR(use_textline_orientation=True, lang='vi', show_log=False)
+    # FIX LỖI 1: Bỏ tham số 'show_log' gây lỗi
+    ocr_engine = PaddleOCR(use_textline_orientation=True, lang='vi')
     HAS_OCR = True
 except Exception as e:
     print(f"⚠️ Cảnh báo: Không thể khởi tạo OCR. Lỗi: {e}")
 
-# --- 2. TỪ ĐIỂN NGÔN NGỮ (TEXT RESOURCES) ---
+# --- 2. TỪ ĐIỂN NGÔN NGỮ ---
 TRANS = {
     "vi": {
         "app_name": "PRO DOCUMENT TOOLKIT",
@@ -28,23 +28,17 @@ TRANS = {
         "menu_pdf": "📄 Chuyển PDF sang Word",
         "menu_ocr": "👁️ Trích xuất chữ (OCR)",
         "lang_btn": "🇬🇧 English",
-        
-        # PDF Section
         "pdf_head": "Chuyển đổi Tài liệu",
         "pdf_sub": "Giữ nguyên định dạng gốc, hỗ trợ xử lý hàng loạt.",
         "pdf_label": "Thả file PDF vào đây hoặc click để tải lên",
         "pdf_btn": "🚀 BẮT ĐẦU CHUYỂN ĐỔI",
         "pdf_success": "Đã chuyển đổi thành công!",
-        
-        # OCR Section
         "ocr_head": "Trích xuất Văn bản (AI)",
         "ocr_sub": "Nhận diện tiếng Việt chính xác từ hình ảnh/scan.",
         "ocr_label": "Tải ảnh lên (PNG, JPG, BMP)",
         "ocr_btn": "🔍 PHÂN TÍCH ẢNH",
         "ocr_rs_text": "Văn bản nhận diện",
         "ocr_rs_img": "Vùng nhận diện",
-        
-        # Common
         "err_nofile": "Vui lòng tải file lên trước!",
         "err_ocr": "Chức năng OCR chưa được cài đặt đúng.",
         "footer": "© 2024 Developed by Chu Viet Kien. Powered by AI."
@@ -55,61 +49,46 @@ TRANS = {
         "menu_pdf": "📄 PDF to Word Converter",
         "menu_ocr": "👁️ Smart OCR Extraction",
         "lang_btn": "🇻🇳 Tiếng Việt",
-        
-        # PDF Section
         "pdf_head": "Document Converter",
         "pdf_sub": "Preserve original formatting, batch processing supported.",
         "pdf_label": "Drop PDF files here or click to upload",
         "pdf_btn": "🚀 START CONVERSION",
         "pdf_success": "Conversion completed successfully!",
-        
-        # OCR Section
         "ocr_head": "Text Extraction (AI)",
         "ocr_sub": "High-accuracy Vietnamese text recognition from images.",
         "ocr_label": "Upload Image (PNG, JPG, BMP)",
         "ocr_btn": "🔍 ANALYZE IMAGE",
         "ocr_rs_text": "Extracted Text",
         "ocr_rs_img": "Detected Zones",
-        
-        # Common
         "err_nofile": "Please upload a file first!",
         "err_ocr": "OCR engine is not properly installed.",
         "footer": "© 2024 Developed by Chu Viet Kien. Powered by AI."
     }
 }
 
-# --- 3. LOGIC XỬ LÝ (BACKEND) ---
+# --- 3. LOGIC XỬ LÝ ---
 
 def convert_pdfs_to_word(pdf_files, lang_code, progress=gr.Progress()):
-    """Xử lý chuyển đổi PDF -> Word"""
     T = TRANS[lang_code]
-    
     if not pdf_files:
-        gr.Warning(T["err_nofile"]) # Dùng thông báo nổi thay vì text box
+        gr.Warning(T["err_nofile"])
         return None
     
     if not isinstance(pdf_files, list):
         pdf_files = [pdf_files]
         
     converted_files = []
-    
     try:
-        # Giả lập loading bar
         progress(0, desc="Starting...")
-        
         for idx, pdf_file in enumerate(pdf_files):
             file_name = os.path.basename(pdf_file.name)
             progress((idx / len(pdf_files)), desc=f"Converting: {file_name}")
-            
             docx_name = os.path.splitext(file_name)[0] + ".docx"
             cv = Converter(pdf_file.name)
             cv.convert(docx_name)
             cv.close()
             converted_files.append(docx_name)
             
-        progress(1, desc="Finalizing...")
-        
-        # Thông báo thành công
         gr.Info(f"✅ {T['pdf_success']}")
         
         if len(converted_files) == 1:
@@ -119,27 +98,21 @@ def convert_pdfs_to_word(pdf_files, lang_code, progress=gr.Progress()):
             with zipfile.ZipFile(zip_name, 'w') as zf:
                 for f in converted_files: zf.write(f)
             return zip_name
-
     except Exception as e:
         gr.Error(f"Error: {str(e)}")
         return None
 
 def run_ocr_func(image, lang_code):
-    """Xử lý OCR"""
     T = TRANS[lang_code]
-    
     if not HAS_OCR:
         gr.Error(T["err_ocr"])
         return None, None, None
-        
     if image is None:
         gr.Warning(T["err_nofile"])
         return None, None, None
 
     try:
-        gr.Info("⏳ AI is reading image...")
-        
-        # Chạy OCR
+        gr.Info("⏳ AI is processing...")
         result = ocr_engine.ocr(image, cls=True)
         txts, boxes = [], []
         
@@ -152,37 +125,30 @@ def run_ocr_func(image, lang_code):
         
         full_text = "\n".join(txts)
         
-        # Vẽ box lên ảnh
         img_cv = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
         for box in boxes:
             pts = np.array(box, np.int32).reshape((-1, 1, 2))
             cv2.polylines(img_cv, [pts], True, (0, 0, 255), 2)
         final_img = cv2.cvtColor(img_cv, cv2.COLOR_BGR2RGB)
         
-        # Lưu file txt
         txt_path = "ocr_result.txt"
         with open(txt_path, "w", encoding="utf-8") as f:
             f.write(full_text)
             
         return full_text, final_img, txt_path
-
     except Exception as e:
         gr.Error(f"OCR Error: {str(e)}")
         return None, None, None
 
-# --- 4. HÀM UI HELPER ---
+# --- 4. UI HELPERS ---
 def change_lang(lang):
     new = "en" if lang == "vi" else "vi"
     T = TRANS[new]
     return (
         new, T["lang_btn"],
-        # Sidebar labels
         gr.update(choices=[T["menu_pdf"], T["menu_ocr"]], label="MENU", value=T["menu_pdf"]),
-        # Header text
         T["app_name"], T["app_desc"], T["footer"],
-        # PDF labels
         T["pdf_head"], T["pdf_sub"], gr.update(label=T["pdf_label"]), T["pdf_btn"],
-        # OCR labels
         T["ocr_head"], T["ocr_sub"], gr.update(label=T["ocr_label"]), T["ocr_btn"], 
         gr.update(label=T["ocr_rs_text"]), gr.update(label=T["ocr_rs_img"])
     )
@@ -191,207 +157,93 @@ def toggle_view(menu_val, lang):
     is_pdf = (menu_val == TRANS["vi"]["menu_pdf"] or menu_val == TRANS["en"]["menu_pdf"])
     return (gr.update(visible=True), gr.update(visible=False)) if is_pdf else (gr.update(visible=False), gr.update(visible=True))
 
-# --- 5. STYLE & THEME (CSS CHUYÊN NGHIỆP) ---
-
-# Tùy chỉnh Theme có sẵn của Gradio
+# --- 5. STYLE & THEME ---
 professional_theme = gr.themes.Soft(
-    primary_hue="blue",     # Màu chủ đạo: Xanh chuyên nghiệp
-    neutral_hue="slate",    # Màu nền/chữ: Xám hiện đại
-    radius_size="md",       # Bo góc vừa phải
-    font=[gr.themes.GoogleFont("Inter"), "ui-sans-serif", "system-ui", "sans-serif"],
+    primary_hue="blue",
+    neutral_hue="slate",
+    radius_size="md",
 ).set(
-    body_background_fill="#f8fafc",  # Màu nền tổng thể xám rất nhạt
-    block_background_fill="#ffffff", # Các khối (block) màu trắng
+    body_background_fill="#f8fafc",
+    block_background_fill="#ffffff",
     block_border_width="1px",
-    block_shadow="0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06)", # Đổ bóng nhẹ (Shadow)
-    button_primary_background_fill="linear-gradient(90deg, #2563eb, #1d4ed8)", # Gradient xanh
-    button_primary_background_fill_hover="linear-gradient(90deg, #1d4ed8, #1e40af)",
+    block_shadow="0 4px 6px -1px rgba(0, 0, 0, 0.1)",
+    button_primary_background_fill="linear-gradient(90deg, #2563eb, #1d4ed8)",
     button_primary_text_color="white",
 )
 
-# CSS tùy chỉnh thêm
 custom_css = """
-/* Ẩn footer mặc định của Gradio */
 footer {display: none !important;}
-
-/* Style cho Header */
-.app-header {
-    text-align: center; 
-    margin-bottom: 30px; 
-    padding: 20px;
-}
-.app-title {
-    font-size: 2rem; 
-    font-weight: 800; 
-    color: #1e293b; 
-    margin-bottom: 0.5rem;
-    letter-spacing: -0.025em;
-}
-.app-desc {
-    font-size: 1.1rem; 
-    color: #64748b;
-}
-
-/* Style cho Card tiêu đề từng tool */
-.tool-header {
-    border-left: 4px solid #2563eb; 
-    padding-left: 15px; 
-    margin-bottom: 20px;
-}
-.tool-title {
-    font-size: 1.5rem; 
-    font-weight: 700; 
-    color: #334155;
-}
-.tool-sub {
-    color: #64748b; 
-    font-size: 0.95rem;
-}
-
-/* Tùy chỉnh nút bấm */
-button.primary-btn {
-    font-weight: 600 !important;
-    font-size: 1rem !important;
-    padding: 12px 24px !important;
-    transition: all 0.2s;
-}
-button.primary-btn:hover {
-    transform: translateY(-1px);
-    box-shadow: 0 10px 15px -3px rgba(37, 99, 235, 0.2);
-}
-
-/* Custom Footer riêng */
-.my-footer {
-    text-align: center; 
-    margin-top: 40px; 
-    color: #94a3b8; 
-    font-size: 0.85rem; 
-    border-top: 1px solid #e2e8f0; 
-    padding-top: 20px;
-}
+.app-header {text-align: center; margin-bottom: 30px; padding: 20px;}
+.app-title {font-size: 2rem; font-weight: 800; color: #1e293b; margin-bottom: 0.5rem;}
+.app-desc {font-size: 1.1rem; color: #64748b;}
+.tool-header {border-left: 4px solid #2563eb; padding-left: 15px; margin-bottom: 20px;}
+.tool-title {font-size: 1.5rem; font-weight: 700; color: #334155;}
+.tool-sub {color: #64748b; font-size: 0.95rem;}
+button.primary-btn {font-weight: 600 !important; padding: 12px 24px !important;}
+.my-footer {text-align: center; margin-top: 40px; color: #94a3b8; font-size: 0.85rem; border-top: 1px solid #e2e8f0; padding-top: 20px;}
 """
 
-# --- 6. XÂY DỰNG GIAO DIỆN (UI LAYOUT) ---
-
-with gr.Blocks(theme=professional_theme, css=custom_css, title="Pro Toolkit") as demo:
+# --- 6. GIAO DIỆN ---
+# Lưu ý: Không truyền theme/css vào đây nữa để tránh Warning
+with gr.Blocks(title="Pro Toolkit") as demo:
     
-    # State quản lý ngôn ngữ
     lang_state = gr.State("vi")
     
-    # === SIDEBAR NAV ===
     with gr.Sidebar():
         gr.Markdown("### 🛠️ DASHBOARD")
-        
         btn_lang = gr.Button(TRANS["vi"]["lang_btn"], size="sm", variant="secondary")
-        
         radio_menu = gr.Radio(
             choices=[TRANS["vi"]["menu_pdf"], TRANS["vi"]["menu_ocr"]],
             value=TRANS["vi"]["menu_pdf"],
             label="MENU",
             interactive=True
         )
-        
-        gr.Markdown("---")
-        gr.Markdown("""
-        **System Status:** 🟢 Core: Online  
-        🟢 AI Engine: Ready
-        """)
+        gr.Markdown("---\n**System Status:** 🟢 Online")
 
-    # === MAIN CONTENT ===
     with gr.Column(elem_classes="main-container"):
-        
-        # -- HEADER CHUNG --
         with gr.Column(elem_classes="app-header"):
             txt_title = gr.Markdown(f"<div class='app-title'>{TRANS['vi']['app_name']}</div>")
             txt_desc = gr.Markdown(f"<div class='app-desc'>{TRANS['vi']['app_desc']}</div>")
 
-        # ==========================
-        # TOOL 1: PDF TO WORD
-        # ==========================
+        # TOOL 1: PDF
         with gr.Group(visible=True) as group_pdf:
-            # Header Tool
             with gr.Column(elem_classes="tool-header"):
                 t1_head = gr.Markdown(f"<div class='tool-title'>{TRANS['vi']['pdf_head']}</div>")
                 t1_sub = gr.Markdown(f"<div class='tool-sub'>{TRANS['vi']['pdf_sub']}</div>")
             
-            # Workspace (Chia cột Input / Output)
             with gr.Row(equal_height=True):
-                # Cột trái: Upload
                 with gr.Column(variant="panel"):
-                    in_pdf = gr.File(
-                        label=TRANS["vi"]["pdf_label"], 
-                        file_types=[".pdf"], 
-                        file_count="multiple",
-                        height=250
-                    )
-                    btn_pdf = gr.Button(
-                        TRANS["vi"]["pdf_btn"], 
-                        variant="primary", 
-                        elem_classes="primary-btn"
-                    )
-                
-                # Cột phải: Kết quả
+                    in_pdf = gr.File(label=TRANS["vi"]["pdf_label"], file_types=[".pdf"], file_count="multiple", height=250)
+                    btn_pdf = gr.Button(TRANS["vi"]["pdf_btn"], variant="primary", elem_classes="primary-btn")
                 with gr.Column(variant="panel"):
-                    gr.Markdown("### 📂 Kết quả tải về")
+                    gr.Markdown("### 📂 Kết quả")
                     out_word = gr.File(label="Download", height=150)
-                    # Dùng Markdown để hiển thị hướng dẫn
-                    gr.Markdown("*File sẽ tự động nén ZIP nếu có nhiều tài liệu.*")
 
-        # ==========================
         # TOOL 2: OCR
-        # ==========================
         with gr.Group(visible=False) as group_ocr:
             with gr.Column(elem_classes="tool-header"):
                 t2_head = gr.Markdown(f"<div class='tool-title'>{TRANS['vi']['ocr_head']}</div>")
                 t2_sub = gr.Markdown(f"<div class='tool-sub'>{TRANS['vi']['ocr_sub']}</div>")
 
             with gr.Row():
-                # Cột trái: Ảnh input + Nút
                 with gr.Column(scale=4, variant="panel"):
-                    in_img = gr.Image(
-                        label=TRANS["vi"]["ocr_label"], 
-                        type="numpy", 
-                        height=400,
-                        sources=["upload", "clipboard"]
-                    )
-                    btn_ocr = gr.Button(
-                        TRANS["vi"]["ocr_btn"], 
-                        variant="primary", 
-                        elem_classes="primary-btn"
-                    )
-
-                # Cột phải: Kết quả (Tab view)
+                    in_img = gr.Image(label=TRANS["vi"]["ocr_label"], type="numpy", height=400)
+                    btn_ocr = gr.Button(TRANS["vi"]["ocr_btn"], variant="primary", elem_classes="primary-btn")
                 with gr.Column(scale=6, variant="panel"):
                     with gr.Tabs():
                         with gr.TabItem("📝 Text"):
-                            out_txt = gr.Textbox(
-                                label=TRANS["vi"]["ocr_rs_text"], 
-                                lines=14, 
-                                show_copy_button=True # Gradio mới đã hỗ trợ, nếu lỗi thì xóa dòng này
-                            )
+                            # FIX LỖI 2: Xóa 'show_copy_button=True' gây crash
+                            out_txt = gr.Textbox(label=TRANS["vi"]["ocr_rs_text"], lines=14)
                             out_file = gr.File(label="Download .txt")
-                        
                         with gr.TabItem("🖼️ Visualize"):
-                            out_img_viz = gr.Image(
-                                label=TRANS["vi"]["ocr_rs_img"], 
-                                interactive=False
-                            )
+                            out_img_viz = gr.Image(label=TRANS["vi"]["ocr_rs_img"], interactive=False)
 
-        # -- FOOTER --
         txt_footer = gr.Markdown(f"<div class='my-footer'>{TRANS['vi']['footer']}</div>")
 
-    # === SỰ KIỆN (EVENTS) ===
-    
-    # 1. Menu Navigation
+    # EVENTS
     radio_menu.change(toggle_view, [radio_menu, lang_state], [group_pdf, group_ocr])
-    
-    # 2. Chức năng PDF
     btn_pdf.click(convert_pdfs_to_word, [in_pdf, lang_state], out_word)
-    
-    # 3. Chức năng OCR
     btn_ocr.click(run_ocr_func, [in_img, lang_state], [out_txt, out_img_viz, out_file])
-    
-    # 4. Đổi ngôn ngữ
     btn_lang.click(change_lang, [lang_state], [
         lang_state, btn_lang, radio_menu,
         txt_title, txt_desc, txt_footer,
@@ -399,7 +251,12 @@ with gr.Blocks(theme=professional_theme, css=custom_css, title="Pro Toolkit") as
         t2_head, t2_sub, in_img, btn_ocr, out_txt, out_img_viz
     ])
 
-# Chạy App
 if __name__ == "__main__":
-    demo.queue() # Bật hàng đợi để xử lý mượt mà
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.queue()
+    # FIX LỖI 3: Chuyển theme/css xuống đây để tương thích Gradio mới nhất
+    demo.launch(
+        server_name="0.0.0.0", 
+        server_port=7860,
+        theme=professional_theme, 
+        css=custom_css
+    )