gradio chat entrypoint moved to cli

Dockerfile

@@ -15,4 +15,4 @@ RUN pip install --no-cache-dir .
 EXPOSE 7860
 
 # Step 6: Command to run your Gradio app (or any Python script)
-CMD ["python", "app.py"]
+CMD ["python", "hf_transformers_demo.py"]

src/knowlang/chat_bot/gradio_demo.py

@@ -1,7 +0,0 @@
-from knowlang.chat_bot.chat_interface import create_chatbot
-from knowlang.configs.config import AppConfig
-
-
-config = AppConfig()
-demo = create_chatbot(config)
-demo.launch()

src/knowlang/cli/commands/chat.py

@@ -0,0 +1,48 @@
+"""Command implementation for the chat interface."""
+import chromadb
+from knowlang.chat_bot.chat_interface import create_chatbot
+from knowlang.configs.config import AppConfig
+from knowlang.cli.types import ChatCommandArgs
+from knowlang.utils.fancy_log import FancyLogger
+
+LOG = FancyLogger(__name__)
+
+def create_config(args: ChatCommandArgs) -> AppConfig:
+    """Create configuration from file or defaults."""
+    if args.config:
+        with open(args.config, 'r') as file:
+            config_data = file.read()
+            return AppConfig.model_validate_json(config_data)
+    return AppConfig()
+
+async def chat_command(args: ChatCommandArgs) -> None:
+    """Execute the chat command.
+    
+    Args:
+        args: Typed command line arguments
+    """
+    config = create_config(args)
+    
+    # Verify database exists
+    try:
+        db_client = chromadb.PersistentClient(path=str(config.db.persist_directory))
+        db_client.get_collection(name=config.db.collection_name)
+    except Exception as e:
+        LOG.error(
+            "Database not found. Please run 'knowlang parse' first to index your codebase."
+            f"\nError: {str(e)}"
+        )
+        return
+    
+    # Create and launch chatbot
+    demo = create_chatbot(config)
+    
+    launch_kwargs = {
+        "server_port": args.server_port,
+        "server_name": args.server_name or "127.0.0.1",
+        "share": args.share,
+    }
+    if args.port:
+        launch_kwargs["port"] = args.port
+        
+    demo.launch(**launch_kwargs)

src/knowlang/cli/parser.py

@@ -3,7 +3,8 @@ import argparse
 from pathlib import Path
 from typing import Union
 
-from knowlang.cli.types import ParseCommandArgs, BaseCommandArgs
+from knowlang.cli.commands.chat import chat_command
+from knowlang.cli.types import ChatCommandArgs, ParseCommandArgs, BaseCommandArgs
 from knowlang.cli.commands.parse import parse_command
 
 def _convert_to_args(parsed_args: argparse.Namespace) -> Union[ParseCommandArgs, BaseCommandArgs]:
@@ -20,6 +21,15 @@ def _convert_to_args(parsed_args: argparse.Namespace) -> Union[ParseCommandArgs,
             output=parsed_args.output,
             command="parse"
         )
+    elif parsed_args.command == "chat":
+        return ChatCommandArgs(
+            **base_args,
+            command="chat",
+            port=parsed_args.port,
+            share=parsed_args.share,
+            server_port=parsed_args.server_port,
+            server_name=parsed_args.server_name
+        )
     
     return BaseCommandArgs(**base_args)
 
@@ -70,6 +80,33 @@ def create_parser() -> argparse.ArgumentParser:
         help="Path to codebase directory or repository URL"
     )
     parse_parser.set_defaults(func=parse_command)
+
+    # Chat command
+    chat_parser = subparsers.add_parser(
+        "chat",
+        help="Launch the chat interface"
+    )
+    chat_parser.add_argument(
+        "--port",
+        type=int,
+        help="Port to run the interface on"
+    )
+    chat_parser.add_argument(
+        "--share",
+        action="store_true",
+        help="Create a shareable link"
+    )
+    chat_parser.add_argument(
+        "--server-port",
+        type=int,
+        help="Port to run the server on (if different from --port)"
+    )
+    chat_parser.add_argument(
+        "--server-name",
+        type=str,
+        help="Server name to listen on (default: 0.0.0.0)"
+    )
+    chat_parser.set_defaults(func=chat_command)
     
     return parser
 

src/knowlang/cli/types.py

@@ -14,4 +14,13 @@ class ParseCommandArgs(BaseCommandArgs):
     """Arguments for the parse command."""
     path: Path
     output: Literal["table", "json"]
-    command: Literal["parse"]  # for command identification
+    command: Literal["parse"]  # for command identification
+
+@dataclass
+class ChatCommandArgs(BaseCommandArgs):
+    """Arguments for the chat command."""
+    command: Literal["chat"]
+    port: Optional[int] = None
+    share: bool = False
+    server_port: Optional[int] = None
+    server_name: Optional[str] = None