import cv2
import numpy as np
import gradio as gr
from pdf2image import convert_from_path
import os
import tempfile
import logging
from PIL import Image

# Set up logging
logging.basicConfig(filename='debug.log', level=logging.DEBUG)
logger = logging.getLogger(__name__)

def validate_poppler_path(poppler_path):
    if not poppler_path:
        return None
    poppler_path = poppler_path.replace('\\', '/')
    if not os.path.isdir(poppler_path):
        return None
    pdftoppm_path = os.path.join(poppler_path, "pdftoppm" + (".exe" if os.name == "nt" else ""))
    if not os.path.isfile(pdftoppm_path):
        return None
    return poppler_path

def calculate_materials_from_dimensions(wall_area, foundation_area):
    return {
        "cement": round(wall_area * 10 + foundation_area * 20, 2),
        "bricks": int(wall_area * 500 + foundation_area * 750),
        "steel": round(wall_area * 2 + foundation_area * 5, 2)
    }

def process_blueprint(uploaded_file, blueprint_width_m=27, blueprint_height_m=9.78, poppler_path=None):
    try:
        filename = uploaded_file.name
        poppler_path = validate_poppler_path(poppler_path)
        
        # Convert input to image
        if filename.endswith(".pdf"):
            images = convert_from_path(uploaded_file.name, poppler_path=poppler_path)
            if not images:
                return {"error": "No pages found in PDF"}
            image = np.array(images[0])
        else:
            image = cv2.imdecode(np.frombuffer(uploaded_file.read(), np.uint8), cv2.IMREAD_COLOR)

        if image is None:
            return {"error": "Failed to load image"}

        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        edges = cv2.Canny(gray, 50, 150)
        lines = cv2.HoughLinesP(edges, 1, np.pi / 180, 100, minLineLength=50, maxLineGap=10)

        total_wall_length_pixels = sum(
            np.sqrt((x2 - x1) ** 2 + (y2 - y1) ** 2)
            for line in lines for x1, y1, x2, y2 in [line[0]]
        ) if lines is not None else 0

        img_h, img_w = image.shape[:2]
        pixel_to_meter_w = blueprint_width_m / img_w
        pixel_to_meter_h = blueprint_height_m / img_h
        avg_pixel_to_meter = (pixel_to_meter_w + pixel_to_meter_h) / 2

        total_wall_length_m = total_wall_length_pixels * avg_pixel_to_meter
        wall_height_m = 3
        wall_area = total_wall_length_m * wall_height_m

        total_area = blueprint_width_m * blueprint_height_m
        foundation_area = total_area * 0.1

        materials = calculate_materials_from_dimensions(wall_area, foundation_area)
        return {
            "cement": f"{materials['cement']} kg",
            "bricks": f"{materials['bricks']} units",
            "steel": f"{materials['steel']} kg"
        }

    except Exception as e:
        return {"error": f"Processing failed: {str(e)}"}

# Gradio UI
interface = gr.Interface(
    fn=process_blueprint,
    inputs=[
        gr.File(label="Upload Blueprint (PDF or Image)", file_types=[".pdf", ".png", ".jpg", ".jpeg"]),
        gr.Number(label="Blueprint Width (meters)", value=27),
        gr.Number(label="Blueprint Height (meters)", value=9.78),
        gr.Textbox(label="Poppler Path (Optional)", placeholder="e.g., C:/poppler/bin")
    ],
    outputs=gr.JSON(label="Estimated Materials"),
    title="Blueprint Estimator",
    description="Upload a blueprint as PDF or image to estimate materials (cement, bricks, steel)."
)

if __name__ == "__main__":
    interface.launch()