Update app.py

app.py

@@ -977,6 +977,25 @@ graph = builder.compile()
 
 @spaces.GPU
 def ProteinAgent(task, protein, up_id, chembl_id, pdb_id, smiles):
+  '''
+  This Agent can perform several protein-related tasks. It can find UNIPROT IDs for a protein, or, given
+  a UNIPROT ID it can find Chembl IDs that match. It can find numbers of and lists of bioactive molecules
+  based on a Chembl ID. It can query the protein databank to find PDB IDs matching a protein name and return
+  the IDs and titles. It can find a particular PDB ID and report information such as how many chains it contains, 
+  the sequence, and any small molecules or ligands bound in the structure.
+
+      Args:
+          task: the task to carry out
+          protein: a protein name
+          up_id: a UNIPROT ID
+          chembl_id: a chembl ID
+          pdb_id: a PDB ID
+          smiles: a SMILES string for a molecule.
+
+      Returns:
+          replies[-1]: a text string containing the information requested
+          img: an image if appropriate, otherwise a blank image.
+  '''
   input = {
     "messages": [
         HumanMessage(f'query_task: {task}, query_protein: {protein}, query_up_id: {up_id}, query_chembl: {chembl_id}, query_pdb: {pdb_id}, query_smiles: {smiles}')