Create app.py

app.py

@@ -0,0 +1,58 @@
+import fitz  # PyMuPDF for PDF manipulation
+import svgwrite  # For creating SVGs
+import gradio as gr
+import io
+import tempfile
+import os
+
+# Step 1: Extract vector-based content from PDF and save as SVG
+def pdf_to_svg(pdf_file, 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)
+        
+        # Extract SVG content from the page at a scale of 2x for higher quality
+        svg_data = page.get_svg_image(matrix=fitz.Matrix(2, 2))  # 2x scaling
+        
+        # Load SVG content into svgwrite for manipulation
+        svg_io = io.StringIO(svg_data)
+        dwg = svgwrite.Drawing(filename="output_diagram.svg")
+        dwg.embed_svg(svg_io)
+        
+        # Adjust stroke width for each path if thickness_scale > 1
+        for element in dwg.elements:
+            if 'stroke-width' in element.attrib:
+                current_width = float(element.attrib['stroke-width'])
+                element.attrib['stroke-width'] = str(current_width * thickness_scale)
+        
+        # Save the final SVG with modifications
+        temp_svg_path = tempfile.mktemp(suffix=".svg", prefix="enhanced_diagram_")
+        dwg.saveas(temp_svg_path)
+        
+        return temp_svg_path  # Return the file 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)
+    if svg_path is None:
+        return "Error: Unable to convert PDF to editable SVG."
+    
+    return svg_path  # Return the SVG file path for download
+
+iface = gr.Interface(
+    fn=adjust_and_convert_pdf,
+    inputs=[
+        gr.File(label="Upload PDF"),
+        gr.Slider(minimum=1, maximum=3, step=0.1, value=1.0, label="Thickness Scale")
+    ],
+    outputs=gr.File(label="Editable SVG"),
+    title="PDF to Editable SVG Converter",
+    description="Upload a PDF containing a diagram to convert it to an editable SVG for CorelDRAW. Adjust the thickness scale for line elements as needed."
+)
+
+# Launch the interface on Hugging Face Spaces
+iface.launch()