import gradio as gr
from pdf2docx import Converter
import os
from pathlib import Path

# Hàm chuyển đổi PDF sang Word
def pdf_to_word(pdf_file):
    try:
        # Đường dẫn tạm thời cho file đầu ra
        output_file = Path(pdf_file).stem + ".docx"
        
        # Chuyển đổi PDF sang Word
        cv = Converter(pdf_file)
        cv.convert(output_file, start=0, end=None)
        cv.close()
        
        # Trả về file Word đã chuyển đổi
        return output_file
    except Exception as e:
        return f"Lỗi: {str(e)}"

# Tạo giao diện Gradio
interface = gr.Interface(
    fn=pdf_to_word,                  # Hàm xử lý
    inputs=gr.File(label="Tải lên file PDF"),  # Input: File PDF
    outputs=gr.File(label="Tải xuống file Word"),  # Output: File Word
    title="Chuyển PDF sang Word",
    description="Tải lên file PDF để chuyển đổi sang định dạng Word (.docx) giữ nguyên định dạng.",
    theme="default"
)

# Chạy ứng dụng
interface.launch()