Update app.py

app.py

@@ -1,38 +1,55 @@
-import json
 import streamlit as st
+import json
 import matplotlib.pyplot as plt
 
-# Page config
-st.set_page_config(layout="wide") 
-
-# Load grid data
-with open('grid.json') as f:
-    grid_data = json.load(f)
-
-# Extract coordinates and labels
-buildings = [[b['coords'][0], b['coords'][1], b['label']] for b in grid_data['buildings']]
-stores = [[s['coords'][0], s['coords'][1]] for s in grid_data['stores']]
-roads = [(r['start'], r['end']) for r in grid_data['roads']]
-
-# Plot map
-fig, ax = plt.subplots()
-ax.axis([0, 10, 0, 10])
-ax.set_xticks([i for i in range(11)])
-ax.set_yticks([i for i in range(11)])
-ax.grid(color='grey', linestyle='-', linewidth=0.25)
-
-# Plot and label buildings
-for building in buildings:
-    ax.scatter(building[0], building[1], color='red')
-    ax.text(building[0], building[1], building[2], color='black', fontsize=8)
-
-# Plot stores
-ax.scatter(*zip(*stores), color='green')
-
-# Draw roads
-for start, end in roads:
-    ax.plot([start[0], end[0]], [start[1], end[1]], color='black')
-
-# Show figure in Streamlit    
-st.pyplot(fig)
-
+# Function to load and parse the JSON data
+def load_data(filename):
+    with open(filename, 'r') as file:
+        data = json.load(file)
+    return data
+
+# Function to draw the grid layout
+def draw_grid(data):
+    # Create a figure and a grid of subplots
+    fig, ax = plt.subplots(figsize=(10, 10))
+    
+    # Setting the grid size
+    nrows, ncols = data['size']['rows'], data['size']['columns']
+    ax.set_xlim(0, ncols)
+    ax.set_ylim(0, nrows)
+    ax.set_xticks(range(ncols+1))
+    ax.set_yticks(range(nrows+1))
+    ax.grid(True)
+    
+    # Plotting each building
+    for building in data['buildings']:
+        # Extracting the building details
+        coords = building['coords']
+        b_type = building['type']
+        size = building['size']
+        
+        # Plotting the building on the grid
+        ax.add_patch(plt.Rectangle((coords[1], nrows-coords[0]-size), size, size, fill=None, edgecolor='r', linewidth=2))
+        ax.text(coords[1]+0.5*size, nrows-coords[0]-0.5*size, b_type, ha='center', va='center')
+    
+    # Setting labels and title
+    ax.set_xlabel('Columns')
+    ax.set_ylabel('Rows')
+    ax.set_title('Village Layout')
+    
+    return fig
+
+# Streamlit application starts here
+def main():
+    st.title('Green Smart Village Layout')
+    
+    # Load and display the data
+    data = load_data('grid.json')  # Adjust this path if your grid.json is located elsewhere
+    st.json(data)
+    
+    # Drawing and displaying the grid layout
+    fig = draw_grid(data)
+    st.pyplot(fig)
+
+if __name__ == "__main__":
+    main()