feat: improve CLI entry point

Author: yuvrajsinh5252

README.md

@@ -58,7 +58,5 @@ lc daily py -e vim
 
 #### Todo
 
-- [ ] Implement test case management
 - [ ] Add support for custom test cases
-- [ ] Improve error handling
 - [ ] Add solution templates

setup.py

@@ -1,17 +1,12 @@
-from setuptools import setup, find_packages
+from setuptools import find_packages, setup
 
 setup(
     name="leetcode-cli",
     version="0.1.0",
     packages=find_packages(),
-    install_requires=[
-        "typer",
-        "requests",
-        "gql[all]"
-    ],
     entry_points={
         "console_scripts": [
-            "lc=src.main:app",
+            "lc=src.main:main",
         ],
-    }
-)
+    },
+)

src/lib/welcome.py

@@ -0,0 +1,18 @@
+import typer
+from rich.console import Console
+
+console = Console()
+
+
+def get_leetcode_ascii():
+    """Return LeetCode ASCII art"""
+    return """
+        |  _  _ |- _   | _
+        |_(/_(/_|_(_()(|(/_
+    """
+
+
+def display_welcome(app: typer.Typer):
+    """Display welcome screen with ASCII art and command list"""
+    console.print(get_leetcode_ascii())
+    console.print("      Welcome to LeetCode CLI", style="bold")

src/main.py

@@ -9,6 +9,7 @@
 from src.commands.solution import solutions
 from src.commands.submit import submit
 from src.commands.test import test
+from src.lib.welcome import display_welcome
 
 app = typer.Typer()
 
@@ -23,5 +24,18 @@
 app.command(name="edit")(edit)
 app.command(name="solutions")(solutions)
 
-if __name__ == "__main__":
+
+@app.callback(invoke_without_command=True)
+def callback(ctx: typer.Context):
+    """LeetCode CLI - A command-line tool for LeetCode problems."""
+    if ctx.invoked_subcommand is None:
+        display_welcome(app)
+        typer.echo(ctx.get_help())
+
+
+def main():
     app()
+
+
+if __name__ == "__main__":
+    main()