Update app.py

app.py

@@ -36,25 +36,24 @@ def reshape_json_data_to_fit_visualize_graph(graph_data):
     return graph_data
 
 
+
 def display_example(index):
     example = dataset[index]
-    # print("This is the example: ")
-    # print(example)
-    # Get the image
     img = example["image"]
 
     # Prepare the graph data
     graph_data = {"nodes": example["nodes"], "edges": example["edges"]}
 
-    # # Convert graph_data to JSON string
-    # json_data = json.dumps(graph_data)
     transformed_graph_data = reshape_json_data_to_fit_visualize_graph(graph_data)
 
-    # print(json_data)
     # Generate the graph visualization
     graph_html = visualize_graph(transformed_graph_data)
 
-    return img, graph_html
+    # Convert graph_data to a formatted JSON string
+    json_data = json.dumps(transformed_graph_data, indent=2)
+
+    return img, graph_html, json_data, transformed_graph_data
+
 
 
 def create_interface():
@@ -73,7 +72,7 @@ def create_interface():
         index_slider.change(
             fn=display_example,
             inputs=[index_slider],
-            outputs=[image_output, graph_output],
+            outputs=[image_output, graph_output, json_output, text_output],
         )
 
     return demo