Create app.py

app.py

@@ -0,0 +1,75 @@
+import gradio as gr
+import matplotlib.pyplot as plt
+from shapely.geometry import Point, Polygon, box
+import ezdxf
+
+def generate_floor_plan(plot_size, num_rooms, building_type):
+    # Parse plot dimensions
+    try:
+        plot_width, plot_height = map(float, plot_size.split('x'))
+    except ValueError:
+        return "Error: Plot size must be in 'width x height' format (e.g., 50x30)."
+
+    # Create plot boundary
+    plot_boundary = box(0, 0, plot_width, plot_height)
+
+    # Calculate room sizes
+    room_width = plot_width / (num_rooms ** 0.5)
+    room_height = plot_height / (num_rooms ** 0.5)
+
+    # Generate room layout
+    rooms = []
+    x, y = 0, 0
+    for _ in range(int(num_rooms ** 0.5)):
+        for _ in range(int(num_rooms ** 0.5)):
+            room = box(x, y, x + room_width, y + room_height)
+            rooms.append(room)
+            x += room_width
+        x = 0
+        y += room_height
+
+    # Add courtyard (10% of plot area at center)
+    courtyard_size = (plot_width * plot_height) * 0.1
+    courtyard_side = courtyard_size ** 0.5
+    courtyard = box(
+        (plot_width - courtyard_side) / 2,
+        (plot_height - courtyard_side) / 2,
+        (plot_width + courtyard_side) / 2,
+        (plot_height + courtyard_side) / 2,
+    )
+
+    # Visualization
+    fig, ax = plt.subplots()
+    ax.set_xlim(0, plot_width)
+    ax.set_ylim(0, plot_height)
+
+    # Plot elements
+    ax.add_patch(plt.Rectangle((0, 0), plot_width, plot_height, edgecolor="black", fill=False, lw=2))  # Plot boundary
+    for room in rooms:
+        ax.add_patch(plt.Rectangle((room.bounds[0], room.bounds[1]), room_width, room_height, edgecolor="blue", fill=False))
+    ax.add_patch(plt.Rectangle((courtyard.bounds[0], courtyard.bounds[1]), courtyard_side, courtyard_side, edgecolor="green", fill=False, lw=2))  # Courtyard
+
+    plt.axis("off")
+    plt.close(fig)
+
+    # Save the plot as an image
+    img_path = "floor_plan.png"
+    fig.savefig(img_path)
+
+    return img_path
+
+# Gradio Interface
+interface = gr.Interface(
+    fn=generate_floor_plan,
+    inputs=[
+        gr.Textbox(label="Plot Size (e.g., 50x30)"),
+        gr.Number(label="Number of Rooms", value=4),
+        gr.Radio(["Residential", "Commercial"], label="Building Type"),
+    ],
+    outputs="image",
+    title="Floor Plan Generator",
+    description="Enter plot size, number of rooms, and building type to generate a floor plan."
+)
+
+if __name__ == "__main__":
+    interface.launch()