Update app.py

app.py

@@ -94,21 +94,23 @@ def estimate(rooms, wall_thickness, openings, floors):
     }
 
 def draw_plan_image(room_shapes):
-    fig, ax = plt.subplots()
+    fig, ax = plt.subplots(figsize=(8, 6))
     for x0, y0, x1, y1 in room_shapes:
         width = x1 - x0
         height = y1 - y0
-        rect = patches.Rectangle((x0, y0), width, height, linewidth=1, edgecolor='black', facecolor='none')
+        rect = patches.Rectangle((x0, y0), width, height, linewidth=1.5, edgecolor='black', facecolor='none')
         ax.add_patch(rect)
-        ax.text(x0 + width / 2, y0 - 2, f"{round(width / 12, 1)} ft", ha='center', fontsize=8)
-        ax.text(x1 + 2, y0 + height / 2, f"{round(height / 12, 1)} ft", va='center', fontsize=8, rotation=90)
+        ax.text(x0 + width / 2, y0 - 5, f"{round(width / 12, 1)} ft", ha='center', fontsize=9)
+        ax.text(x1 + 2, y0 + height / 2, f"{round(height / 12, 1)} ft", va='center', fontsize=9, rotation=90)
+    ax.set_title("2D Room Layout with Dimensions")
     ax.set_aspect('equal')
     ax.axis('off')
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
-    fig.savefig(temp_file.name, bbox_inches='tight')
+    fig.savefig(temp_file.name, bbox_inches='tight', dpi=150)
     plt.close(fig)
     return temp_file.name
 
+
 def generate_pdf(data_dict, calc_details, room_shapes):
     image_path = draw_plan_image(room_shapes)
 
@@ -144,12 +146,14 @@ def generate_pdf(data_dict, calc_details, room_shapes):
 
     pdf.ln(5)
     pdf.set_font("Arial", 'B', size=12)
-    pdf.cell(200, 10, "2D Plan with Dimensions (in ft)", ln=True)
-    pdf.image(image_path, x=10, y=None, w=180)
+    pdf.cell(200, 10, "📐 2D Plan with Dimensions (in feet)", ln=True)
+    
+    try:
+        pdf.image(image_path, x=10, y=None, w=180)
+    except Exception as e:
+        pdf.set_font("Arial", '', size=10)
+        pdf.cell(200, 10, f"Error embedding plan image: {e}", ln=True)
 
-    tmp = NamedTemporaryFile(delete=False, suffix=".pdf")
-    pdf.output(tmp.name)
-    return tmp.name
 
 def plot_rooms(shapes):
     fig, ax = plt.subplots()