Created init and base system prompt

ProjectCreator.py

@@ -1,38 +1,115 @@
-'''
-Init
-'''
-
 # import libraries
 import argparse
+import os
+
 from modules.init import init
 
 # load the classes
 initClass = init()
 
 
-'''
+"""
 Parser functions
-'''
+"""
+
+
 # init funtion
-def initParser():
-    
-    pass
+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!")
+    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", required=True, type=str)
-    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 = 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()
@@ -40,4 +117,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()