Create app.py

app.py

@@ -0,0 +1,93 @@
+import gradio as gr
+import matplotlib.pyplot as plt
+import random
+import os
+
+# Function to generate a room layout based on dimensions and desired upgrades
+def generate_room_plan(dimensions, current_use, upgrades):
+    try:
+        width, height = map(float, dimensions.split('x'))
+    except ValueError:
+        return "Error: Room dimensions must be in 'width x height' format (e.g., 15x12)."
+
+    # Create a figure
+    fig, ax = plt.subplots(figsize=(8, 6))
+    ax.set_xlim(0, width)
+    ax.set_ylim(0, height)
+
+    # Draw room boundary
+    ax.plot([0, width, width, 0, 0], [0, 0, height, height, 0], color="black", linewidth=2, label="Room Boundary")
+
+    # Add furniture placement based on upgrades
+    if "bed" in upgrades.lower():
+        bed_width, bed_height = width * 0.3, height * 0.2
+        ax.add_patch(plt.Rectangle((1, 1), bed_width, bed_height, color="blue", alpha=0.5, label="Bed"))
+    if "sofa" in upgrades.lower():
+        sofa_width, sofa_height = width * 0.3, height * 0.1
+        ax.add_patch(plt.Rectangle((width - sofa_width - 1, 1), sofa_width, sofa_height, color="green", alpha=0.5, label="Sofa"))
+    if "table" in upgrades.lower():
+        table_width, table_height = width * 0.2, height * 0.15
+        ax.add_patch(plt.Rectangle((width / 2 - table_width / 2, height / 2 - table_height / 2), table_width, table_height, color="brown", alpha=0.5, label="Table"))
+
+    # Add labels and legend
+    ax.set_title(f"{current_use.capitalize()} Layout with Upgrades")
+    ax.legend(loc="upper right")
+    ax.axis('off')
+
+    # Save the figure to a file
+    output_path = "room_layout.png"
+    plt.savefig(output_path)
+    plt.close(fig)
+
+    # Return the file path
+    return output_path
+
+# Function to estimate renovation costs
+def estimate_cost(dimensions, upgrades):
+    try:
+        width, height = map(float, dimensions.split('x'))
+    except ValueError:
+        return "Error: Room dimensions must be in 'width x height' format (e.g., 15x12)."
+
+    area = width * height
+    base_cost = area * 50  # Basic cost per square foot
+
+    upgrade_costs = {
+        "bed": 500,
+        "sofa": 300,
+        "table": 200,
+        "paint": 150,
+        "lighting": 250
+    }
+
+    total_cost = base_cost
+    for upgrade in upgrades.split(','):
+        upgrade = upgrade.strip().lower()
+        if upgrade in upgrade_costs:
+            total_cost += upgrade_costs[upgrade]
+
+    return f"Estimated Renovation Cost: ${total_cost:.2f}"
+
+# Gradio interface
+def renovation_planner(dimensions, current_use, upgrades):
+    layout = generate_room_plan(dimensions, current_use, upgrades)
+    cost = estimate_cost(dimensions, upgrades)
+    return layout, cost
+
+interface = gr.Interface(
+    fn=renovation_planner,
+    inputs=[
+        gr.Textbox(label="Room Dimensions (e.g., 15x12)"),
+        gr.Textbox(label="Current Use (e.g., Bedroom, Living Room)"),
+        gr.Textbox(label="Desired Upgrades (e.g., Bed, Sofa, Paint, Lighting)"),
+    ],
+    outputs=[
+        gr.Image(label="Room Layout"),
+        gr.Textbox(label="Renovation Cost Estimate"),
+    ],
+    title="Single-Room Renovation Planner",
+    description="Provide room dimensions, current use, and desired upgrades to get a layout and cost estimate."
+)
+
+if __name__ == "__main__":
+    interface.launch()