nanom commited on
Commit
6bd90ad
·
1 Parent(s): ddd94be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -12,30 +12,30 @@ def postag(tk):
12
  tag = ""
13
  plural_tags = ["NNS", "NNPS"]
14
  if tk.tag_ in plural_tags:
15
- tag = " (Plural)"
16
  else:
17
- tag = " ({})".format(tk.tag_)
18
  return tag
19
 
20
  def genFlatDepthTree(expr):
21
  doc = nlp(expr)
22
  root = next(doc.sents).root
23
- node = Node(root.text+"\t (Root)"+postag(root)+" - [0]", parent=None)
24
 
25
  def tree(tk, last_node, depth):
26
  if tk.n_lefts + tk.n_rights > 0:
27
  for ch in tk.children:
28
- tree(ch, Node(ch.text+"\t"+postag(ch)+" - ["+str(depth+1)+"]", parent=last_node), depth+1)
29
 
30
  tree(root, node, 0)
31
  flat_tree = ""
32
 
33
  for pre, fill, node in RenderTree(node):
34
- flat_tree += """{}{}\n""".format(pre, node.name)
35
 
36
  img_tree = displacy.render(doc, style='dep', options={'distance': 100})
37
 
38
- return "<center><div style='max-width: 800px; overflow-x:auto;'>"+img_tree+"</div></center>", flat_tree
39
 
40
  # def syntacticTree(expr):
41
  # doc = nlp(expr)
@@ -64,7 +64,7 @@ with demo:
64
  gr.Image(value="https://img.unocero.com/2019/11/facebook-app-para-hacer-memes-1-1024x576.jpg",label="", type="URL")
65
  with gr.Column():
66
  input_expr = gr.Textbox(label="Input", placeholder="Enter an expression here")
67
- out_flat_tree = gr.Text(label="Flat tree", value="")
68
 
69
  out_image_tree = gr.HTML(label="")
70
 
 
12
  tag = ""
13
  plural_tags = ["NNS", "NNPS"]
14
  if tk.tag_ in plural_tags:
15
+ tag = " <b>(Plural)</b>"
16
  else:
17
+ tag = " <b>({})</b>".format(tk.tag_)
18
  return tag
19
 
20
  def genFlatDepthTree(expr):
21
  doc = nlp(expr)
22
  root = next(doc.sents).root
23
+ node = Node(root.text+":(Root) "+postag(root), parent=None)
24
 
25
  def tree(tk, last_node, depth):
26
  if tk.n_lefts + tk.n_rights > 0:
27
  for ch in tk.children:
28
+ tree(ch, Node(ch.text+":"+str(depth+1)+postag(ch), parent=last_node), depth+1)
29
 
30
  tree(root, node, 0)
31
  flat_tree = ""
32
 
33
  for pre, fill, node in RenderTree(node):
34
+ flat_tree += """{}{}</br>""".format(pre, node.name)
35
 
36
  img_tree = displacy.render(doc, style='dep', options={'distance': 100})
37
 
38
+ return "<center><div style='max-width: 800px; overflow-x:auto;'>"+img_tree+"</div></center>", "<div>"+flat_tree+"</div>"
39
 
40
  # def syntacticTree(expr):
41
  # doc = nlp(expr)
 
64
  gr.Image(value="https://img.unocero.com/2019/11/facebook-app-para-hacer-memes-1-1024x576.jpg",label="", type="URL")
65
  with gr.Column():
66
  input_expr = gr.Textbox(label="Input", placeholder="Enter an expression here")
67
+ out_flat_tree = gr.HTML(label="Flat tree", value="")
68
 
69
  out_image_tree = gr.HTML(label="")
70