gabboud commited on
Commit
79a965a
·
1 Parent(s): 12aac99

fix and simplify logic

Browse files
Files changed (2) hide show
  1. app.py +2 -7
  2. utils/handle_events.py +3 -5
app.py CHANGED
@@ -140,16 +140,11 @@ with gr.Blocks(title="RFD3 Test") as demo:
140
  display_state.value = "Please Select a Batch and Design number to show sequence"
141
 
142
  def generate(config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args):
143
- if config_upload is None and scaffold_upload is None:
144
  return None, None
145
- if config_upload is not None and scaffold_upload is None:
146
  gen_directory, gen_results = generation_with_input_config(config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args)
147
  return gen_directory, gen_results
148
- if config_upload is not None and scaffold_upload is not None:
149
- gen_directory, gen_results = generation_with_input_config(config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args)
150
- return gen_directory, gen_results
151
- else:
152
- return None, None
153
 
154
  run_btn.click(give_run_status, inputs=[num_batches, num_designs_per_batch, config_upload, scaffold_upload], outputs=runtextbox).then(
155
  generate, inputs=[config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args], outputs=[gen_directory, gen_results]
 
140
  display_state.value = "Please Select a Batch and Design number to show sequence"
141
 
142
  def generate(config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args):
143
+ if config_upload is None:
144
  return None, None
145
+ else:
146
  gen_directory, gen_results = generation_with_input_config(config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args)
147
  return gen_directory, gen_results
 
 
 
 
 
148
 
149
  run_btn.click(give_run_status, inputs=[num_batches, num_designs_per_batch, config_upload, scaffold_upload], outputs=runtextbox).then(
150
  generate, inputs=[config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args], outputs=[gen_directory, gen_results]
utils/handle_events.py CHANGED
@@ -121,11 +121,9 @@ def give_run_status(num_batches, num_designs_per_batch, config_upload, scaffold_
121
 
122
 
123
  """
124
- if config_upload is None and scaffold_upload is None:
125
  return gr.update(value="Please ensure you have uploaded a configuration file: .yaml or .json")
126
- if config_upload is not None and scaffold_upload is None:
127
  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)}")
128
- if config_upload is not None and scaffold_upload is not None:
129
- return gr.update(value=f"Running conditional generation for {num_batches} batches of {num_designs_per_batch}, config uploaded from file {os.path.basename(config_upload)}, scaffold uploaded as {os.path.basename(scaffold_upload)}")
130
  else:
131
- return gr.update(value=f"Not implemented yet")
 
121
 
122
 
123
  """
124
+ if config_upload is None:
125
  return gr.update(value="Please ensure you have uploaded a configuration file: .yaml or .json")
126
+ elif scaffold_upload is None:
127
  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)}")
 
 
128
  else:
129
+ return gr.update(value=f"Running conditional generation for {num_batches} batches of {num_designs_per_batch}, config uploaded from file {os.path.basename(config_upload)}, scaffold uploaded as {os.path.basename(scaffold_upload)}")