Update app.py

app.py

@@ -23,10 +23,10 @@ def find_qsar_guid_from_prompt(prompt: str, qsar_guid_index: int) -> str:
         qsar_guid_index: The qsar guid index to select given the prompt that the user gave, so if the user asks for solubility, the qsar guid should be possible_qsar_guids[0].key
     """
     
-    return possible_qsar_guids[qsar_guid_index].key
+    return possible_qsar_guids[qsar_guid_index]['key']
 
 @tool
-def qsartoolbox_qsar_model_tool(smiles:str, qsarModelGUID: str)-> str:
+def qsartoolbox_qsar_model_tool(smiles:str, qsarModelGUID: str)-> Optional[str]:
     """This machine learning model serves as an interface for the QSAR Toolbox API, 
     allowing users to input SMILES strings and select a QSAR model to generate predictive values for various properties.
     
@@ -35,7 +35,13 @@ def qsartoolbox_qsar_model_tool(smiles:str, qsarModelGUID: str)-> str:
         qsarModelGUID: a qsar model guid to use 
     """
 
-    return jaqpot.qsartoolbox_qsar_model_predict_sync(smiles, qsarModelGUID)
+    results = jaqpot.qsartoolbox_qsar_model_predict_sync(smiles, qsarModelGUID)
+    result = next((x for x in results if x["Value"] != ''), None) 
+
+    if(result != None):
+        return result['Value']
+    else
+        return None
 
 final_answer = FinalAnswerTool()