Create app.py

app.py

@@ -1,15 +1,23 @@
 import gradio as gr
+import cv2
+import numpy as np
 
-def generate_floorplan(image):
-    # Filhal ye demo ke liye sirf message dega
-    return "Naksha ban raha hai... (yahan AI model connect hoga)"
+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
 
-iface = gr.Interface(
-    fn=generate_floorplan,
-    inputs=gr.Image(type="filepath", label="Makan ki tasveer upload karo"),
-    outputs="text",
+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."
 )
 
-iface.launch()
+if __name__ == "__main__":
+    demo.launch()