fix main run button logic

app.py

@@ -8,7 +8,7 @@ import spaces
 from atomworks.io.utils.visualize import view
 from lightning.fabric import seed_everything
 from rfd3.engine import RFD3InferenceConfig, RFD3InferenceEngine
-from utils import download_weights
+from utils.download_weights import download_weights
 from utils.pipelines import test_rfd3_from_notebook, unconditional_generation
 #from gradio_molecule3d import Molecule3D
 from utils.handle_events import *
@@ -181,16 +181,24 @@ with gr.Blocks(title="RFD3 Test") as demo:
         
     run_btn.click(give_run_status, inputs=[config_ready, scaffold_ready, num_batches, num_designs_per_batch, length], outputs=runtextbox).then(
         generate, inputs=[config_ready, scaffold_ready, num_batches, num_designs_per_batch, length], outputs=[gen_directory, gen_results]
-    )
-        
-    run_btn.click(unconditional_generation, inputs=[num_batches, num_designs_per_batch, length], outputs=[gen_directory, gen_results]).then(
+    ).then(
         update_batch_choices,
         inputs=gen_results,
-        outputs=batch_dropdown).then(
-            download_results_as_zip,
-            inputs=gen_directory,
-            outputs=output_file
-        )
+        outputs=batch_dropdown
+    ).then(
+        download_results_as_zip,
+        inputs=gen_directory,
+        outputs=output_file
+    )
+        
+    #run_btn.click(unconditional_generation, inputs=[num_batches, num_designs_per_batch, length], outputs=[gen_directory, gen_results]).then(
+    #    update_batch_choices,
+    #    inputs=gen_results,
+    #    outputs=batch_dropdown).then(
+    #        download_results_as_zip,
+    #        inputs=gen_directory,
+    #        outputs=output_file
+    #    )
 
     batch_dropdown.change(update_designs, inputs=[batch_dropdown, gen_results], outputs=[design_dropdown])
     design_dropdown.change()

utils/__init__.py

@@ -1,4 +1,4 @@
-from utils.download_weights import *
-from utils.pipelines import *
-
-__all__ = ["download_weights"]
+#from utils.download_weights import *
+#from utils.pipelines import *
+#
+#__all__ = ["download_weights"]

utils/handle_events.py

@@ -53,4 +53,6 @@ def show_pdb(batch, design, result):
     pdb_path= next(d["pdb_path"] for d in result if d["batch"] == int(batch) and d["design"] == int(design))
     with open(pdb_path, 'r') as f:
         pdb_str = f.read()
-    return gr.update(value=f"Selected Batch: {batch}, Design: {design}, saved at {pdb_str}:\n {pdb_str}", visible=True)
+    print(pdb_path)
+    print(pdb_str)
+    return gr.update(value=f"Selected Batch: {batch}, Design: {design}, saved at {pdb_path}:\n {pdb_str}", visible=True)