from tadas.snowflake import client as snowflake_utils


def model_predict_proba(input_table, output_table, ml_model_name, ml_model_version, model_features):
    sql = f"""
    create or replace transient table {output_table} as
    with tadas as model {ml_model_name} version {ml_model_version}
    select
        round(
            tadas ! predict_proba(
            {',\n    '.join(model_features)}
            )['output_feature_1']
            -- round to 2 decimal places
            ,2
        ) as tadas_30day_score,
        *
    from {input_table}
    where consecutive_trend > 0
    """
    with snowflake_utils.snowflake_connection() as conn:
        result = snowflake_utils.execute(query=sql, connection=conn)

    return result