update graph to json

app.py

@@ -101,36 +101,39 @@ def clear_graph():
 
 import json
 
+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,
+    try:
+        # 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
+                    "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)
+        
+        # Instead of writing to a file, return the JSON data as a message
+        return f"Graph Data in JSON Format:\n{json_data}"
+
+    except Exception as e:
+        # If any error occurs, catch it and display it in the message_output
+        return f"An error occurred: {str(e)}"
 
 # Gradio interface
 demo = gr.Blocks()