main.py

"""CLI entry point for tools suite access. As needed, add appropriate argument options as the project grows.
"""
import argparse
import model
from fetch import (
    nasa_firms_api,
    setup_auth,
    batch_data_downloader_selenium,
    retrieve_eonet_cross_reference,
    copernicus_sentiel_query,
)

if __name__ == "__main__":
    """
    Parses command-line arguments and executes the appropriate functionality.

    This script serves as the entry point for the Payload AI Software Suite. It
    allows users to run a model or execute specific data-fetching functions.

    Command-line Arguments:
        --run-model: Run the model.
        --nasa-firms: Fetch data from NASA FIRMS API.
        --setup-auth: Set up OAuth2 authentication for Copernicus.
        --batch-download: Download images using Selenium.
        --eonet-crossref: Fetch wildfire data from the EONET API.
        --copernicus-query: Query Sentinel data from Copernicus.
        --coordinates: Specify coordinates for the query in the format: LON LAT.
        --time-range: Time range for the query in the format: FROM TO
                      (e.g., '2023-01-01T00:00:00Z 2023-01-03T23:59:59Z').

    Raises:
        SystemExit: If invalid arguments are provided.
    """
    parser = argparse.ArgumentParser(
        prog='Payload AI Software Suite',
        description='Remote sensing mission core tools for wildfire image classification and data retrieval'
    )
    parser.add_argument('--run-model', required=False, action='store_true', help="Run the model")
    parser.add_argument('--nasa-firms', required=False, action='store_true', help="Fetch data from NASA FIRMS API")
    parser.add_argument('--setup-auth', required=False, action='store_true', help="Set up OAuth2 authentication for Copernicus")
    parser.add_argument('--batch-download', required=False, action='store_true', help="Download images using Selenium")
    parser.add_argument('--eonet-crossref', required=False, action='store_true', help="Fetch wildfire data from the EONET API")
    parser.add_argument('--copernicus-query', required=False, action='store_true', help="Query Sentinel data from Copernicus")
    parser.add_argument('--coordinates', required=False, nargs=2, type=float, metavar=('LON', 'LAT'),
                        help="Specify coordinates for the query in the format: LON LAT")
    parser.add_argument('--time-range', required=False, nargs=2, metavar=('FROM', 'TO'),
                        help="Time range for the query in the format: FROM TO (e.g., '2023-01-01T00:00:00Z 2023-01-03T23:59:59Z')")

    args = parser.parse_args()
    if args.run_model:
        model.train()
    elif args.nasa_firms:
        nasa_firms_api()
    elif args.setup_auth:
        setup_auth()
    elif args.batch_download:
        batch_data_downloader_selenium()
    elif args.eonet_crossref:
        retrieve_eonet_cross_reference()
    elif args.copernicus_query:
        copernicus_sentiel_query()
    else:
        print("No valid arguments provided. Use -h for help.")