add web demo

Former-commit-id: 25ea647e5ac36b497b8e176b123fdee39be3fd30

src/llmtuner/webui/chat.py

@@ -11,10 +11,12 @@ from llmtuner.webui.locales import ALERTS
 
 class WebChatModel(ChatModel):
 
-    def __init__(self):
+    def __init__(self, *args):
         self.model = None
         self.tokenizer = None
         self.generating_args = GeneratingArguments()
+        if len(args) != 0:
+            super().__init__(*args)
 
     def load_model(
         self,

src/llmtuner/webui/components/chatbot.py

@@ -1,4 +1,4 @@
-from typing import Dict, Tuple
+from typing import Dict, Optional, Tuple
 
 import gradio as gr
 from gradio.blocks import Block
@@ -8,9 +8,10 @@ from llmtuner.webui.chat import WebChatModel
 
 
 def create_chat_box(
-    chat_model: WebChatModel
+    chat_model: WebChatModel,
+    visible: Optional[bool] = False
 ) -> Tuple[Block, Component, Component, Dict[str, Component]]:
-    with gr.Box(visible=False) as chat_box:
+    with gr.Box(visible=visible) as chat_box:
         chatbot = gr.Chatbot()
 
         with gr.Row():