Enhance CLI with subcommand handling and usage instructions

Author: RETR0-OS

ModelForge/cli.py

@@ -50,7 +50,24 @@ def check_huggingface_login():
 def main():
     """
     Main entry point for ModelForge CLI.
+
+    Subcommands:
+        modelforge        — start the web server
+        modelforge cli    — launch the interactive CLI wizard
     """
+    if len(sys.argv) > 1:
+        subcommand = sys.argv[1]
+        if subcommand == "cli":
+            from .notebook_cli.wizard import main as cli_main
+            cli_main()
+            return
+        else:
+            print(f"Unknown subcommand: '{subcommand}'")
+            print("Usage:")
+            print("  modelforge        Start the web server")
+            print("  modelforge cli    Launch the interactive CLI wizard")
+            sys.exit(1)
+
     print("\n" + "=" * 80)
     print("  __  __           _      _ _____                     ")
     print(" |  \\/  |         | |    | |  ___|                    ")

ModelForge/notebook_cli/wizard.py

@@ -32,10 +32,10 @@
     prompt_task,
 )
 
-_BANNER = """
+_BANNER = r"""
  __  __           _      _  _____
 |  \/  | ___   __| | ___| ||  ___|__  _ __ __ _  ___
-| |\/| |/ _ \ / _` |/ _ \ || |_ / _ \| '__/ _` |/ _ \\
+| |\/| |/ _ \ / _` |/ _ \ || |_ / _ \| '__/ _` |/ _ \
 | |  | | (_) | (_| |  __/ ||  _| (_) | | | (_| |  __/
 |_|  |_|\___/ \__,_|\___|_||_|  \___/|_|  \__, |\___|
                                            |___/