Update app.py

app.py

@@ -19,7 +19,7 @@ model.to(device)
 
 # Load spaCy model for dependency parsing
 os.system('python -m spacy download en_core_web_sm')
-nlp = spacy.load("en_core_web_sm") ## add latin-spacy
+nlp = spacy.load("en_core_web_sm")
 
 # Function for generating text and tokenizing
 def historical_generation(prompt, max_new_tokens=600):
@@ -56,7 +56,7 @@ def historical_generation(prompt, max_new_tokens=600):
     # Create highlighted text output
     highlighted_text = []
     for token in tokens:
-        clean_token = token.replace("Ġ", "")
+        clean_token = token.replace("Ġ", "")  # Remove "Ġ"
         token_type = tokenizer.convert_ids_to_tokens([tokenizer.convert_tokens_to_ids(token)])[0]
         highlighted_text.append((clean_token, token_type))
 
@@ -92,7 +92,11 @@ def full_interface(prompt, max_new_tokens):
     tokens_input, pos_count_input, html_input = text_analysis(prompt)
 
     # The "Send" button should now appear after these outputs are generated
-    return generated_highlight, pos_count_input, html_input, gr.update(visible=True), generated_text
+    return generated_highlight, pos_count_input, html_input, gr.update(visible=True), generated_text, gr.update(visible=False), gr.update(visible=True)
+
+# Reset function to restore button states
+def reset_interface():
+    return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
 
 # Gradio interface components
 with gr.Blocks() as iface:
@@ -108,6 +112,9 @@ with gr.Blocks() as iface:
     send_button = gr.Button(value="Generate Dependency Parse for Generated Text", visible=False)
     dependency_parse_generated = gr.HTML(label="Dependency Parse Visualization (Generated Text)")
 
+    # Reset button, hidden initially
+    reset_button = gr.Button(value="Start Again", visible=False)
+
     # Button behavior for generating final parse visualization
     send_button.click(
         generate_dependency_parse, 
@@ -115,11 +122,19 @@ with gr.Blocks() as iface:
         outputs=[dependency_parse_generated]
     )
 
-    # Main interface logic
-    prompt.submit(
+    # Main interface logic: when clicked, "Generate" button hides itself and shows the reset button
+    generate_button = gr.Button(value="Generate Text and Initial Outputs")
+    generate_button.click(
         full_interface, 
         inputs=[prompt, max_new_tokens], 
-        outputs=[highlighted_text, tokenizer_info, dependency_parse_input, send_button, dependency_parse_generated]
+        outputs=[highlighted_text, tokenizer_info, dependency_parse_input, send_button, dependency_parse_generated, generate_button, reset_button]
+    )
+
+    # Reset button logic: hide itself and re-show the "Generate" button
+    reset_button.click(
+        reset_interface, 
+        inputs=None, 
+        outputs=[generate_button, send_button, reset_button]
     )
 
-iface.launch()
+iface.launch()