Update app.py

app.py

@@ -1,50 +1,77 @@
 import gradio as gr
 from pdf2docx import Converter
 import os
+import zipfile
 import time
 
-# --- Hàm xử lý chính (có thêm thanh tiến trình) ---
-def convert_pdf_to_word(pdf_file, progress=gr.Progress()):
-    if pdf_file is None:
-        return None, "⚠️ Vui lòng tải file lên trước!"
+# --- Hàm xử lý chính (Hỗ trợ Batch Processing) ---
+def convert_pdfs_to_word(pdf_files, progress=gr.Progress()):
+    # Kiểm tra đầu vào
+    if not pdf_files:
+        return None, "⚠️ Vui lòng tải ít nhất một file lên!"
     
-    # 1. Định nghĩa tên file đầu ra
-    # Lấy tên gốc của file pdf để đặt tên cho file docx cho chuyên nghiệp
-    original_name = os.path.basename(pdf_file.name)
-    docx_filename = os.path.splitext(original_name)[0] + "_converted.docx"
+    # Nếu pdf_files không phải là list (trường hợp lỗi hiếm gặp), đưa nó vào list
+    if not isinstance(pdf_files, list):
+        pdf_files = [pdf_files]
+
+    converted_files = []
+    total_files = len(pdf_files)
     
     try:
-        progress(0.1, desc="Đang khởi tạo...")
-        # 2. Khởi tạo converter
-        cv = Converter(pdf_file.name)
-        
-        progress(0.3, desc="Đang phân tích layout...")
-        # 3. Thực hiện chuyển đổi
-        cv.convert(docx_filename, start=0, end=None)
-        
-        progress(0.8, desc="Đang hoàn tất...")
-        cv.close()
+        # Vòng lặp xử lý từng file
+        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"
+            
+            # Cập nhật thanh tiến trình
+            msg = f"Đang xử lý file {index + 1}/{total_files}: {original_name}"
+            progress((index / total_files), desc=msg)
+            
+            # Thực hiện convert
+            cv = Converter(pdf_file.name)
+            cv.convert(docx_filename, start=0, end=None)
+            cv.close()
+            
+            converted_files.append(docx_filename)
+
+        # --- Giai đoạn đóng gói kết quả ---
+        progress(0.95, desc="Đang đóng gói file...")
         
+        # Trường hợp 1: Chỉ có 1 file -> Trả về file docx
+        if len(converted_files) == 1:
+            final_output = converted_files[0]
+            status_msg = f"✅ Hoàn tất! Đã chuyển đổi: {os.path.basename(final_output)}"
+            
+        # Trường hợp 2: Có nhiều file -> Nén thành ZIP
+        else:
+            zip_filename = "All_Converted_Documents.zip"
+            with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
+                for file in converted_files:
+                    # arcname là tên file bên trong file zip (bỏ đường dẫn dài dòng)
+                    zipf.write(file, arcname=os.path.basename(file))
+            
+            final_output = zip_filename
+            status_msg = f"✅ Hoàn tất! Đã nén {len(converted_files)} file vào ZIP."
+
         progress(1.0, desc="Xong!")
-        return docx_filename, f"✅ Chuyển đổi thành công: {docx_filename}"
-        
+        return final_output, status_msg
+
     except Exception as e:
         return None, f"❌ Có lỗi xảy ra: {str(e)}"
 
 # --- Cấu hình giao diện (UI) ---
 
-# Tùy chỉnh CSS để giao diện đẹp hơn
 custom_css = """
 #convert_btn {
     background: linear-gradient(90deg, #4b6cb7 0%, #182848 100%);
     color: white;
     border: none;
     font-weight: bold;
+    font-size: 16px;
 }
-footer {visibility: hidden} /* Ẩn footer mặc định của Gradio */
+footer {visibility: hidden}
 """
 
-# Sử dụng Theme Soft để giao diện mềm mại
 theme = gr.themes.Soft(
     primary_hue="blue",
     secondary_hue="slate",
@@ -55,46 +82,44 @@ theme = gr.themes.Soft(
 
 with gr.Blocks(theme=theme, css=custom_css, title="PDF to Word Pro") as demo:
     
-    # --- Phần Tiêu đề ---
+    # --- Header ---
     with gr.Row():
-        with gr.Column(scale=1):
-            pass # Cột trống để căn lề
+        with gr.Column(scale=1): pass
         with gr.Column(scale=8):
             gr.Markdown(
                 """
-                # 📝 PDF TO WORD CONVERTER
-                ### Chuyên dụng cho Đề thi Tiếng Anh & Tài liệu Học tập
+                # 📝 PDF TO WORD CONVERTER PRO
+                ### Chuyên dụng cho Đề thi Tiếng Anh & Tài liệu Học tập (Hỗ trợ nhiều file)
                 """
             )
-        with gr.Column(scale=1):
-            pass
+        with gr.Column(scale=1): pass
 
-    gr.HTML("<hr>") # Đường kẻ ngang
+    gr.HTML("<hr>")
 
-    # --- Phần Chức năng chính ---
+    # --- Main Content ---
     with gr.Row():
-        # Cột bên trái: Upload
+        # Cột Input
         with gr.Column(variant="panel"):
             gr.Markdown("### 1. Tải tài liệu")
-            input_file = gr.File(
-                label="Upload file PDF",
+            # file_count="multiple" cho phép chọn nhiều file cùng lúc
+            input_files = gr.File(
+                label="Chọn các file PDF (Giữ Ctrl để chọn nhiều)",
                 file_types=[".pdf"],
-                file_count="single",
+                file_count="multiple", 
                 height=250
             )
             
-        # Nút bấm ở giữa (hoặc dưới input)
-        # Cột bên phải: Kết quả
+        # Cột Output
         with gr.Column(variant="panel"):
             gr.Markdown("### 2. Tải về kết quả")
-            output_file = gr.File(label="File Word đã chuyển đổi", height=150)
+            output_file = gr.File(label="File Word hoặc ZIP", height=150)
             status_text = gr.Textbox(label="Trạng thái", interactive=False)
 
-    # Nút bấm to nằm riêng một hàng
+    # Nút Action
     with gr.Row():
         btn = gr.Button("🚀 CHUYỂN ĐỔI NGAY", elem_id="convert_btn", scale=1)
 
-    # --- Phần thông tin tác giả & Bản quyền ---
+    # --- Footer Info ---
     gr.HTML("<br><br>")
     
     with gr.Accordion("ℹ️ Thông tin phát triển & Bản quyền", open=False):
@@ -103,22 +128,22 @@ with gr.Blocks(theme=theme, css=custom_css, title="PDF to Word Pro") as demo:
             ### 👨‍💻 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.0.0
+            * **Phiên bản:** 1.1.0 (Batch Support)
             
             ### 📜 Bản quyền & Miễn trừ trách nhiệm
             * Công cụ này sử dụng thư viện mã nguồn mở `pdf2docx`.
+            * Nếu bạn tải lên nhiều file, hệ thống sẽ tự động nén thành file `.zip`.
             * Chúng tôi không lưu trữ tài liệu của bạn. File sẽ tự động xóa sau khi phiên làm việc kết thúc.
             * © 2024 Chu Viet Kien. All rights reserved.
             """
         )
 
-    # --- Kết nối sự kiện ---
+    # --- Events ---
     btn.click(
-        fn=convert_pdf_to_word, 
-        inputs=[input_file], 
+        fn=convert_pdfs_to_word, 
+        inputs=[input_files], 
         outputs=[output_file, status_text]
     )
 
-# Chạy ứng dụng
 if __name__ == "__main__":
     demo.launch()