Update app.py

app.py

@@ -68,15 +68,18 @@ with gr.Blocks() as demo:
     with gr.Row():
         # 左邊：控制按鈕與狀態欄
         with gr.Column(scale=1):
-            start_btn = gr.Button("開始對話")
-            end_btn = gr.Button("結束對話")
+            with gr.Row():
+                start_btn = gr.Button("開始對話")
+                end_btn = gr.Button("結束對話")
             textbox_input = gr.Textbox(label="Prompt", lines=1, placeholder="Enter prompt here...")
             button = gr.Button("Submit")
-            
+start_btn.click(fn=start_chat, outputs=status_box)
+end_btn.click(fn=finish_chat, outputs=status_box)           
         # 右邊：Prompt 輸入與 Markdown 輸出
         with gr.Column(scale=3):
-            status_box = gr.Textbox(label="狀態", interactive=False)
-            TPS_box = gr.Textbox(label="TPS", interactive=False)            
+            with gr.Row():
+                status_box = gr.Textbox(label="狀態", interactive=False)
+                TPS_box = gr.Textbox(label="TPS", interactive=False)            
             textbox_output = gr.Markdown(label="robot answer:", elem_id="scroll_output")