support pdb input

app.py

@@ -116,6 +116,9 @@ with gr.Blocks(title="RFD3 Test") as demo:
         if config_ready == "upload" and scaffold_ready == "no_input":
             gen_directory, gen_results = generation_with_input_config(config_upload, num_batches, num_designs_per_batch, extra_args)
             return gen_directory, gen_results
+        if config_ready=="upload" and scaffold_ready=="upload":
+            gen_directory, gen_results = generation_with_input_config(config_upload, num_batches, num_designs_per_batch, extra_args)
+            return gen_directory, gen_results         
         else:
             return None, None
         

utils/handle_events.py

@@ -106,7 +106,7 @@ def validate_scaffold_ready_with_file(scaffold_upload):
     else:
         return "Please upload a scaffold file or press 'No Scaffold/Target' for unconditional generation.", None
     
-def give_run_status(config_ready, scaffold_ready, num_batches, num_designs_per_batch, config_upload):
+def give_run_status(config_ready, scaffold_ready, num_batches, num_designs_per_batch, config_upload, scaffold_upload):
     """
     Once the user presses on the Run Generation button, this function checks whether both config and scaffold are ready, and gives a status message about the generation run that is about to happen. If both config and scaffold are not ready, it prompts the user to ensure they are both validated before running the generation.
 
@@ -117,13 +117,13 @@ def give_run_status(config_ready, scaffold_ready, num_batches, num_designs_per_b
     scaffold_ready: gr.State
         status string for scaffold readiness, is "upload" if scaffold file is uploaded and valid, "no_input" if no scaffold/target is selected for unconditioned generation, and None if scaffold is not ready for generation
     num_batches: gr.Number
-        number of batches input by the user in the manual config, used for the generation status message
+        number of batches input by the user manually
     num_designs_per_batch: gr.Number
-        number of designs per batch input by the user in the manual config, used for the generation
-    length: gr.Number
-        length input by the user in the manual config, used for the generation status message
+        number of designs per batch input by the user manually
     config_upload: gr.File
-        uploaded config file, used for the generation status message if upload config is selected
+        uploaded config file, is None if no file is uploaded
+    scaffold_upload: gr.File
+        uploaded scaffold file, is None if no file is uploaded
 
     Returns:
     -------
@@ -135,5 +135,7 @@ def give_run_status(config_ready, scaffold_ready, num_batches, num_designs_per_b
         return gr.update(value="Please ensure both config and scaffold are validated before running the generation.")
     elif config_ready == "upload" and scaffold_ready == "no_input":
         return gr.update(value=f"Running unconditional generation for {num_batches} batches of {num_designs_per_batch}, config uploaded from file {os.path.basename(config_upload)}")
+    elif config_ready=="upload" and scaffold_ready=="upload":
+        return gr.update(value=f"Running conditional generation for {num_batches} batches of {num_designs_per_batch}, config uploaded from file {config_upload}, scaffold uploaded a {scaffold_upload}")
     else:
         return gr.update(value=f"Not implemented yet")