Update app.py

app.py

@@ -1,46 +1,57 @@
 import fitz  # PyMuPDF for PDF manipulation
+import cairosvg  # For precise SVG rendering
 import gradio as gr
 import tempfile
-import os
+import re
 
-# Step 1: Convert PDF page directly to high-quality SVG preserving vector fidelity
-def pdf_to_svg(pdf_file, thickness_scale=1.0):
+# Convert inches to pixels for dimension scaling
+def inches_to_pixels(inches, dpi=300):
+    return int(inches * dpi)
+
+# Convert PDF to SVG with specified dimensions and thickness scaling
+def pdf_to_svg(pdf_file, width_in, height_in, thickness_scale=1.0):
     try:
         # Open the PDF and load the first page
         pdf_document = fitz.open(pdf_file.name)
         page = pdf_document.load_page(0)
         
-        # Use an even larger matrix scale for improved clarity, attempting to retain vector quality
-        svg_data = page.get_svg_image(matrix=fitz.Matrix(8, 8))  # 8x scaling for high detail
+        # Render PDF page as SVG data using a high-resolution matrix
+        svg_data = page.get_svg_image(matrix=fitz.Matrix(8, 8))  # 8x for high fidelity
         
-        # Check if SVG data was extracted
+        # Ensure we have valid SVG data
         if not svg_data:
             print("Error: SVG data extraction failed.")
             return None
         
-        # Create a temporary SVG file and write the SVG data into it
-        temp_svg_path = tempfile.mktemp(suffix=".svg", prefix="enhanced_diagram_")
-        
-        # Apply thickness adjustment directly if needed
+        # Adjust stroke width in the SVG data if thickness scaling is applied
         if thickness_scale != 1.0:
-            import re
             svg_data = re.sub(
                 r'stroke-width="([\d.]+)"',
                 lambda match: f'stroke-width="{float(match.group(1)) * thickness_scale}"',
                 svg_data
             )
         
+        # Convert width and height to pixels at 300 DPI for SVG scaling
+        width_px = inches_to_pixels(width_in)
+        height_px = inches_to_pixels(height_in)
+        
+        # Embed dimensions into the SVG
+        svg_data = re.sub(r'width="([\d.]+)pt"', f'width="{width_px}px"', svg_data)
+        svg_data = re.sub(r'height="([\d.]+)pt"', f'height="{height_px}px"', svg_data)
+        
+        # Save SVG data to a temporary file
+        temp_svg_path = tempfile.mktemp(suffix=".svg", prefix="enhanced_diagram_")
         with open(temp_svg_path, 'w') as svg_file:
             svg_file.write(svg_data)
         
-        return temp_svg_path  # Return the file path of the saved SVG
+        return temp_svg_path  # Return path of the saved SVG
     except Exception as e:
         print(f"Error during SVG conversion: {str(e)}")
         return None
 
-# Step 2: Hugging Face Gradio Interface
-def adjust_and_convert_pdf(pdf_file, thickness_scale=1.0):
-    svg_path = pdf_to_svg(pdf_file, thickness_scale=thickness_scale)
+# Gradio Interface with width and height inputs
+def adjust_and_convert_pdf(pdf_file, width_in, height_in, thickness_scale=1.0):
+    svg_path = pdf_to_svg(pdf_file, width_in, height_in, thickness_scale=thickness_scale)
     
     # Handle cases where the SVG path is None
     if svg_path is None:
@@ -48,15 +59,18 @@ def adjust_and_convert_pdf(pdf_file, thickness_scale=1.0):
     
     return svg_path  # Return the SVG file path for download
 
+# Gradio interface with width, height, and thickness scale inputs
 iface = gr.Interface(
     fn=adjust_and_convert_pdf,
     inputs=[
         gr.File(label="Upload PDF"),
+        gr.Number(label="Width (inches)", value=5.0),
+        gr.Number(label="Height (inches)", value=4.0),
         gr.Slider(minimum=1, maximum=3, step=0.1, value=1.0, label="Thickness Scale")
     ],
     outputs=gr.File(label="Editable SVG"),
-    title="High-Quality PDF to SVG Converter",
-    description="Upload a PDF containing a diagram to convert it to a high-quality, editable SVG for CorelDRAW. Adjust the thickness scale for line elements if necessary."
+    title="PDF to Fully Editable SVG Converter for CorelDRAW",
+    description="Upload a PDF to convert it to a fully editable SVG with specific width and height dimensions in inches. Ideal for CorelDRAW editing."
 )
 
 # Launch the interface on Hugging Face Spaces