add adjacency matrix

app.py

@@ -130,9 +130,10 @@ def plot_graph_sentence(sentence, graph_type="both"):
   elif graph_type == "both":
     graphs = construct_both_graph(docs)
   g = to_jraph(graphs[0])
+  adj_mat = get_adjacency_matrix(g)
   nx_graph = convert_jraph_to_networkx_graph(g)
   pos = half_circle_layout(len(graphs[0]["nodes"]))
-  plot = plt.figure(figsize=(25, 6))
+  plot = plt.figure(figsize=(12, 6))
   nx.draw(nx_graph, pos=pos,
           labels={i: e for i,e in enumerate(graphs[0]["nodes"])},
           with_labels = True, edge_color="blue",
@@ -142,17 +143,33 @@ def plot_graph_sentence(sentence, graph_type="both"):
     nx_graph, pos=pos,
     edge_labels=graphs[0]["edge_labels"],
     font_color='red'
-)
-  return plot
+  )
+  adj_mat_plot, ax = plt.subplots(figsize=(6, 6))
+  ax.matshow(adj_mat)
+  return [gr.update(value=plot), gr.update(value=adj_mat_plot)]
     
 def get_list_sentences(id):
+  id =  int(min(id, len(dataset["train"]) - 1))
   return gr.update(choices = dataset["train"][id]["transcript"].split("."))
 
 with gr.Blocks() as demo:
-    id = gr.Slider(maximum=len(dataset["train"]) - 1, label="Record #")
-    sentence = gr.Dropdown(label="Transcript sentence", choices = dataset["train"][0]["transcript"].split("."), interactive = True)
-    plot = gr.Plot(label="Dependency graph")
-    id.change(get_list_sentences, id, sentence)
-    sentence.change(plot_graph_sentence, sentence, plot)
+  with gr.Tab("From transcript"):
+    with gr.Row():
+      with gr.Column():
+        id = gr.Number(label="Transcript")
+      with gr.Column(scale=3):
+        sentence_transcript = gr.Dropdown(label="Sentence", choices = dataset["train"][0]["transcript"].split("."), interactive = True)
+  with gr.Tab("Type sentence"):
+    with gr.Row():
+      sentence_typed = gr.Textbox(label="Sentence", interactive = True)
+  with gr.Row():
+    with gr.Column(scale=2):
+      plot_graph = gr.Plot(label="Word graph")
+    with gr.Column():
+      plot_adj = gr.Plot(label="Word graph adjacency matrix")
+
+  id.change(get_list_sentences, id, sentence)  
+  sentence_transcript.change(plot_graph_sentence, sentence_transcript, [plot_graph, plot_adj])
+  sentence_typed.change(plot_graph_sentence, sentence_typed, [plot_graph, plot_adj])
 
 demo.launch()