import os
import threading
from pdf2docx import Converter
import time
import logging

# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

def convert_pdf_to_docx(pdf_path, docx_path, progress_callback=None):
    """
    將單個PDF文件轉(zhuǎn)換為Word文件。

    Args:
        pdf_path: PDF文件路徑。
        docx_path: Word文件路徑。
        progress_callback: 可選的回調(diào)函數(shù)，用于更新進度條。
    """
    try:
        cv = Converter(pdf_path)
        cv.convert(docx_path, start=0, end=None)  # 可以指定轉(zhuǎn)換的頁碼范圍
        cv.close()
        logging.info(f"Successfully converted {pdf_path} to {docx_path}")
        if progress_callback:
            progress_callback(1)  # 假設(shè)每個文件完成時進度增加1
    except Exception as e:
        logging.error(f"Error converting {pdf_path}: {e}")

def batch_convert(pdf_files, output_dir, progress_callback=None):
    """
    批量將PDF文件轉(zhuǎn)換為Word文件。

    Args:
        pdf_files: PDF文件列表。
        output_dir: 輸出目錄。
        progress_callback: 可選的回調(diào)函數(shù)，用于更新進度條。
    """
    total_files = len(pdf_files)
    converted_count = 0
    threads = []

    for pdf_file in pdf_files:
        pdf_filename = os.path.basename(pdf_file)
        docx_filename = os.path.splitext(pdf_filename)[0] + ".docx"
        docx_path = os.path.join(output_dir, docx_filename)

        # 創(chuàng)建線程進行轉(zhuǎn)換
        thread = threading.Thread(target=convert_pdf_to_docx, args=(pdf_file, docx_path, progress_callback))
        threads.append(thread)
        thread.start()

    # 等待所有線程完成
    for thread in threads:
        thread.join()

    logging.info("Batch conversion completed.")

# 示例用法 (需要替換為實際的文件列表和目錄)
if __name__ == '__main__':
    pdf_files = ["path/to/file1.pdf", "path/to/file2.pdf"]  # 替換為你的PDF文件列表
    output_dir = "path/to/output"  # 替換為你的輸出目錄

    # 創(chuàng)建輸出目錄
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    # 模擬進度條回調(diào)函數(shù)
    def update_progress(increment):
        global converted_count
        converted_count += increment
        print(f"Progress: {converted_count}/{len(pdf_files)}")

    start_time = time.time()
    batch_convert(pdf_files, output_dir, update_progress)
    end_time = time.time()

    print(f"Total time taken: {end_time - start_time:.2f} seconds")

import tkinter as tk
from tkinter import filedialog, messagebox
import os
import threading
from pdf2docx import Converter
import time
import logging

# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

def convert_pdf_to_docx(pdf_path, docx_path, progress_callback=None):
    """
    將單個PDF文件轉(zhuǎn)換為Word文件。

    Args:
        pdf_path: PDF文件路徑。
        docx_path: Word文件路徑。
        progress_callback: 可選的回調(diào)函數(shù)，用于更新進度條。
    """
    try:
        cv = Converter(pdf_path)
        cv.convert(docx_path, start=0, end=None)  # 可以指定轉(zhuǎn)換的頁碼范圍
        cv.close()
        logging.info(f"Successfully converted {pdf_path} to {docx_path}")
        if progress_callback:
            progress_callback(1)  # 假設(shè)每個文件完成時進度增加1
    except Exception as e:
        logging.error(f"Error converting {pdf_path}: {e}")
        return False
    return True

def batch_convert(pdf_files, output_dir, progress_callback=None):
    """
    批量將PDF文件轉(zhuǎn)換為Word文件。

    Args:
        pdf_files: PDF文件列表。
        output_dir: 輸出目錄。
        progress_callback: 可選的回調(diào)函數(shù)，用于更新進度條。
    """
    total_files = len(pdf_files)
    converted_count = 0
    threads = []
    success_count = 0

    for pdf_file in pdf_files:
        pdf_filename = os.path.basename(pdf_file)
        docx_filename = os.path.splitext(pdf_filename)[0] + ".docx"
        docx_path = os.path.join(output_dir, docx_filename)

        # 創(chuàng)建線程進行轉(zhuǎn)換
        thread = threading.Thread(target=convert_pdf_to_docx_wrapper, args=(pdf_file, docx_path, progress_callback))
        threads.append(thread)
        thread.start()

    # 等待所有線程完成
    for thread in threads:
        thread.join()

    logging.info("Batch conversion completed.")
    return success_count

def convert_pdf_to_docx_wrapper(pdf_path, docx_path, progress_callback):
    global success_count
    if convert_pdf_to_docx(pdf_path, docx_path, progress_callback):
        success_count += 1

def select_pdf_files():
    global pdf_files
    pdf_files = filedialog.askopenfilenames(filetypes=[("PDF files", "*.pdf")])
    pdf_listbox.delete(0, tk.END)
    for file in pdf_files:
        pdf_listbox.insert(tk.END, os.path.basename(file))

def select_output_dir():
    global output_dir
    output_dir = filedialog.askdirectory()
    output_dir_label.config(text=f"Output Directory: {output_dir}")

def start_conversion():
    global pdf_files, output_dir, success_count
    if not pdf_files:
        messagebox.showerror("Error", "Please select PDF files.")
        return
    if not output_dir:
        messagebox.showerror("Error", "Please select an output directory.")
        return

    total_files = len(pdf_files)
    progress_var.set(0)
    progress_bar["maximum"] = total_files
    success_count = 0

    def update_progress(increment):
        progress_var.set(progress_var.get() + increment)
        root.update_idletasks()  # 強制更新GUI

    def conversion_thread():
        start_time = time.time()
        successful_conversions = batch_convert(pdf_files, output_dir, update_progress)
        end_time = time.time()
        messagebox.showinfo("Info", f"Conversion completed.  Successfully converted {successful_conversions} out of {total_files} files.  Total time taken: {end_time - start_time:.2f} seconds")

    # 啟動后臺線程
    threading.Thread(target=conversion_thread).start()

# GUI setup
root = tk.Tk()
root.title("PDF to Word Converter")

pdf_files = []
output_dir = ""
success_count = 0

# PDF file selection
pdf_button = tk.Button(root, text="Select PDF Files", command=select_pdf_files)
pdf_button.pack(pady=10)

pdf_listbox = tk.Listbox(root, width=50)
pdf_listbox.pack()

# Output directory selection
output_button = tk.Button(root, text="Select Output Directory", command=select_output_dir)
output_button.pack(pady=10)

output_dir_label = tk.Label(root, text="Output Directory: ")
output_dir_label.pack()

# Progress bar
progress_var = tk.IntVar()
progress_bar = tk.Scale(root, variable=progress_var, orient=tk.HORIZONTAL, length=300, showvalue=False)
progress_bar.pack(pady=10)

# Start conversion button
convert_button = tk.Button(root, text="Start Conversion", command=start_conversion)
convert_button.pack(pady=10)

root.mainloop()