Update app.py

app.py

@@ -72,9 +72,7 @@ with gr.Blocks() as demo:
                 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)           
+            button = gr.Button("Submit")        
         # 右邊：Prompt 輸入與 Markdown 輸出
         with gr.Column(scale=3):
             with gr.Row():
@@ -82,7 +80,8 @@ end_btn.click(fn=finish_chat, outputs=status_box)
                 TPS_box = gr.Textbox(label="TPS", interactive=False)            
             textbox_output = gr.Markdown(label="robot answer:", elem_id="scroll_output")
 
-
+    start_btn.click(fn=start_chat, outputs=status_box)
+    end_btn.click(fn=finish_chat, outputs=status_box)   
 
     # 當按鈕被按下時，呼叫 generate_stream 並更新 textbox_output
     button.click(fn=generate_stream, inputs=textbox_input, outputs=[textbox_output, TPS_box])