propagate rfd3 subprocess stderr to run status textbox

app.py

@@ -141,13 +141,13 @@ with gr.Blocks(title="RFD3 Test") as demo:
     
     def generate(config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args):
         if config_upload is None:
-            return None, None
+            return gr.update(), None, None
         else:
-            gen_directory, gen_results = generation_with_input_config(config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args)
-            return gen_directory, gen_results
+            textbox_update, gen_directory, gen_results = generation_with_input_config(config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args)
+            return textbox_update, gen_directory, gen_results
         
     run_btn.click(give_run_status, inputs=[num_batches, num_designs_per_batch, config_upload, scaffold_upload], outputs=runtextbox).then(
-        generate, inputs=[config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args], outputs=[gen_directory, gen_results]
+        generate, inputs=[config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args], outputs=[runtextbox, gen_directory, gen_results]
     ).then(
         update_batch_choices,
         inputs=gen_results,
@@ -179,4 +179,4 @@ with gr.Blocks(title="RFD3 Test") as demo:
     #visualize_btn.click(load_viewer, inputs=[batch_dropdown, design_dropdown, gen_results], outputs=viewer)
 
 if __name__ == "__main__":
-    demo.launch(show_error=True)
+    demo.launch()

utils/pipelines.py

@@ -115,6 +115,8 @@ def generation_with_input_config(input_file, pdb_file, num_batches, num_designs_
     
     Returns:
     -------
+        textbox_update: gr.update,
+            A gr.update object to update the textbox with the status of the generation. propagates subprocess errors to the textbox if the generation fails.
         directory: str,
             The path to the directory where the generated structures are saved.
         results: list of dicts,
@@ -143,7 +145,7 @@ def generation_with_input_config(input_file, pdb_file, num_batches, num_designs_
             command += f" {extra_args}"
         print(f"Running command: {command}")
         start = perf_counter()
-        subprocess.run(command, shell=True, check=True, text=True)
+        res = subprocess.run(command, shell=True, check=True, text=True, capture_output=True)
         print("Command took", perf_counter() - start, "seconds to run.")
 
 
@@ -160,7 +162,7 @@ def generation_with_input_config(input_file, pdb_file, num_batches, num_designs_
                 results.append({"batch": batch, "design": design, "cif_path": cif_path, "pdb_path": pdb_path})
             
         print(results)
-        return directory, results
+        return gr.update(value="Generation Successful"), directory, results
     
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(f"Subprocess error:\n{e.stderr}") from e
+        return gr.update(f"Subprocess error:\n{e.stderr}"), None, None