Improve CLI

great_ai/__main__.py

@@ -40,7 +40,7 @@ GREAT_AI_LOGGING_CONFIG = {
 }
 
 
-def main() -> None:
+def serve() -> None:
     args = parse_arguments()
     should_auto_reload = not _is_in_production_mode(logger=None)
 
@@ -200,10 +200,14 @@ class GreatAIReload(BaseReload):
             sock.close()
 
 
-if __name__ == "__main__":
+def main() -> None:
     try:
-        main()
+        serve()
     except KeyboardInterrupt:
         exit()
     except Exception as e:
         logger.error(e)
+
+
+if __name__ == "__main__":
+    main()

great_ai/large_file/__main__.py

@@ -11,7 +11,7 @@ from .parse_arguments import parse_arguments
 logger = get_logger("large_file")
 
 
-def main() -> None:
+def handle_command() -> None:
     parser, args = parse_arguments()
 
     large_file = get_class(args)
@@ -61,11 +61,15 @@ def get_class(args: Namespace) -> Type[LargeFileBase]:
     return large_file
 
 
-if __name__ == "__main__":
+def main() -> None:
     try:
-        main()
+        handle_command()
     except KeyboardInterrupt:
         logger.warning("Exiting")
         exit()
     except Exception as e:
         logger.exception(e)
+
+
+if __name__ == "__main__":
+    main()

great_ai/parse_arguments.py

@@ -12,7 +12,7 @@ def parse_arguments() -> Namespace:
         help="the name of the file containing your to-be-served function such as `main.py`\n",
     )
 
-    default_host = "0.0.0.0"
+    default_host = "127.0.0.1"
     parser.add_argument(
         "--host",
         type=str,