# import libraries
import argparse
import os

from modules.init import init

# load the classes
initClass = init()


"""
Parser functions
"""


# init funtion
def initParser(args):
    print("Creating project idea. This may take some time based on hardware.")
    # get systemPrompt
    print("Creating system prompt. (1/3")
    systemPrompt: str = initClass.generatePrompt(args)
    # print(systemPrompt)

    # Let the LLM generate a system prompt
    print("Generating Instructions (.MD) file. (2/3)")
    instructions = initClass.generateMd(systemPrompt, args.model)

    print(f"Creating instructions file. (3/3)")
    with open(f"{os.path.abspath(os.getcwd())}/Instructions.md", "w") as f:
        f.write(instructions)

    print(
        f"-- COMPLETED --\nInstructions: {os.path.abspath(os.getcwd())}/Instructions.md"
    )
    return


"""
Main class
"""


def main():
    # create the parser and subparser
    parser = argparse.ArgumentParser(
        description="Use LLM's to create project ideas. Powered by ollama!"
    )
    subparsers = parser.add_subparsers(required=True)

    parser_init = subparsers.add_parser(
        "init", help="Creates a new project using LLM's"
    )
    parser_init.add_argument(
        "-m",
        "--model",
        help="Defines the ollama model. [Default; ministral-3]",
        default="ministral-3",
        required=False,
        type=str,
    )
    parser_init.add_argument(
        "-l",
        "--language",
        help="Defines which programming language you want to create the project in. This will be send to the LLM [default: Any]",
        required=False,
        default="Any",
        type=str,
    )
    parser_init.add_argument(
        "-f",
        "--framework",
        help="Defines a framework if prefered. [default; Not specified]",
        required=False,
        default="Not specified",
        type=str,
    )
    parser_init.add_argument(
        "-e",
        "--packages",
        help="Defines if external packages are allowed",
        required=False,
        default=True,
        type=bool,
    )
    parser_init.add_argument(
        "-d",
        "--difficulty",
        help="Set your difficulty to the LLM [Default; Beginner]",
        default="Beginner",
        required=False,
    )
    parser_init.add_argument(
        "-t",
        "--time",
        help="Set the estimated time you want to work on the project. [Default; 1 Hour]",
        default="1 Hour",
        type=str,
    )
    parser_init.add_argument(
        "-p",
        "--project",
        help="Set the type of project. (For example; CLI-app, Website, Etc) [Default; Any]",
        default="Any",
        type=str,
    )
    parser_init.add_argument(
        "-c",
        "--custom",
        help="Give the llm custom instructions if needed. [Default; None]",
        default="none",
        type=str,
    )
    parser_init.set_defaults(func=initParser)

    args = parser.parse_args()
    args.func(args)


if __name__ == "__main__":
    main()