Update app.py

app.py

@@ -434,33 +434,7 @@ def mask(sentence):
     masked_sentences = [predictions[i]['sequence'] for i in range(len(predictions))]
     return masked_sentences
 
-# Function to generate the tree and return the Graphviz source
-# def generate_tree(original_sentence: str) -> str:
-#     paraphrased_sentences = generate_paraphrase(original_sentence)
-#     first_paraphrased_sentence = paraphrased_sentences[0]
-#     masked_sentence = mask_non_stopword(first_paraphrased_sentence)
-#     masked_versions = mask(masked_sentence)
-#     dot = graphviz.Digraph()
-#     dot.attr(rankdir='LR', size='10,10', dpi=' 2743')
 
-#     existing_nodes = set()
-
-#     def add_paraphrases(parent, paraphrases):
-#         if parent not in existing_nodes:
-#             dot.node(parent, parent, shape='box')
-#             existing_nodes.add(parent)
-
-#         for paraphrase in paraphrases:
-#             if paraphrase not in existing_nodes:
-#                 dot.node(paraphrase, paraphrase, shape='box')
-#                 existing_nodes.add(paraphrase)
-#             dot.edge(parent, paraphrase)
-
-#     add_paraphrases(original_sentence, paraphrased_sentences) #whenever a new branch is to be created call this function along with the original sentence and the list of its paraphrases
-#     add_paraphrases(paraphrased_sentences[0], masked_versions)
-
-#     graph_path = dot.render(filename='paraphrase_tree_dynamic', format='png')
-#     return masked_sentence, masked_versions, dot.source
 
 
 
@@ -483,7 +457,8 @@ def generate_plot(original_sentence):
         (1, 6),
         (1, 7),
         (1, 8),
-        (1, 9)
+        (1, 9),
+        (1, 10)
     ]
     wrapped_nodes = ['<br>'.join(textwrap.wrap(node, width=30)) for node in nodes]
     fig = go.Figure()