Update app.py

app.py

@@ -13,6 +13,14 @@ hf_hub.snapshot_download(model_id, local_dir=model_path)
 pipe = ov_genai.LLMPipeline(model_path, "CPU")
 #pipe.start_chat()
 
+def start_chat():
+    pipe.start_chat()
+    return "✅ 開始對話！"
+
+def finish_chat():
+    pipe.finish_chat()
+    return "🛑 結束對話！"
+	
 # 建立推論函式：使用 streamer 並回傳 generator 結果
 def generate_stream(prompt):
     q = queue.Queue()
@@ -47,7 +55,12 @@ with gr.Blocks(css="""
 }
 """) as demo:
     gr.Markdown("## 🧠 OpenVINO Streaming Demo with Gradio Textbox")
-
+	
+    start_btn = gr.Button("開始對話")
+    end_btn = gr.Button("結束對話")
+    start_btn.click(fn=start_chat, outputs=status_box)
+    end_btn.click(fn=finish_chat, outputs=status_box)
+	
     textbox_input = gr.Textbox(label="Prompt", lines=1, placeholder="Enter prompt here...")
     textbox_output = gr.Textbox(label="Output", elem_id="scrollbox", lines=10)