Show graph visualization for bolt queries (#761)

* Add graph visualization support for bolt protocol requests

Fixes #557 by converting bolt responses to compatible format
for graph tab display. Transforms nodes and relationships to
standardized format with ~id, ~entityType, ~labels/~type, and
~properties field.

* Add graph visualization support for bolt protocol requests

Fixes #557 by converting bolt responses to compatible format
for graph tab display. Transforms nodes and relationships to
standardized format with ~id, ~entityType, ~labels/~type, and
~properties field.

* Update graph_magic.py

---------

Co-authored-by: Neel Shah <neeljs@amazon.com>

Author: theneelshah

src/graph_notebook/magics/graph_magic.py

@@ -3764,7 +3764,33 @@ def handle_opencypher_query(self, line, cell, local_ns):
                 results_df, has_results = oc_results_df(res, res_format)
                 if has_results:
                     titles.append('Console')
-                # Need to eventually add code to parse and display a network for the bolt format here
+                # Create graph visualization for bolt response
+                try:
+                    # Wrap bolt response in expected format
+                    # Required because the graph visualizer need the data to be present in a certain format 
+                    transformed_res = {"results": res} if isinstance(res, list) else {"results": []}
+                    
+                    gn = OCNetwork(group_by_property=args.group_by, display_property=args.display_property,
+                                   group_by_raw=args.group_by_raw,
+                                   group_by_depth=args.group_by_depth,
+                                   edge_display_property=args.edge_display_property,
+                                   tooltip_property=args.tooltip_property,
+                                   edge_tooltip_property=args.edge_tooltip_property,
+                                   label_max_length=args.label_max_length,
+                                   edge_label_max_length=args.rel_label_max_length,
+                                   ignore_groups=args.ignore_groups)
+                    gn.add_results(transformed_res)
+                    logger.debug(f'number of nodes is {len(gn.graph.nodes)}')
+                    if len(gn.graph.nodes) > 0:
+                        self.graph_notebook_vis_options['physics']['disablePhysicsAfterInitialSimulation'] \
+                            = args.stop_physics
+                        self.graph_notebook_vis_options['physics']['simulationDuration'] = args.simulation_duration
+                        force_graph_output = Force(network=gn, options=self.graph_notebook_vis_options)
+                        titles.append('Graph')
+                        children.append(force_graph_output)
+                except (TypeError, ValueError) as network_creation_error:
+                    logger.debug(f'Unable to create network from bolt result. Skipping from result set: {res}')
+                    logger.debug(f'Error: {network_creation_error}')
 
         if not args.silent:
             if args.mode != 'explain':

src/graph_notebook/neptune/client.py

@@ -606,7 +606,34 @@ def opencyper_bolt(self, query: str, **kwargs):
         with driver.session(database=self.neo4j_database) as session:
             try:
                 res = session.run(query, kwargs)
-                data = res.data()
+                data = []
+                for record in res:
+                    record_dict = {}
+                    for key in record.keys():
+                        value = record[key]
+                        if hasattr(value, 'id') and (hasattr(value, 'labels') or hasattr(value, 'type')):  # Node or Relationship
+                            if hasattr(value, 'labels'):  # Node
+                                record_dict[key] = {
+                                    '~id': str(value.id),
+                                    '~entityType': 'node',
+                                    '~labels': list(value.labels),
+                                    '~properties': dict(value)
+                                }
+                            elif hasattr(value, 'type'):  # Relationship
+                                
+                                start_node = value.nodes[0] if value.nodes else None
+                                end_node = value.nodes[1] if len(value.nodes) > 1 else None
+                                record_dict[key] = {
+                                    '~id': str(value.id),
+                                    '~entityType': 'relationship',
+                                    '~start': str(start_node.id) if start_node else '',
+                                    '~end': str(end_node.id) if end_node else '',
+                                    '~type': value.type,
+                                    '~properties': dict(value)
+                                }
+                        else:
+                            record_dict[key] = value
+                    data.append(record_dict)
             except AuthError:
                 print("Neo4J Bolt request failed with an authentication error. Please ensure that the 'neo4j' section "
                       "of your %graph_notebook_config contains the correct credentials and auth setting.")

src/graph_notebook/network/opencypher/OCNetwork.py

@@ -26,8 +26,7 @@
 
 class OCNetwork(EventfulNetwork):
     """OCNetwork extends the EventfulNetwork class and uses the add_results method to parse any response that returns
-    nodes/relationships as part (or all) of the response.  Currently this only works with HTTPS response format but in
-    the future, we will work to support Bolt based responses as well.
+    nodes/relationships as part (or all) of the response.
     """
 
     def __init__(self, graph: MultiDiGraph = None, callbacks=None, label_max_length=DEFAULT_LABEL_MAX_LENGTH,