Update app.py

app.py

@@ -126,6 +126,7 @@ def create_graph(entities: list[str], relationships: list[tuple[str, str, str]])
         A networkx Graph object.
     """
     logging.debug(f"create_graph called with entities: {entities}, relationships: {relationships}")
+    print(f"create_graph input:\nentities: {entities}\nrelationships: {relationships}")
     try:
         G = nx.Graph()
         G.add_nodes_from(entities)
@@ -138,6 +139,9 @@ def create_graph(entities: list[str], relationships: list[tuple[str, str, str]])
             else:
                 logging.warning(f"Edge ({subject}, {object_}) not found in the graph.")
 
+        if not nx.is_graph(G):  # Add this check
+            logging.error("Error: create_graph did not create a valid networkx graph")
+            return nx.Graph()  # Return an empty graph
         return G
     except Exception as e:
         error_message = f"Error in create_graph: {str(e)}"
@@ -155,6 +159,10 @@ import networkx as nx
 def create_bokeh_plot(graph: nx.Graph, layout_type: str):
     """Creates a Bokeh plot of the given networkx graph."""
     try:
+        if not nx.is_graph(graph): # Add this check
+            logging.error("Error: create_bokeh_plot received an invalid networkx graph")
+            return None # Or return a placeholder plot
+
         if layout_type == 'spring':
             pos = nx.spring_layout(graph, seed=42)
         elif layout_type == 'fruchterman_reingold':
@@ -201,70 +209,73 @@ def create_bokeh_plot(graph: nx.Graph, layout_type: str):
 def create_plotly_plot(graph: nx.Graph, layout_type: str):
     """Creates a Plotly plot of the given networkx graph."""
     try:
-      import plotly.graph_objects as go
-
-      if layout_type == 'spring':
-          pos = nx.spring_layout(graph, seed=42)
-      elif layout_type == 'fruchterman_reingold':
-          pos = nx.fruchterman_reingold_layout(graph, seed=42)
-      elif layout_type == 'circular':
-          pos = nx.circular_layout(graph)
-      elif layout_type == 'random':
-          pos = nx.random_layout(graph, seed=42)
-      elif layout_type == 'spectral':
-          pos = nx.spectral_layout(graph)
-      elif layout_type == 'shell':
-          pos = nx.shell_layout(graph)
-      else:
-          pos = nx.spring_layout(graph, seed=42)  # Default layout
-
-      edge_x = []
-      edge_y = []
-      for edge in graph.edges():
-          x0, y0 = pos[edge[0]]
-          x1, y1 = pos[edge[1]]
-          edge_x.append(x0)
-          edge_y.append(y0)
-          edge_x.append(x1)
-          edge_y.append(y1)
-          edge_x.append(None)
-          edge_y.append(None)
-
-      edge_trace = go.Scatter(
-          x=edge_x, y=edge_y,
-          line=dict(width=0.5, color='#888'),
-          hoverinfo='none',
-          mode='lines')
-
-      node_x = []
-      node_y = []
-      for node in graph.nodes():
-          x, y = pos[node]
-          node_x.append(x)
-          node_y.append(y)
-
-      node_trace = go.Scatter(
-          x=node_x, y=node_y,
-          mode='markers',
-          hoverinfo='text',
-          marker=dict(
-              showscale=False,
-              colorscale='YlGnBu',
-              reversescale=True,
-              color=[],
-              size=10,
-              line_width=2))
-
-      node_adjacencies = []
-      node_text = []
-      for node, adjacencies in enumerate(graph.adjacency()):
-          node_adjacencies.append(len(adjacencies[1]))
-          node_text.append(f"{adjacencies[0]} (# of connections: {len(adjacencies[1])})")
-
-      node_trace.marker.color = node_adjacencies
-      node_trace.text = node_text
-
-      fig = go.Figure(data=[edge_trace, node_trace],
+        if not nx.is_graph(graph): # Add this check
+            logging.error("Error: create_plotly_plot received an invalid networkx graph")
+            return None # Or return a placeholder plot
+        import plotly.graph_objects as go
+
+        if layout_type == 'spring':
+            pos = nx.spring_layout(graph, seed=42)
+        elif layout_type == 'fruchterman_reingold':
+            pos = nx.fruchterman_reingold_layout(graph, seed=42)
+        elif layout_type == 'circular':
+            pos = nx.circular_layout(graph)
+        elif layout_type == 'random':
+            pos = nx.random_layout(graph, seed=42)
+        elif layout_type == 'spectral':
+            pos = nx.spectral_layout(graph)
+        elif layout_type == 'shell':
+            pos = nx.shell_layout(graph)
+        else:
+            pos = nx.spring_layout(graph, seed=42)  # Default layout
+
+        edge_x = []
+        edge_y = []
+        for edge in graph.edges():
+            x0, y0 = pos[edge[0]]
+            x1, y1 = pos[edge[1]]
+            edge_x.append(x0)
+            edge_y.append(y0)
+            edge_x.append(x1)
+            edge_y.append(y1)
+            edge_x.append(None)
+            edge_y.append(None)
+
+        edge_trace = go.Scatter(
+            x=edge_x, y=edge_y,
+            line=dict(width=0.5, color='#888'),
+            hoverinfo='none',
+            mode='lines')
+
+        node_x = []
+        node_y = []
+        for node in graph.nodes():
+            x, y = pos[node]
+            node_x.append(x)
+            node_y.append(y)
+
+        node_trace = go.Scatter(
+            x=node_x, y=node_y,
+            mode='markers',
+            hoverinfo='text',
+            marker=dict(
+                showscale=False,
+                colorscale='YlGnBu',
+                reversescale=True,
+                color=[],
+                size=10,
+                line_width=2))
+
+        node_adjacencies = []
+        node_text = []
+        for node, adjacencies in enumerate(graph.adjacency()):
+            node_adjacencies.append(len(adjacencies[1]))
+            node_text.append(f"{adjacencies[0]} (# of connections: {len(adjacencies[1])})")
+
+        node_trace.marker.color = node_adjacencies
+        node_trace.text = node_text
+
+        fig = go.Figure(data=[edge_trace, node_trace],
                   layout=go.Layout(
                       title='Knowledge Graph',
                       titlefont_size=16,
@@ -280,7 +291,7 @@ def create_plotly_plot(graph: nx.Graph, layout_type: str):
                       yaxis=dict(showgrid=False, zeroline=False, showticklabels=False))
                       )
 
-      return fig
+        return fig
 
     except Exception as e:
         error_message = f"Error creating Plotly plot: {str(e)}"
@@ -410,9 +421,11 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
 
     def process_and_update(text: str, entity_types: str, predicates: str, layout_type: str, visualization_type: str):
         G, fig, output = process_text(text, entity_types, predicates, layout_type, visualization_type)
+        print(f"process_and_update: G = {G}") # Debug
         return G, fig, output
 
     def update_graph_wrapper(G: nx.Graph, layout_type: str, visualization_type: str):
+        print(f"update_graph_wrapper: G = {G}") # Debug
         if G is not None:
             fig, _ = update_graph(G, layout_type, visualization_type)
             return fig