Update app.py

app.py

@@ -72,9 +72,8 @@ def generate_dependency_parse(generated_text):
     tokens_generated, pos_count_generated, html_generated = text_analysis(generated_text)
     return html_generated
 
-def generate_dependency_parse(generated_text):
-    tokens_generated, pos_count_generated, html_generated = text_analysis(generated_text)
-    return html_generated
+def display_dependency_parse(generated_text):
+    return generate_dependency_parse(generated_text), gr.update(visible=True)
 
 def full_interface(prompt, max_new_tokens, top_k, temperature, top_p, repetition_penalty):
     generated_highlight, generated_text = historical_generation(
@@ -85,7 +84,7 @@ def full_interface(prompt, max_new_tokens, top_k, temperature, top_p, repetition
     return generated_text, generated_highlight, pos_count_input, html_input, gr.update(visible=True), generated_text, gr.update(visible=False), gr.update(visible=True)
 
 def reset_interface():
-    return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+    return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
 
 with gr.Blocks(theme=gr.themes.Base()) as iface:  
 
@@ -112,9 +111,9 @@ with gr.Blocks(theme=gr.themes.Base()) as iface:
     dependency_parse_generated = gr.HTML(label="Dependency Parse Visualization (Generated Text)", visible=False)
     
     send_button.click(
-        generate_dependency_parse, 
+        display_dependency_parse, 
         inputs=[generated_text_output], 
-        outputs=[dependency_parse_generated, gr.update(visible=True)]
+        outputs=[dependency_parse_generated]
     )
     
     reset_button = gr.Button(value="♻️Start Again", visible=False)
@@ -129,7 +128,7 @@ with gr.Blocks(theme=gr.themes.Base()) as iface:
     reset_button.click(
         reset_interface, 
         inputs=None, 
-        outputs=[generate_button, send_button, reset_button, generated_text_output, dependency_parse_generated]
+        outputs=[generate_button, send_button, reset_button, generated_text_output, highlighted_text, tokenizer_info, dependency_parse_input, dependency_parse_generated]
     )
 
 iface.launch()