back to old

app.py

@@ -1,41 +1,65 @@
+import os
+import subprocess
 import gradio as gr
 from pptx import Presentation
-from fpdf import FPDF
-import tempfile
-import os
-from PIL import Image
+from pptx.util import Inches
+from pdf2image import convert_from_path
+
+
+subprocess.run(['bash', 'setup.sh'], check=True)
+
+
+def pptx_to_pdf(input_pptx, output_pdf):
+    subprocess.run(['unoconv', '-f', 'pdf', '-o', output_pdf, input_pptx], check=True)
+
 
-def pptx_to_pdf(pptx_file):
-    # Create a temporary directory to store intermediate images
-    with tempfile.TemporaryDirectory() as tmpdirname:
-        pdf = FPDF()
+def pdf_to_pptx(input_pdf, output_pptx):
+    # Specify a directory to save images
+    output_dir = "pdf_images"
+    os.makedirs(output_dir, exist_ok=True)
+    
+    # Convert PDF to images and save in the specified directory
+    images = convert_from_path(input_pdf, output_folder=output_dir)
+    
+    # Create a new PowerPoint presentation
+    presentation = Presentation()
+    
+    for i, img in enumerate(images):
+        img_path = os.path.join(output_dir, f"page_{i+1}.png")
+        img.save(img_path, 'PNG')
         
-        # Load the PowerPoint file
-        presentation = Presentation(pptx_file.name)
+        slide_layout = presentation.slide_layouts[5]  # Use a blank layout
+        slide = presentation.slides.add_slide(slide_layout)
+        left = top = Inches(0)
+        pic = slide.shapes.add_picture(img_path, left, top, width=Inches(10), height=Inches(7.5))
         
-        for i, slide in enumerate(presentation.slides):
-            # Save each slide as an image
-            img_path = os.path.join(tmpdirname, f"slide_{i}.png")
-            slide.shapes[0].image.save(img_path)
-
-            # Add the image to the PDF
-            pdf.add_page()
-            pdf.image(img_path, 0, 0, 210, 297)  # A4 size (210mm x 297mm)
-
-        # Save the PDF to a file
-        pdf_output_path = os.path.join(tmpdirname, "output.pdf")
-        pdf.output(pdf_output_path)
-
-        return pdf_output_path
-
-# Create a Gradio interface
-iface = gr.Interface(
-    fn=pptx_to_pdf,
-    inputs=gr.File(file_types=[".pptx"]),
-    outputs=gr.File(file_types=[".pdf"]),
-    title="PPTX to PDF Converter",
-    description="Upload a PowerPoint file to convert it to a PDF without watermarks."
+    presentation.save(output_pptx)
+
+def convert_pptx_to_pdf(pptx_file):
+    output_pdf = "output.pdf"
+    pptx_to_pdf(pptx_file.name, output_pdf)
+    return output_pdf
+
+def convert_pdf_to_pptx(pdf_file):
+    output_pptx = "output_converted.pptx"
+    pdf_to_pptx(pdf_file.name, output_pptx)
+    return output_pptx
+
+pptx_to_pdf_interface = gr.Interface(
+    fn=convert_pptx_to_pdf,
+    inputs=gr.File(label="Upload PPTX File", file_count="single"),
+    outputs=gr.File(label="Download PDF File"),
+    title="Convert PPTX to PDF"
 )
 
-# Launch the app
-iface.launch()
+pdf_to_pptx_interface = gr.Interface(
+    fn=convert_pdf_to_pptx,
+    inputs=gr.File(label="Upload PDF File", file_count="single"),
+    outputs=gr.File(label="Download PPTX File"),
+    title="Convert PDF to PPTX"
+)
+
+app = gr.TabbedInterface([pptx_to_pdf_interface, pdf_to_pptx_interface], ["PPTX to PDF", "PDF to PPTX"])
+
+if __name__ == "__main__":
+    app.launch()