vladimir.manuylov commited on
Commit
bea5b09
·
1 Parent(s): 047aecb

fixing empty seq bug

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -64,6 +64,9 @@ def generate_smiles_for_sequence(protein_sequence: str, num_samples: int, reques
64
 
65
  return ",\n".join(unique_smiles)
66
 
 
 
 
67
 
68
  def log_run(client_ip: str, seq: str):
69
  rec = {
@@ -138,7 +141,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
138
  "More attempts increase runtime but can improve diversity."
139
  )
140
  )
141
- submit_btn = gr.Button("Generate Molecules", variant="primary")
142
 
143
  with gr.Column(scale=3):
144
  output_smiles = gr.Textbox(
@@ -148,6 +151,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
148
  interactive=True
149
  )
150
 
 
151
  gr.Markdown("### Examples")
152
  gr.Examples(
153
  examples=[
@@ -168,6 +172,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
168
  *Model developed by Gero AI. For more details, check out the [original repository](https://github.com/gero-science/ProtoBind-Diff).*
169
  """
170
  )
 
171
 
172
  submit_btn.click(
173
  fn=generate_smiles_for_sequence,
 
64
 
65
  return ",\n".join(unique_smiles)
66
 
67
+ def enable_btn(seq: str):
68
+ return gr.Button.update(interactive=len(seq) >= 10)
69
+
70
 
71
  def log_run(client_ip: str, seq: str):
72
  rec = {
 
141
  "More attempts increase runtime but can improve diversity."
142
  )
143
  )
144
+ submit_btn = gr.Button("Generate Molecules", variant="primary", interactive=False)
145
 
146
  with gr.Column(scale=3):
147
  output_smiles = gr.Textbox(
 
151
  interactive=True
152
  )
153
 
154
+
155
  gr.Markdown("### Examples")
156
  gr.Examples(
157
  examples=[
 
172
  *Model developed by Gero AI. For more details, check out the [original repository](https://github.com/gero-science/ProtoBind-Diff).*
173
  """
174
  )
175
+ protein_sequence.change(enable_btn, inputs=protein_sequence, outputs=submit_btn)
176
 
177
  submit_btn.click(
178
  fn=generate_smiles_for_sequence,