Update app.py

app.py

@@ -328,11 +328,11 @@ def sub_rings(smile_in: str, number_subs = 1) -> str:
      
       img = Draw.MolsToGridImage(mols, legends=legends, molsPerRow=3, subImgSize=(200,200),useSVG=False,returnPNG=False)
 
-    except:
+  except:
       out_text = "Invalid SMILES string"
       img = None
         
-    return None, None
+  return None, None
 
 with gr.Blocks() as gradio_app:
   gr.Markdown(
@@ -358,6 +358,8 @@ with gr.Blocks() as gradio_app:
 
   
   smile = gr.Textbox(label="SMILES for hit expansion")
+  
+
   with gr.Row():
     mask_btn = gr.Button("Generate analogues with Mask-filling.")
     sub_btn = gr.Button("Generate analogues with substitutions.")
@@ -367,7 +369,7 @@ with gr.Blocks() as gradio_app:
     mol_pic = gr.Image(label="Molecule Images:")
 
 
-  @mask_btn.click(inputs=[smile], outputs=[results, mol_pic])
+  @mask_btn.click(inputs=[smile, percent_mask], outputs=[results, mol_pic])
   def do_genmask(smile):
     return gen_mask(smile)
   
@@ -375,10 +377,6 @@ with gr.Blocks() as gradio_app:
   def do_subrings(smile):
     return sub_rings(smile)
 
-  @smile.submit(inputs=[smile], outputs=[results, mol_pic])
-  def do_genmask(smile,struct_type):
-    return gen_mask(smile)
-
 
 if __name__ == "__main__":
     gradio_app.launch(mcp_server=True)