Update app.py

app.py

@@ -73,7 +73,7 @@ def generate_dependency_parse(generated_text):
     return html_generated
 
 def display_dependency_parse(generated_text):
-    return generate_dependency_parse(generated_text), gr.update(visible=True)
+    return generate_dependency_parse(generated_text)
 
 def full_interface(prompt, max_new_tokens, top_k, temperature, top_p, repetition_penalty):
     generated_highlight, generated_text = historical_generation(
@@ -81,7 +81,8 @@ def full_interface(prompt, max_new_tokens, top_k, temperature, top_p, repetition
     )
 
     tokens_input, pos_count_input, html_input = text_analysis(prompt)
-    return generated_text, generated_highlight, pos_count_input, html_input, gr.update(visible=True), generated_text, gr.update(visible=False), gr.update(visible=True)
+    # Set the visibility of the generated text and highlight components
+    return generated_text, generated_highlight, pos_count_input, html_input, gr.update(visible=True), "", gr.update(visible=True), gr.update(visible=False)
 
 def reset_interface():
     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)
@@ -106,25 +107,24 @@ with gr.Blocks(theme=gr.themes.Base()) as iface:
     highlighted_text = gr.HighlightedText(label="🎅🏻⌚Tokenized", combine_adjacent=True, show_legend=True)
     tokenizer_info = gr.JSON(label="📉Tokenizer Info (Input Text)")
     dependency_parse_input = gr.HTML(label="👁️Visualization")
-    
-    send_button = gr.Button(value="🎅🏻⌚OCRonos-Vintage 👁️Visualization", visible=False)
     dependency_parse_generated = gr.HTML(label="Dependency Parse Visualization (Generated Text)", visible=False)
     
-    send_button.click(
-        display_dependency_parse, 
-        inputs=[generated_text_output], 
-        outputs=[dependency_parse_generated]
-    )
-    
+    send_button = gr.Button(value="🎅🏻⌚OCRonos-Vintage 👁️Visualization", visible=False)
     reset_button = gr.Button(value="♻️Start Again", visible=False)
 
     generate_button = gr.Button(value="🎅🏻⌚Generate Historical Text")
     generate_button.click(
         full_interface, 
         inputs=[prompt, max_new_tokens, top_k, temperature, top_p, repetition_penalty], 
-        outputs=[generated_text_output, highlighted_text, tokenizer_info, dependency_parse_input, send_button, generated_text_output, generate_button, reset_button]
+        outputs=[generated_text_output, highlighted_text, tokenizer_info, dependency_parse_input, send_button, dependency_parse_generated, generate_button, reset_button]
     )
 
+    send_button.click(
+        display_dependency_parse, 
+        inputs=[generated_text_output], 
+        outputs=[dependency_parse_generated]
+    )
+    
     reset_button.click(
         reset_interface, 
         inputs=None,