update app

app.py

@@ -3,13 +3,29 @@ import networkx as nx
 import matplotlib.pyplot as plt
 from io import BytesIO
 from PIL import Image
+import matplotlib.patches as mpatches
 
 # Initialize the lesson plan graph
 lesson_graph = nx.DiGraph()
 
+# Define color map for node types
+color_map = {
+    "User": "#FF9999",  # Light Red
+    "Subject": "#66B2FF",  # Light Blue
+    "Grade Level": "#99FF99",  # Light Green
+    "Learning Objective": "#FFCC99",  # Light Orange
+    "Activity": "#FF99FF",  # Light Purple
+    "Assessment": "#FFFF99",  # Light Yellow
+    "Resource": "#99FFFF",  # Light Cyan
+    "School Board": "#CCCCCC"  # Light Gray
+}
+
 def add_to_graph(teacher_name, subject, grade_level, learning_objective, activity, assessment, resource, school_board):
     global lesson_graph
     
+    # Clear previous graph
+    lesson_graph.clear()
+    
     # Add nodes to the graph
     lesson_graph.add_node(teacher_name, type="User")
     lesson_graph.add_node(subject, type="Subject")
@@ -34,18 +50,28 @@ def add_to_graph(teacher_name, subject, grade_level, learning_objective, activit
     search_string = f"{subject} {grade_level} {learning_objective} {activity} {resource}".strip()
     
     # Visualize the graph
-    plt.figure(figsize=(12, 8))
-    pos = nx.spring_layout(lesson_graph)
-    nx.draw(lesson_graph, pos, with_labels=True, node_color="lightblue", 
-            font_size=8, font_weight="bold", node_size=3000, node_shape="o")
+    plt.figure(figsize=(14, 10))
+    pos = nx.spring_layout(lesson_graph, k=0.9, iterations=50)
     
-    # Add node labels
-    labels = nx.get_node_attributes(lesson_graph, 'type')
-    nx.draw_networkx_labels(lesson_graph, pos, labels, font_size=6)
+    # Draw nodes with color coding
+    for node, node_type in nx.get_node_attributes(lesson_graph, 'type').items():
+        nx.draw_networkx_nodes(lesson_graph, pos, nodelist=[node], node_color=color_map[node_type], 
+                               node_size=3000, alpha=0.8)
+    
+    nx.draw_networkx_edges(lesson_graph, pos, edge_color='gray', arrows=True, arrowsize=20)
+    nx.draw_networkx_labels(lesson_graph, pos, font_size=8, font_weight="bold")
     
     # Add edge labels
     edge_labels = nx.get_edge_attributes(lesson_graph, 'relationship')
-    nx.draw_networkx_edge_labels(lesson_graph, pos, edge_labels=edge_labels, font_size=6)
+    nx.draw_networkx_edge_labels(lesson_graph, pos, edge_labels=edge_labels, font_size=7)
+    
+    # Create legend
+    legend_elements = [mpatches.Patch(color=color, label=node_type) for node_type, color in color_map.items()]
+    plt.legend(handles=legend_elements, loc='upper left', bbox_to_anchor=(1, 1), title="Node Types")
+    
+    plt.title("Your Educational Landscape", fontsize=16)
+    plt.axis('off')
+    plt.tight_layout()
     
     # Save the plot to a bytes object
     buf = BytesIO()
@@ -61,14 +87,14 @@ def add_to_graph(teacher_name, subject, grade_level, learning_objective, activit
 def clear_graph():
     global lesson_graph
     lesson_graph.clear()
-    return "Canvas cleared. You can start a new lesson plan."
+    return "Landscape cleared. You can start a new lesson plan."
 
 # Gradio interface
 demo = gr.Blocks()
 
 with demo:
-    gr.Markdown("# EduCanvas: Craft Your Lesson Masterpiece")
-    gr.Markdown("Welcome to EduCanvas, where lesson planning becomes an art. Design, visualize, and perfect your educational masterpieces with ease.")
+    gr.Markdown("# EduScape: Design Your Educational Landscape")
+    gr.Markdown("Welcome to EduScape, where lesson planning becomes an adventure in crafting educational journeys. Design, visualize, and perfect your learning landscapes with ease.")
     
     with gr.Row():
         teacher_name = gr.Textbox(label="Teacher Name")
@@ -87,12 +113,12 @@ with demo:
         resource = gr.Textbox(label="Resource/Material")
     
     with gr.Row():
-        generate_btn = gr.Button("Paint Your Lesson Plan")
-        clear_btn = gr.Button("Clear Canvas")
+        generate_btn = gr.Button("Map Your Lesson Plan")
+        clear_btn = gr.Button("Clear Landscape")
     
     search_output = gr.Textbox(label="Content Discovery Search String")
-    graph_output = gr.Image(label="Your Lesson Masterpiece")
-    message_output = gr.Textbox(label="Canvas Status")
+    graph_output = gr.Image(label="Your Educational Landscape")
+    message_output = gr.Textbox(label="Landscape Status")
     
     generate_btn.click(
         add_to_graph, 
@@ -102,5 +128,5 @@ with demo:
     
     clear_btn.click(clear_graph, outputs=message_output)
 
-# Launch the EduCanvas app
+# Launch the EduScape app
 demo.launch()