python code for AME Solar energy prediction