Update app.py

app.py

@@ -10,8 +10,8 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Basic: Read and write graphs", "Basic: Simple graph", 
                                   "Basic: Simple graph Directed", "Drawing: Custom Node Position",
                                   "Drawing: Cluster Layout", "Drawing: Degree Analysis",
-                                  "Drawing: Ego Graph", "Drawing: Eigenvalues", "Drawing: Four Grids"])
-
+                                  "Drawing: Ego Graph", "Drawing: Eigenvalues", "Drawing: Four Grids",
+                                  "Drawing: House With Colors"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -19,6 +19,63 @@ def draw_graph(G, pos=None, title="Graph Visualization"):
     nx.draw(G, pos=pos, with_labels=True, node_color='lightblue', node_size=500, font_size=10, font_weight='bold')
     st.pyplot(plt)
 
+# Function to display Drawing: House With Colors
+def display_house_with_colors():
+    st.title("Drawing: House With Colors")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        # Create the house graph and explicitly set positions
+        G = nx.house_graph()
+        pos = {0: (0, 0), 1: (1, 0), 2: (0, 1), 3: (1, 1), 4: (0.5, 2.0)}
+
+        # Plot nodes with different properties for the "wall" and "roof" nodes
+        nx.draw_networkx_nodes(G, pos, node_size=3000, nodelist=[0, 1, 2, 3], node_color="tab:blue")
+        nx.draw_networkx_nodes(G, pos, node_size=2000, nodelist=[4], node_color="tab:orange")
+        nx.draw_networkx_edges(G, pos, alpha=0.5, width=6)
+        
+        # Customize axes
+        ax = plt.gca()
+        ax.margins(0.11)
+        plt.tight_layout()
+        plt.axis("off")
+        st.pyplot(plt)
+
+    elif option == "Create your own":
+        # Allow the user to specify node positions and colors
+        st.write("Specify positions for the house graph nodes.")
+
+        positions = {}
+        for i in range(5):
+            x = st.number_input(f"X-coordinate for node {i}:", min_value=-10, max_value=10, value=0)
+            y = st.number_input(f"Y-coordinate for node {i}:", min_value=-10, max_value=10, value=0)
+            positions[i] = (x, y)
+
+        # Allow the user to specify colors for wall and roof nodes
+        wall_color = st.color_picker("Wall color:", "#0000FF")
+        roof_color = st.color_picker("Roof color:", "#FFA500")
+
+        if st.button("Generate"):
+            # Create the house graph with the specified positions
+            G_custom = nx.house_graph()
+
+            # Plot nodes with user-defined properties for wall and roof nodes
+            nx.draw_networkx_nodes(G_custom, positions, node_size=3000, nodelist=[0, 1, 2, 3], node_color=wall_color)
+            nx.draw_networkx_nodes(G_custom, positions, node_size=2000, nodelist=[4], node_color=roof_color)
+            nx.draw_networkx_edges(G_custom, positions, alpha=0.5, width=6)
+
+            # Customize axes
+            ax = plt.gca()
+            ax.margins(0.11)
+            plt.tight_layout()
+            plt.axis("off")
+            st.pyplot(plt)
+
+# Display Drawing: House With Colors if selected
+if sidebar_option == "Drawing: House With Colors":
+    display_house_with_colors()
+
 # Function to display Four Grids visualization for Drawing: Four Grids
 def display_four_grids():
     st.title("Drawing: Four Grids")