update generation message when run completes

app.py

@@ -64,7 +64,7 @@ with gr.Blocks(title="RFD3 Test") as demo:
             )
 
             config_validation_btn = gr.Button("Validate Config")
-            config_textbox = gr.Textbox(value ="Waiting for config validation...")
+            config_textbox = gr.Textbox(label="Configuration status", value ="Waiting for config validation...")
 
             with gr.Accordion(label="Advanced Options", open=False):
                 extra_args = gr.Textbox(
@@ -80,9 +80,9 @@ with gr.Blocks(title="RFD3 Test") as demo:
             with gr.Row():
                 scaffold_validation_btn = gr.Button("Validate Scaffold")
                 no_input_btn = gr.Button("No Scaffold/Target")
-            scaffold_textbox = gr.Textbox(value ="Waiting for scaffold validation...")
+            scaffold_textbox = gr.Textbox(label="Scaffold/Target status", value ="Waiting for scaffold validation...")
     run_btn = gr.Button("Run Generation", variant="primary")
-    runtextbox = gr.Textbox(value="Waiting for generation run...")
+    runtextbox = gr.Textbox(label="Run status", value="Waiting for generation run...")
 
 
 
@@ -119,7 +119,7 @@ with gr.Blocks(title="RFD3 Test") as demo:
         else:
             return None, None
         
-    run_btn.click(give_run_status, inputs=[config_ready, scaffold_ready, num_batches, num_designs_per_batch, config_upload], outputs=runtextbox).then(
+    run_btn.click(give_run_status, inputs=[config_ready, scaffold_ready, num_batches, num_designs_per_batch, config_upload], outputs=runtextbox.then(
         generate, inputs=[config_ready, scaffold_ready, num_batches, num_designs_per_batch, config_upload, extra_args], outputs=[gen_directory, gen_results]
     ).then(
         update_batch_choices,
@@ -129,6 +129,10 @@ with gr.Blocks(title="RFD3 Test") as demo:
         download_results_as_zip,
         inputs=gen_directory,
         outputs=output_file
+    ).then(
+        lambda gen_results: gr.update(value="Generation completed!") if gen_results is not None else gr.update(), 
+        inputs=gen_results,
+        outputs=runtextbox
     )