Update app.py

app.py

@@ -2,6 +2,7 @@ import cv2
 import numpy as np
 from PIL import Image
 import gradio as gr
+from pdf2image import convert_from_path  # Import the PDF-to-image conversion library
 
 # Function to calculate materials based on blueprint dimensions
 def calculate_materials_from_dimensions(wall_area, foundation_area):
@@ -27,8 +28,14 @@ def calculate_materials_from_dimensions(wall_area, foundation_area):
 
 # Function to process the blueprint and extract dimensions
 def process_blueprint(image_path):
-    # Open the image
-    image = cv2.imread(image_path)
+    # Open the image (supporting both image and PDF inputs)
+    if image_path.lower().endswith(".pdf"):
+        # Convert the PDF to an image (first page)
+        images = convert_from_path(image_path, first_page=1, last_page=1)
+        image = np.array(images[0])
+    else:
+        # Open the image file
+        image = cv2.imread(image_path)
     if image is None:
         raise ValueError("Could not load the image")
 
@@ -86,12 +93,12 @@ def process_blueprint(image_path):
 # Set up Gradio interface
 interface = gr.Interface(
     fn=process_blueprint,
-    inputs=gr.Image(type="filepath", label="Upload Blueprint Image"),
+    inputs=gr.File(label="Upload Blueprint (Image or PDF)", type="filepath"),  # Support both image and PDF files
     outputs=gr.JSON(label="Material Estimates"),
     title="Blueprint Material Estimator",
-    description="Upload a blueprint image to estimate construction materials."
+    description="Upload a blueprint image or PDF to estimate construction materials."
 )
 
 # Launch the interface
 if __name__ == "__main__":
-    interface.launch(share=False)
+    interface.launch(share=False)