Update app.py

app.py

@@ -450,7 +450,7 @@ builder.add_edge("reflect_node", END)
 graph = builder.compile()
 
 @spaces.GPU
-def PropAgent(smiles, reference, task):
+def PropAgent(task, smiles, reference):
 
   #if Substitution_image.png exists, remove it
   if os.path.exists('Substitution_image.png'):
@@ -501,6 +501,7 @@ with gr.Blocks(fill_height=True) as forest:
       pic = gr.Image(label="Molecule")
 
 
-      calc_btn.click(PropAgent, inputs = [smiles, ref, task], outputs = [props, pic])
+      calc_btn.click(PropAgent, inputs = [task, smiles, ref], outputs = [props, pic])
+      task.submit(PropAgent, inputs = [task, smiles, ref], outputs = [props, pic])
 
 forest.launch(debug=False, mcp_server=True)