Merge pull request #15 from codedog-ai/feature_rearrage_layout

webui/qa.py

@@ -37,11 +37,22 @@ def recreate(instruction):
     task = QaTask(instruction=instruction, knowledge=load_vectorstore("example").as_retriever())
 
 
+def clear(msg, chatbot, session_id, telemetry, docs):
+    return "", "", '',  '{"tokens":0}', '["docs"]'
+
+
 with gr.Blocks() as qa_ui:
+    with gr.Row():
+        with gr.Column(scale=6):
+            instruction = gr.Textbox(label="Instruction", value=DEFAULT_INSTRUCTION, interactive=False)
+        with gr.Column(scale=1):
+            apply = gr.Button(value="更换Prompt")
+            clear_btn = gr.Button(value="清空")
+
     with gr.Row():
         with gr.Column(scale=6):
             with gr.Row():
-                chatbot = gr.Chatbot(height=800)
+                chatbot = gr.Chatbot(height=400)
             with gr.Row():
                 msg = gr.Textbox()
         with gr.Column(scale=1):
@@ -51,13 +62,7 @@ with gr.Blocks() as qa_ui:
                 telemetry = gr.JSON(value="""{"tokens":0}""", label="Telemetry")
             with gr.Row():
                 docs = gr.JSON(value="""["docs"]""", label="Docs")
-            with gr.Row():
-                clear = gr.ClearButton([msg, chatbot, session_id, telemetry, docs])
-    with gr.Row():
-        with gr.Column(scale=6):
-            instruction = gr.Textbox(label="Instruction", value=DEFAULT_INSTRUCTION, interactive=False)
-        with gr.Column(scale=1):
-            apply = gr.Button(value="更换Prompt")
 
+    clear_btn.click(clear, [msg, chatbot, session_id, telemetry, docs], [msg, chatbot, session_id, telemetry, docs])
     msg.submit(respond, [msg, chatbot, session_id], [msg, chatbot, session_id, telemetry, docs])
     apply.click(recreate, [instruction], [])