Create app.py

app.py

@@ -0,0 +1,70 @@
+import fitz  # PyMuPDF
+import gradio as gr
+from PIL import Image, ImageEnhance
+import io
+
+# Step 1: Convert the entire PDF page to a high-quality image
+def pdf_to_image(pdf_file, zoom=4):  # Increased zoom for better quality
+    try:
+        print(f"Opening PDF: {pdf_file.name}")
+
+        # Open the PDF file
+        pdf_document = fitz.open(pdf_file.name)
+        print("PDF opened successfully.")
+
+        # Load the first page
+        page = pdf_document.load_page(0)
+        print("Loaded the first page of the PDF.")
+
+        # Render the page to an image with a higher zoom level for better quality
+        mat = fitz.Matrix(zoom, zoom)  # Increase zoom for higher resolution
+        pix = page.get_pixmap(matrix=mat, alpha=False)  # No transparency
+
+        # Save the pixmap to a PNG byte stream
+        image_bytes = io.BytesIO(pix.tobytes("png"))
+        img = Image.open(image_bytes)
+        return img
+    except Exception as e:
+        print(f"Error during PDF to image conversion: {str(e)}")
+        return None
+
+# Step 2: Adjust image dimensions and brightness
+def adjust_image(pdf_file, height, width, brightness):
+    print("Processing the uploaded PDF...")
+    img = pdf_to_image(pdf_file)
+
+    if img is None:
+        print("Failed to convert PDF to image.")
+        return "Unable to convert PDF to image."
+
+    try:
+        # Resize the image
+        img_resized = img.resize((int(width), int(height)), Image.LANCZOS)  # High-quality resampling
+        print(f"Image resized to: {width}x{height}")
+
+        # Adjust brightness
+        enhancer = ImageEnhance.Brightness(img_resized)
+        img_bright = enhancer.enhance(brightness)
+        print(f"Brightness adjusted by factor: {brightness}")
+
+        return img_bright
+    except Exception as e:
+        print(f"Error during image adjustment: {str(e)}")
+        return "Error adjusting the image."
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=adjust_image,
+    inputs=[
+        gr.File(label="Upload PDF"),
+        gr.Number(label="Height", value=1000),
+        gr.Number(label="Width", value=800),
+        gr.Slider(minimum=0.5, maximum=2, step=0.1, value=1.0, label="Brightness"),
+    ],
+    outputs="image",
+    title="Plate Design Creation - Convert & Adjust PDF Page",
+    description="Upload a PDF containing a technical diagram or nameplate. Convert the page to an image and adjust its dimensions and brightness."
+)
+
+# Launch the interface
+iface.launch()