Update app.py

app.py

@@ -440,15 +440,26 @@ def generate_tree(original_sentence: str) -> str:
     masked_sentence = mask_non_stopword(first_paraphrased_sentence)
     masked_versions = mask(masked_sentence)
     dot = graphviz.Digraph()
-    dot.attr(rankdir='LR', size='8,10!', dpi='72')
-    dot.node("Original", original_sentence)
-    dot.node("Paraphrased", first_paraphrased_sentence)
-    dot.edge("Original", "Paraphrased")
-    for i, masked in enumerate(masked_versions):
-        node_id = f"Masked_{i}"
-        dot.node(node_id, masked)
-        dot.edge("Paraphrased", node_id)
-    return masked_sentence, dot.source
+    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
 
 # Function for the Gradio interface
 def model(prompt):
@@ -459,11 +470,11 @@ def model(prompt):
     for i in range(len(common_subs)):
         common_subs[i]["Paraphrased Sentence"] = res[i]
     result = highlight_phrases_with_colors(res, common_grams)
-    masked_sentence, tree_source = generate_tree(sentence)
+    masked_sentence, masked_versions, tree_source = generate_tree(sentence)
     graph = graphviz.Source(tree_source)
-    svg_content = graph.pipe(format='svg').decode('utf-8')
+    png_content = graph.render(filename='paraphrase_tree_dynamic', format='png')
     # tree = f'<div style="width: 100%; overflow-x: auto;">{svg_content}</div>'
-    return generated, generated, result, masked_sentence, svg_content
+    return generated, generated, result, masked_sentence, masked_versions, png_content
 
 with gr.Blocks(theme = gr.themes.Monochrome()) as demo:
     gr.Markdown("# Paraphrases the Text and Highlights the Non-melting Points")
@@ -488,12 +499,15 @@ with gr.Blocks(theme = gr.themes.Monochrome()) as demo:
         masked_sentence = gr.Textbox(label="Masked Sentence")
 
     with gr.Row():
-        tree = gr.HTML(label="Tree")
+        masked_versions = gr.Textbox(label="Sentence Generated by Masking Model")
+
+    with gr.Row():
+        tree = gr.Image(label="Paraphrase Tree")
 
-    submit_button.click(model, inputs=user_input, outputs=[ai_output, selected_sentence, html_output, masked_sentence, tree])
+    submit_button.click(model, inputs=user_input, outputs=[ai_output, selected_sentence, html_output, masked_sentence, masked_versions, tree])
     clear_button.click(lambda: "", inputs=None, outputs=user_input)
-    clear_button.click(lambda: "", inputs=None, outputs=[ai_output, selected_sentence, html_output, masked_sentence, tree])
+    clear_button.click(lambda: "", inputs=None, outputs=[ai_output, selected_sentence, html_output, masked_sentence, masked_versions, tree])
 
 # Launch the demo
-demo.launch()
+demo.launch(share=True)