import gradio as gr
import cv2
import numpy as np

def make_floorplan(image):
    # Convert image to grayscale
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    # Edge detection
    edges = cv2.Canny(gray, 100, 200)
    # Convert edges back to 3 channels
    edges_rgb = cv2.cvtColor(edges, cv2.COLOR_GRAY2BGR)
    return edges_rgb

demo = gr.Interface(
    fn=make_floorplan,
    inputs=gr.Image(type="numpy", label="Makan ki tasveer upload karo"),
    outputs=gr.Image(type="numpy", label="Output Floorplan"),
    title="Floorplan AI",
    description="Makan ki image upload karo aur naksha banao."
)

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