| import re |
|
|
| STYLES = { |
| "ranch": "ranch", "rambler": "ranch", |
| "colonial": "colonial", "cape": "colonial", |
| "modern": "modern", "contemporary": "modern", |
| "victorian": "victorian", |
| "craftsman": "craftsman", |
| "cabin": "cabin", "cottage": "cabin", |
| "tudor": "tudor", |
| "farmhouse": "farmhouse", |
| } |
|
|
| ROOF_STYLES = { |
| "ranch": "hip", "rambler": "hip", |
| "colonial": "gable", "cape": "gable", |
| "modern": "flat", "contemporary": "flat", |
| "victorian": "steep-gable", |
| "craftsman": "low-gable", |
| "cabin": "gable", "cottage": "gable", |
| "tudor": "steep-gable", |
| "farmhouse": "gable", |
| } |
|
|
| ROOM_COLORS = { |
| "living": "#e8dcc8", "family": "#e8dcc8", |
| "kitchen": "#d4e8d0", "dining": "#e0d0c0", |
| "bedroom": "#d0d8e8", "master": "#c8d0e8", |
| "bath": "#d8e8e8", "bathroom": "#d8e8e8", |
| "office": "#e0d0d8", "den": "#e0d0d8", |
| "garage": "#d0d0d0", |
| "hall": "#f0ece8", "hallway": "#f0ece8", |
| "laundry": "#e8e0d8", |
| "closet": "#ece8e0", |
| "loft": "#e8e0d0", |
| "basement": "#c8c0b8", |
| "porch": "#d8d0c8", "deck": "#c8b8a8", |
| "patio": "#c8b8a8", |
| } |
|
|
|
|
| def parse_description(text): |
| text = text.lower().strip() |
|
|
| sqft = _extract_sqft(text) |
| width, depth = _extract_dimensions(text, sqft) |
| style = _extract_style(text) |
| stories = _extract_stories(text) |
| bedrooms = _extract_bedrooms(text) |
| bathrooms = _extract_bathrooms(text) |
| garage = _has_garage(text) |
| extra_rooms = _extract_extra_rooms(text) |
|
|
| rooms = _generate_rooms(width, depth, stories, bedrooms, bathrooms, garage, extra_rooms, style) |
|
|
| roof_style = ROOF_STYLES.get(style, "gable") |
| return { |
| "structure": { |
| "width": width, |
| "depth": depth, |
| "stories": stories, |
| "style": style, |
| "roof_style": roof_style, |
| "sqft": sqft or (width * depth * stories), |
| }, |
| "rooms": rooms, |
| "features": { |
| "bedrooms": int(bedrooms), |
| "bathrooms": float(bathrooms), |
| "garage": garage, |
| "stories": int(stories), |
| }, |
| "summary": _generate_summary(width, depth, stories, style, bedrooms, bathrooms), |
| } |
|
|
|
|
| def _extract_sqft(text): |
| m = re.search(r'(\d+[,.]?\d*)\s*(sq\s*ft|square\s*feet|sqft)', text) |
| if m: |
| return int(float(m.group(1).replace(",", ""))) |
| m = re.search(r'(\d+[,.]?\d*)\s*x\s*(\d+[,.]?\d*)\s*(?:feet|ft|foot)', text) |
| if m: |
| w = float(m.group(1).replace(",", "")) |
| d = float(m.group(2).replace(",", "")) |
| return int(w * d) |
| return None |
|
|
|
|
| def _extract_dimensions(text, sqft): |
| m = re.search(r'(\d+)\s*(?:feet|ft|foot)\s*(?:x|by|and)?\s*(\d+)\s*(?:feet|ft|foot)', text) |
| if m: |
| return int(m.group(1)), int(m.group(2)) |
| m = re.search(r'(\d+)\s*x\s*(\d+)', text) |
| if m: |
| return int(m.group(1)), int(m.group(2)) |
| m = re.search(r'(\d+)\s*(?:feet|ft|foot)\s*(?:wide|width)?', text) |
| if m: |
| w = int(m.group(1)) |
| if sqft: |
| d = max(1, sqft // w) |
| else: |
| d = int(w * 0.75) |
| return w, d |
| m = re.search(r'(\d+)\s*(?:by|wide|width)\s*(\d+)', text) |
| if m: |
| return int(m.group(1)), int(m.group(2)) |
| if sqft: |
| w = int(sqft ** 0.5) |
| d = max(1, sqft // w) |
| return w, d |
| return 40, 30 |
|
|
|
|
| def _extract_style(text): |
| for style, normalized in STYLES.items(): |
| if style in text: |
| return normalized |
| if "2 story" in text or "two story" in text or "two-story" in text: |
| return "colonial" |
| return "modern" |
|
|
|
|
| def _extract_stories(text): |
| m = re.search(r'(\d+)\s*[-]?\s*(?:story|floor|level|storey)', text) |
| if m: |
| return min(int(m.group(1)), 3) |
| if "single story" in text or "ranch" in text or "rambler" in text: |
| return 1 |
| return 1 |
|
|
|
|
| def _extract_bedrooms(text): |
| m = re.search(r'(\d+)\s*(?:bedroom|bed\b)', text) |
| if m: |
| return int(m.group(1)) |
| m = re.search(r'(\d+)\s*br\b', text) |
| if m: |
| return int(m.group(1)) |
| return 3 |
|
|
|
|
| def _extract_bathrooms(text): |
| m = re.search(r'(\d+)\s*(?:bathroom|bath\b)', text) |
| if m: |
| return int(m.group(1)) |
| m = re.search(r'(\d+)\s*ba\b', text) |
| if m: |
| return int(m.group(1)) |
| m = re.search(r'(\d+\.?\d*)\s*-?\s*bath', text) |
| if m: |
| return float(m.group(1)) |
| return 2 |
|
|
|
|
| def _has_garage(text): |
| if re.search(r'\bgarage\b', text): |
| return True |
| if re.search(r'\bcar\s*(?:garage|port)\b', text): |
| return True |
| return False |
|
|
|
|
| def _extract_extra_rooms(text): |
| extras = [] |
| room_keywords = ["office", "den", "library", "gym", "theater", "workshop", |
| "laundry", "mudroom", "sunroom", "conservatory", "studio", |
| "loft", "basement", "attic", "porch", "deck", "patio", |
| "wine cellar", "pantry", "playroom", "nursery"] |
| for kw in room_keywords: |
| if kw in text: |
| extras.append(kw) |
| return extras |
|
|
|
|
| def _generate_rooms(width, depth, stories, bedrooms, bathrooms, garage, extras, style): |
| rooms = [] |
| color = lambda name: ROOM_COLORS.get(name, "#e8e0d8") |
|
|
| w3 = width / 3.0 |
| d2 = depth / 2.0 |
|
|
| rooms.append({"name": "Living Room", "x": 0, "z": 0, "w": w3 * 1.5, "d": d2, "color": color("living"), "floor": 0}) |
| rooms.append({"name": "Kitchen", "x": w3 * 1.5, "z": 0, "w": w3 * 0.8, "d": d2, "color": color("kitchen"), "floor": 0}) |
| rooms.append({"name": "Dining", "x": w3 * 1.5 + w3 * 0.8, "z": 0, "w": w3 * 0.7, "d": d2, "color": color("dining"), "floor": 0}) |
|
|
| bed_colors = ["#d0d8e8", "#c8d0e8", "#d8d0e0", "#d0d0e0", "#d8d8e8"] |
| bed_w = min(w3 * 0.7, 14) |
| bed_d = min(d2 * 0.6, 14) |
|
|
| for i in range(bedrooms): |
| bx = (i % 3) * w3 |
| bz = d2 + (i // 3) * bed_d |
| if bx + bed_w > width: |
| bx = width - bed_w |
| name = "Master Bedroom" if i == 0 else f"Bedroom {i+1}" |
| rooms.append({"name": name, "x": bx, "z": bz, "w": bed_w, "d": bed_d, "color": bed_colors[i % len(bed_colors)], "floor": 0}) |
|
|
| bath_w = 6 |
| bath_d = 6 |
| for i in range(int(bathrooms)): |
| bx = width - bath_w - (i * bath_w) % (width / 3) |
| bz = 0 if i % 2 == 0 else d2 - bath_d |
| if bx < 0: |
| bx = width - bath_w |
| rooms.append({"name": f"{'Master ' if i == 0 and bedrooms > 0 else ''}Bathroom" if i == 0 else f"Bathroom {i+1}", |
| "x": bx, "z": bz, "w": bath_w, "d": bath_d, "color": color("bathroom"), "floor": 0}) |
|
|
| if garage: |
| gw = min(width * 0.35, 24) |
| gd = min(depth * 0.4, 24) |
| rooms.append({"name": "Garage", "x": 0, "z": depth - gd, "w": gw, "d": gd, "color": color("garage"), "floor": 0}) |
|
|
| hall_w = 4 |
| rooms.append({"name": "Hallway", "x": 0, "z": d2, "w": hall_w, "d": bed_d if bedrooms > 0 else d2 * 0.5, "color": color("hallway"), "floor": 0}) |
|
|
| for extra in extras: |
| ex = width * 0.6 |
| ez = depth * 0.3 |
| ew = min(w3 * 0.6, 12) |
| ed = min(d2 * 0.5, 10) |
| if extra in ("deck", "patio", "porch"): |
| rooms.append({"name": extra.capitalize(), "x": width * 0.3, "z": -8, "w": width * 0.4, "d": 8, "color": color(extra), "floor": 0, "exterior": True}) |
| else: |
| rooms.append({"name": extra.capitalize(), "x": ex, "z": ez, "w": ew, "d": ed, "color": color(extra), "floor": 0}) |
|
|
| if stories > 1: |
| for room in rooms: |
| if room["floor"] == 0 and room["name"] not in ("Garage",): |
| upper = room.copy() |
| upper["name"] = f"Upper {room['name']}" |
| upper["floor"] = 1 |
| upper["color"] = "#e0dcd8" |
| rooms.append(upper) |
|
|
| return rooms |
|
|
|
|
| def _generate_summary(width, depth, stories, style, bedrooms, bathrooms): |
| sqft = width * depth * stories |
| return f"A {style} {stories}-story home, {width}ft x {depth}ft — approximately {sqft} sqft. {bedrooms} bedrooms, {bathrooms} bathrooms." |
|
|
|
|
| if __name__ == "__main__": |
| tests = [ |
| "2000 sq ft ranch house with 3 bedrooms and 2 baths", |
| "modern 2-story house 30x40 with 4 bedrooms 3 bathrooms and a 2 car garage", |
| "small cabin 800 sq ft with 1 bedroom", |
| "victorian 3500 sq ft 5 bedroom 3 bath with office and library", |
| "colonial with 4 bedrooms and 2.5 baths", |
| ] |
| for t in tests: |
| r = parse_description(t) |
| print(f"\n{'='*60}") |
| print(f"Input: {t}") |
| print(f"Structure: {r['structure']}") |
| print(f"Rooms: {len(r['rooms'])}") |
| for room in r['rooms'][:5]: |
| print(f" {room['name']}: {room['w']:.0f}x{room['d']:.0f} @ ({room['x']:.0f},{room['z']:.0f}) floor={room.get('floor',0)}") |
| print(f"Summary: {r['summary']}") |
|
|
