Update app.py

app.py

@@ -2,6 +2,9 @@ import streamlit as st
 from groq import Groq
 import json
 import os
+import matplotlib.pyplot as plt
+import numpy as np
+import plotly.graph_objects as go
 from dotenv import load_dotenv
 
 # Load environment variables
@@ -15,7 +18,7 @@ except Exception as e:
 
 # App title
 st.title("🏠 AI Architect: Home Planner")
-st.markdown("Get a professionally designed house plan in seconds!")
+st.markdown("Get an easy-to-understand house plan with 2D/3D visualization!")
 
 # --- USER INPUTS ---
 st.header("📏 Step 1: Plot Details")
@@ -43,51 +46,122 @@ lawn = st.checkbox("Lawn/Garden 🌳")
 # --- AI GENERATION ---
 if st.button("Generate Professional Plan"):
     prompt = f"""
-    Design a house plan for a {plot_length}x{plot_width} ft plot following architectural best practices:
-    - Space Efficiency: Minimize wasted space, use standard dimensions
-    - Room Requirements:
-      - {master_bed} master bedroom(s) (min 12x14 ft)
-      - {bedrooms} standard bedroom(s) (min 10x12 ft)
-      - {kitchens} kitchen(s) (min 8x10 ft)
-      - {study_rooms} study room(s) (min 8x8 ft)
-      - {lounges} lounge(s) (min 12x16 ft)
-    - Washrooms:
-      - {attached_washrooms} attached (min 6x8 ft)
-      - {common_washrooms} common (min 5x7 ft)
-    - Extras: {"Parking (min 12x20 ft)" if parking else ""}, {"Lawn" if lawn else ""}
-
-    Respond in JSON with:
-    1. '2d_layout': Scaled room placements with dimensions
-    2. '3d_tips': How to visualize flow
-    3. 'materials': Cost-effective material suggestions
-    4. 'warnings': Violations of standard dimensions
+    Design a house plan for a {plot_length}x{plot_width} ft plot following architectural best practices.
+    Return response in JSON format with:
+    1. 'floor_plan': 2D layout with room dimensions and positions
+    2. 'visualization_tips': Simple tips for visualizing the space
+    3. 'material_suggestions': Recommended materials
+    4. 'warnings': Any size violations
     """
     
     try:
         response = client.chat.completions.create(
             messages=[{"role": "user", "content": prompt}],
-            model="llama3-70b-8192",  # Updated to currently supported model
+            model="llama3-70b-8192",
             response_format={"type": "json_object"}
         )
         plan = json.loads(response.choices[0].message.content)
 
         st.success("✅ Professionally Designed Plan Generated!")
         
-        st.header("📐 2D Layout")
-        st.write(plan["2d_layout"])
+        # --- 2D Visualization ---
+        st.header("📐 2D Floor Plan")
+        
+        # Create figure
+        fig, ax = plt.subplots(figsize=(10, 8))
+        
+        # Plot rooms
+        for room in plan['floor_plan']['rooms']:
+            rect = plt.Rectangle(
+                (room['x'], room['y']), 
+                room['width'], 
+                room['height'],
+                linewidth=2,
+                edgecolor='black',
+                facecolor=np.random.rand(3,))
+            ax.add_patch(rect)
+            
+            # Add room label
+            center_x = room['x'] + room['width']/2
+            center_y = room['y'] + room['height']/2
+            ax.text(center_x, center_y, 
+                   f"{room['width']}x{room['height']}ft",
+                   ha='center', va='center', fontsize=8)
+        
+        # Plot washrooms
+        for washroom in plan['floor_plan']['washrooms']:
+            rect = plt.Rectangle(
+                (washroom['x'], washroom['y']), 
+                washroom['width'], 
+                washroom['height'],
+                linewidth=2,
+                edgecolor='black',
+                facecolor='lightblue',
+                hatch='/')
+            ax.add_patch(rect)
+        
+        # Set plot limits and labels
+        ax.set_xlim(0, plot_length)
+        ax.set_ylim(0, plot_width)
+        ax.set_aspect('equal')
+        ax.set_xlabel('Length (ft)')
+        ax.set_ylabel('Width (ft)')
+        ax.set_title('2D Floor Plan')
+        st.pyplot(fig)
+        
+        # --- 3D Visualization ---
+        st.header("✨ 3D Visualization")
+        
+        # Create simple 3D representation
+        fig_3d = go.Figure()
+        
+        # Add rooms as 3D boxes
+        for room in plan['floor_plan']['rooms']:
+            fig_3d.add_trace(go.Mesh3d(
+                x=[room['x'], room['x'], room['x']+room['width'], room['x']+room['width'], room['x']],
+                y=[room['y'], room['y']+room['height'], room['y']+room['height'], room['y'], room['y']],
+                z=[0, 0, 0, 0, 0],
+                opacity=0.8,
+                color='lightgray',
+                showscale=False
+            ))
+            fig_3d.add_trace(go.Mesh3d(
+                x=[room['x'], room['x'], room['x']+room['width'], room['x']+room['width'], room['x']],
+                y=[room['y'], room['y']+room['height'], room['y']+room['height'], room['y'], room['y']],
+                z=[10, 10, 10, 10, 10],
+                opacity=0.8,
+                color='lightgray',
+                showscale=False
+            ))
+        
+        fig_3d.update_layout(
+            scene=dict(
+                xaxis=dict(title='Length (ft)'),
+                yaxis=dict(title='Width (ft)'),
+                zaxis=dict(title='Height (ft)'),
+                aspectmode='manual',
+                aspectratio=dict(x=1, y=plot_width/plot_length, z=0.2)
+            ),
+            margin=dict(r=20, l=10, b=10, t=30)
+        )
+        st.plotly_chart(fig_3d)
         
-        st.header("✨ 3D Flow Tips")
-        st.write(plan["3d_tips"])
+        # --- Additional Information ---
+        st.header("💡 Visualization Tips")
+        for tip in plan['visualization_tips']:
+            st.markdown(f"- {tip}")
         
-        st.header("🏗️ Material Recommendations")
-        st.write(plan["materials"])
+        st.header("🏗️ Material Suggestions")
+        for material in plan['material_suggestions']:
+            st.markdown(f"- {material}")
         
-        st.header("⚠️ Professional Warnings")
-        st.write(plan.get("warnings", "No violations detected."))
+        st.header("⚠️ Warnings")
+        for warning in plan['warnings']:
+            st.markdown(f"- ⚠️ {warning}")
         
     except Exception as e:
         st.error(f"Error: {e}. Check your API key or try again later.")
 
 # Footer
 st.markdown("---")
-st.caption("✅ Follows standard architectural dimensions | Powered by Groq API")
+st.caption("✅ Easy-to-understand visualizations | Powered by Groq API")