import argparse
import requests
import sys

URL = "https://biosig.lab.uq.edu.au/toxcsm/api/predict"

def main(args):   
    params = {
        "email_address": args.email_address,
        "smiles_string": args.smiles_string,
        "pred_type": args.pred_type
    }

    req = requests.post(URL, data=params)
    print(req.json())
    return True

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Job submission for a single SMILES\n')
    parser.add_argument('--smiles_string',  dest='smiles_string', type=str, help='Single SMILES string', required=False, default=None)    
    parser.add_argument('--pred_type',  dest='pred_type', type=str, help='prediction mode: all (default), organic, environmental, human_dose_response, nuclear_response, stress_response', required=False, default="all") 
    parser.add_argument('--email_address',  dest='email_address', type=str, help='E-mail address string', required=False, default="")

    args = parser.parse_args()
    if args.smiles_string == None:
        print('usage: python toxcsm_api_smiles_post.py --smiles_string SMILES_STRING [--pred_type PREDICTION_TYPE] [--email_address YOUR_EMAIL]')
        sys.exit(1)

    main(args)