update buttons

app.py

@@ -99,6 +99,39 @@ def clear_graph():
     lesson_graph.clear()
     return "Landscape cleared. You can start a new lesson plan."
 
+import json
+
+def graph_to_json():
+    # Convert the lesson_graph to a dictionary format
+    graph_data = {
+        "nodes": [
+            {
+                "id": node,
+                "type": lesson_graph.nodes[node]["type"],
+                "description": lesson_graph.nodes[node].get("description", "")
+            }
+            for node in lesson_graph.nodes()
+        ],
+        "edges": [
+            {
+                "source": u,
+                "target": v,
+                "relationship": lesson_graph.edges[u, v]["relationship"]
+            }
+            for u, v in lesson_graph.edges()
+        ]
+    }
+    
+    # Convert dictionary to JSON string
+    json_data = json.dumps(graph_data, indent=4)
+    
+    # Save to a file
+    file_name = "/mnt/data/lesson_graph.json"
+    with open(file_name, "w") as f:
+        f.write(json_data)
+    
+    return file_name
+
 # Gradio interface
 demo = gr.Blocks()
 
@@ -143,6 +176,14 @@ with demo:
         inputs=[teacher_name, subject, grade_level, learning_objective, activity, assessment, resource, school_board],
         outputs=[search_output, graph_output]
     )
+
+    # Adding the button to download the lesson graph as JSON
+    download_btn = gr.Button("Download Lesson Graph (JSON)")
+    
+    download_btn.click(
+        graph_to_json,
+        outputs=gr.File(label="Download JSON File")
+    )
     
     clear_btn.click(clear_graph, outputs=message_output)