import streamlit as st
from pdf2image import convert_from_bytes
from PIL import Image
import io
import fitz  # PyMuPDF for PDF handling
import docx
from fpdf import FPDF
import os
os.system("apt-get update && apt-get install -y poppler-utils")

st.set_page_config(page_title="PDF & Image Converter", page_icon="📄", layout="centered")
st.title("📄 PDF & Image Converter")
st.markdown("Convert PDFs to images, Word, and vice versa.")

# File uploader
uploaded_file = st.file_uploader("Upload a file (PDF, PNG, JPG, JPEG, DOCX)", type=["pdf", "png", "jpg", "jpeg", "docx"])

if uploaded_file:
    file_type = uploaded_file.type

    # ---------------- PDF TO IMAGE ----------------
    if file_type == "application/pdf":
        st.subheader("Convert PDF to Image")
        images = convert_from_bytes(uploaded_file.read())

        format_option = st.selectbox("Select image format", ["PNG", "JPEG", "JPG"])
        quality = st.slider("Image Quality", 50, 100, 90)

        zip_buffer = io.BytesIO()
        with st.spinner("Converting PDF to images..."):
            for i, img in enumerate(images):
                img_format = format_option.lower()
                img_byte_arr = io.BytesIO()
                img.save(img_byte_arr, format=img_format.upper(), quality=quality)
                st.image(img, caption=f"Page {i+1}", use_column_width=True)
                st.download_button(f"Download Page {i+1}", img_byte_arr.getvalue(), file_name=f"page_{i+1}.{img_format}")

    # ---------------- PDF TO WORD ----------------
        st.subheader("Convert PDF to Word")
        if st.button("Convert to DOCX"):
            doc = docx.Document()
            pdf_reader = fitz.open(stream=uploaded_file.read(), filetype="pdf")
            for page in pdf_reader:
                text = page.get_text()
                doc.add_paragraph(text)
            word_buffer = io.BytesIO()
            doc.save(word_buffer)
            st.download_button("Download DOCX", word_buffer.getvalue(), file_name="converted.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")

    # ---------------- IMAGE TO PDF ----------------
    elif file_type in ["image/png", "image/jpeg", "image/jpg"]:
        st.subheader("Convert Image to PDF")
        image = Image.open(uploaded_file)
        img_pdf_buffer = io.BytesIO()
        image.convert("RGB").save(img_pdf_buffer, format="PDF")
        st.download_button("Download PDF", img_pdf_buffer.getvalue(), file_name="converted.pdf", mime="application/pdf")

    # ---------------- WORD TO PDF ----------------
    elif file_type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
        st.subheader("Convert Word to PDF")
        if st.button("Convert to PDF"):
            doc = docx.Document(uploaded_file)
            pdf = FPDF()
            pdf.set_auto_page_break(auto=True, margin=10)
            pdf.add_page()
            pdf.set_font("Arial", size=12)
            for para in doc.paragraphs:
                pdf.cell(200, 10, txt=para.text, ln=True)
            pdf_buffer = io.BytesIO()
            pdf.output(pdf_buffer, "F")
            st.download_button("Download PDF", pdf_buffer.getvalue(), file_name="converted.pdf", mime="application/pdf")

st.markdown("---")
st.markdown("💡 Developed with ❤️ using Streamlit, PyMuPDF, pdf2image, and PIL")