README.md

@@ -6,9 +6,7 @@ colorTo: purple
 sdk: gradio
 sdk_version: 3.36.1
 app_file: app.py
-pinned: false
-license: apache-2.0
-duplicated_from: FredZhang7/chat-test
+pinned: true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -59,7 +59,7 @@ model_cache_path = [elm for elm in Path(cache_loc).rglob("*") if Path(model_name
 logger.debug(f"{model_cache_path=}")
 
 if model_cache_path:
-    model_size_gb = Path(model_cache_path[0]).stat().st_size / 2**30
+    model_size_gb = model_cache_path[0].stat().st_size / 2**30
     logger.info(f"{model_name=} {model_size_gb=:.2f} GB")
 
 
@@ -68,12 +68,21 @@ def respond(message, chat_history):
     chat_history.append((message, response))
     return "", chat_history
 
-
-with gr.Blocks() as block:
+theme = gr.themes.Soft(text_size="sm")
+with gr.Blocks(theme=theme) as block:
     chatbot = gr.Chatbot()
-    msg = gr.Textbox()
 
-clear = gr.ClearButton([msg, chatbot])
-msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    with gr.Column():
+        with gr.Column(scale=12):
+            msg = gr.Textbox()
+        with gr.Column(scale=1, min_width=16):
+            btn = gr.Button()
+        with gr.Column(scale=1, min_width=16):
+            clear = gr.ClearButton([msg, chatbot])
+
+    # do not clear prompt
+    msg.submit(lambda x, y: [x] + respond(x, y)[1:], [msg, chatbot], [msg, chatbot])
+
+    btn.click(respond, [msg, chatbot], [msg, chatbot])
 
 block.queue().launch()