Update app.py

app.py

@@ -1,17 +1,17 @@
 import os
-import datetime
-from PIL import Image
-from fpdf import FPDF
 import gradio as gr
-from PyPDF2 import PdfReader, PdfWriter
+from pdf_to_svg import convert_pdf_to_svg
+from svg_editor import apply_transformations
+from svg_to_pdf import convert_svg_to_pdf
+import datetime
+import tempfile
 
 # Folder paths
 INPUT_FOLDER = "./input/"
 OUTPUT_FOLDER = "./output/"
 LOG_FOLDER = "./logs/"
-PROCESSED_IMAGE = os.path.join(OUTPUT_FOLDER, "processed_image.png")
-OUTPUT_PDF = os.path.join(OUTPUT_FOLDER, "processed_file.pdf")
-LOG_FILE = os.path.join(LOG_FOLDER, "conversion_log.txt")
+SVG_OUTPUT = os.path.join(OUTPUT_FOLDER, "processed_diagram.svg")
+PDF_OUTPUT = os.path.join(OUTPUT_FOLDER, "processed_diagram.pdf")
 
 # Ensure required folders exist
 os.makedirs(INPUT_FOLDER, exist_ok=True)
@@ -19,89 +19,62 @@ os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 os.makedirs(LOG_FOLDER, exist_ok=True)
 
 # Function to log conversion steps
-def log_conversion(status, input_file, output_file):
-    with open(LOG_FILE, "a") as log_file:
+def log_conversion(status, input_pdf, output_svg, output_pdf):
+    log_path = os.path.join(LOG_FOLDER, "conversion_log.txt")
+    with open(log_path, "a") as log_file:
         log_file.write(f"{datetime.datetime.now()} - Status: {status}\n")
-        log_file.write(f"Input File: {input_file}\n")
-        log_file.write(f"Output File: {output_file}\n\n")
+        log_file.write(f"Input PDF: {input_pdf}\n")
+        log_file.write(f"Output SVG: {output_svg}\n")
+        log_file.write(f"Output PDF: {output_pdf}\n\n")
 
-# Processing function for PDF
-def process_pdf(uploaded_pdf, width_inches, height_inches, dpi=300):
+# Gradio interface function
+def process_pdf(uploaded_file, brightness, text_scale, line_scale, width, height):
     try:
-        # Save the uploaded PDF temporarily
-        input_pdf_path = os.path.join(INPUT_FOLDER, "uploaded_file.pdf")
-        with open(input_pdf_path, "wb") as f:
-            f.write(uploaded_pdf.read())
-
-        # Read the PDF
-        reader = PdfReader(input_pdf_path)
-        writer = PdfWriter()
+        # Save uploaded file
+        input_pdf = os.path.join(INPUT_FOLDER, "uploaded_file.pdf")
+        with open(input_pdf, "wb") as f:
+            f.write(uploaded_file.read())
 
-        for page_num, page in enumerate(reader.pages):
-            # Extract the image from the PDF page
-            if "/XObject" in page["/Resources"]:
-                xObject = page["/Resources"]["/XObject"].get_object()
-                for obj in xObject:
-                    if xObject[obj]["/Subtype"] == "/Image":
-                        size = (xObject[obj]["/Width"], xObject[obj]["/Height"])
-                        data = xObject[obj].get_data()
-                        img = Image.frombytes("RGB", size, data)
+        # Step 1: Convert PDF to SVG
+        convert_pdf_to_svg(input_pdf, SVG_OUTPUT, width, height)
 
-                        # Convert the image to grayscale and resize
-                        grayscale_image = img.convert("L")
-                        width_pixels = int(width_inches * dpi)
-                        height_pixels = int(height_inches * dpi)
-                        resized_image = grayscale_image.resize((width_pixels, height_pixels))
+        # Step 2: Apply transformations to the SVG
+        apply_transformations(SVG_OUTPUT, brightness, text_scale, line_scale)
 
-                        # Save processed image temporarily
-                        processed_image_path = os.path.join(OUTPUT_FOLDER, f"processed_page_{page_num + 1}.png")
-                        resized_image.save(processed_image_path)
+        # Step 3: Convert SVG to PDF
+        convert_svg_to_pdf(SVG_OUTPUT, PDF_OUTPUT)
 
-                        # Add the processed image to the new PDF
-                        pdf = FPDF(unit="in", format=(width_inches, height_inches))
-                        pdf.add_page()
-                        pdf.image(processed_image_path, x=0, y=0, w=width_inches, h=height_inches)
+        # Step 4: Log the process
+        log_conversion("Success", input_pdf, SVG_OUTPUT, PDF_OUTPUT)
 
-                        # Save the final PDF
-                        pdf.output(OUTPUT_PDF)
-
-        # Log success
-        log_conversion("Success", input_pdf_path, OUTPUT_PDF)
-
-        # Return processed PDF and log file
-        return OUTPUT_PDF, LOG_FILE
+        # Return output files for download
+        return (SVG_OUTPUT, PDF_OUTPUT)
 
     except Exception as e:
-        log_conversion("Failed", "Uploaded PDF", "Processing Error")
-        return None, f"Error: {str(e)}"
-
-# Gradio Interface
-def gradio_app():
-    with gr.Blocks() as demo:
-        gr.Markdown("# PDF Grayscale Converter and Resizer")
-
-        with gr.Row():
-            pdf_input = gr.File(type="file", label="Upload PDF File")
-            with gr.Column():
-                width_input = gr.Number(value=8.5, label="Width (in inches)")
-                height_input = gr.Number(value=11.0, label="Height (in inches)")
-
-        process_button = gr.Button("Process PDF")
-
-        with gr.Row():
-            pdf_output = gr.File(label="Download Processed PDF")
-            log_file_output = gr.File(label="Download Log File")
-
-        # Define the button click functionality
-        process_button.click(
-            process_pdf,
-            inputs=[pdf_input, width_input, height_input],
-            outputs=[pdf_output, log_file_output],
-        )
-
-    return demo
-
-# Launch the Gradio app
-if __name__ == "__main__":
-    app = gradio_app()
-    app.launch()
+        log_conversion("Failed", input_pdf, SVG_OUTPUT, PDF_OUTPUT)
+        return str(e)
+
+# Gradio interface setup
+inputs = [
+    gr.File(label="Upload your PDF file", type="file"),
+    gr.Slider(minimum=0.5, maximum=5.0, step=0.5, default=1.0, label="Brightness (0.5 to 5.0)"),
+    gr.Slider(minimum=0.5, maximum=2.0, step=0.1, default=1.0, label="Text Size Scale (0.5 to 2.0)"),
+    gr.Slider(minimum=0.5, maximum=5.0, step=0.5, default=1.0, label="Line Thickness Scale (0.5 to 5.0)"),
+    gr.Number(value=8.0, label="Width (in inches)", precision=0.1),
+    gr.Number(value=11.0, label="Height (in inches)", precision=0.1)
+]
+
+outputs = [
+    gr.File(label="Download SVG", type="file"),
+    gr.File(label="Download PDF", type="file")
+]
+
+# Launch Gradio interface
+gr.Interface(
+    fn=process_pdf,
+    inputs=inputs,
+    outputs=outputs,
+    live=True,
+    title="PDF to Editable SVG and PDF Converter",
+    description="Upload a PDF to convert it to both an editable SVG and a PDF with transformations applied."
+).launch()