Update app.py

app.py

@@ -20,7 +20,7 @@ from deepface import DeepFace
 warnings.filterwarnings("ignore")
 
 # ==========================================
-# 0. 環境變數與 RAG 初始化
+#  環境變數與 RAG 初始化
 # ==========================================
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -86,7 +86,7 @@ if not df.empty and 'RAG_Content' in df.columns:
         print(f"⚠️ RAG 初始化失敗: {e}")
 
 # ==========================================
-# 1. 核心功能函式 (餐廳推薦邏輯)
+#  核心功能函式 (餐廳推薦邏輯)
 # ==========================================
 
 def get_restaurant_data(mood_score_str, food_choice):
@@ -208,7 +208,7 @@ def mood_agent_logic(score_input, food_input, diary_input, debug_mode):
     yield final_response, image_output, map_html
 
 # ==========================================
-# 2. 情緒辨識整合模組 (維持不變)
+#  [修改] 情緒辨識整合模組
 # ==========================================
 
 # 定義中文字典 (繪圖用)
@@ -231,10 +231,23 @@ def putText(img, x, y, text, size=50, color=(255, 255, 255)):
     except:
         return img
 
-# 關鍵功能：偵測情緒並回傳「分數選項」 (維持不變)
-def detect_emotion_and_map(frame):
-    if frame is None:
-        return frame, None
+# ★ 新增：按鈕切換函式 (控制按鈕的文字與顏色)
+def toggle_recording(is_recording):
+    # 切換狀態 (True <-> False)
+    new_state = not is_recording
+    
+    if new_state:
+        # 如果變成錄製中，按鈕變成紅色「停止」
+        return new_state, gr.Button(value="🛑 停止錄製 (鎖定結果)", variant="stop")
+    else:
+        # 如果變成停止，按鈕變成綠色「開始」
+        return new_state, gr.Button(value="📸 開始即時偵測", variant="secondary")
+
+# ★ 修改：偵測邏輯 (加入 is_recording 判斷)
+def detect_emotion_stream(frame, is_recording):
+    # 如果沒有在錄製，或是沒有畫面，就什麼都不更新 (讓畫面凍結在最後一幀)
+    if not is_recording or frame is None:
+        return gr.update(), gr.update()
     
     detected_emotion = "neutral" 
     mapped_score = "3 (普通)"    
@@ -248,7 +261,7 @@ def detect_emotion_and_map(frame):
         # 2. 畫在圖片上
         frame = putText(frame, 20, 40, detected_emotion)
 
-        # 3. 橋樑：將情緒轉換為餐廳系統的分數
+        # 3. 分數轉換
         if detected_emotion == 'happy':
             mapped_score = "5 (超棒)"
         elif detected_emotion == 'surprise':
@@ -261,17 +274,19 @@ def detect_emotion_and_map(frame):
             mapped_score = "1 (心情差)"
 
     except Exception as e:
-        print(f"DeepFace Error: {e}")
         pass
     
     return frame, mapped_score
 
 # ==========================================
-# 5. Gradio 介面建構 (修改版：鏡頭與結果分離)
+#  Gradio 介面建構 (修改版)
 # ==========================================
 
 with gr.Blocks(title="AI 心情食堂") as demo:
     
+    # ★ 1. 定義一個變數來記錄狀態 (預設是 False 停止中)
+    recording_state = gr.State(value=False)
+
     with gr.Column(visible=True) as main_app_col: 
         gr.Markdown(f"## 🍱 AI 心情食堂導航")
         gr.Markdown("請看著鏡頭，讓 AI 幫你判斷今天的心情分數！")
@@ -280,23 +295,20 @@ with gr.Blocks(title="AI 心情食堂") as demo:
             # 左側：輸入區
             with gr.Column(scale=1):
                 
-                # ★ 修改重點：將鏡頭與結果分開顯示 ★
                 gr.Markdown("### 📸 步驟 1：心情偵測 (選用)")
-                
-                # 這裡用 Row 把兩個影像並排 (左邊鏡頭，右邊截圖)
                 with gr.Row():
-                    # 左邊：永遠是即時鏡頭 (不設為 output)
-                    webcam_input = gr.Image(sources=["webcam"], label="即時鏡頭 (請看這裡)", streaming=True)
-                    
-                    # 右邊：顯示偵測後的「靜態截圖」
-                    captured_image = gr.Image(label="偵測結果截圖", interactive=False)
+                    # 左邊：即時鏡頭
+                    webcam_input = gr.Image(sources=["webcam"], label="即時鏡頭", streaming=True)
+                    # 右邊：結果畫面 (互動模式設為 False)
+                    captured_image = gr.Image(label="即時偵測結果", interactive=False)
                 
-                detect_btn = gr.Button("📸 截圖並偵測心情 👇", variant="secondary")
+                # 按鈕預設為「開始」
+                detect_btn = gr.Button("📸 開始即時偵測", variant="secondary")
                 
                 gr.Markdown("### 📝 步驟 2：確認與補充")
                 score_input = gr.Radio(
                     ["1 (心情差)", "2 (不太好)", "3 (普通)", "4 (不錯)", "5 (超棒)"], 
-                    label="1. 心情分數 (AI 會自動填入，也可手動改)", 
+                    label="1. 心情分數 (AI 自���填入)", 
                     value="3 (普通)"
                 )
                 food_input = gr.Radio(["吃飯", "吃麵", "隨便"], label="2. 想吃什麼", value="隨便")
@@ -305,7 +317,7 @@ with gr.Blocks(title="AI 心情食堂") as demo:
                 debug_mode_btn = gr.Checkbox(label="🔧 顯示 Prompt 除錯資訊", value=False)
                 submit_btn = gr.Button("🚀 送出給 Agent", variant="primary")
 
-            # 右側：Agent 輸出區 (維持不變)
+            # 右側：Agent 輸出區
             with gr.Column(scale=1):
                 agent_output = gr.Markdown(label="AI 回應")
                 image_output = gr.Image(label="AI 推薦美食圖", type="pil", width=400)
@@ -313,11 +325,19 @@ with gr.Blocks(title="AI 心情食堂") as demo:
 
     # Events 
     
-    # ★ 修改重點：按鈕點擊後，輸出目標改為 captured_image，不再覆蓋 webcam_input
+    # ★ 事件 1：按鈕點擊 -> 切換錄製狀態 -> 更新按鈕樣子
     detect_btn.click(
-        fn=detect_emotion_and_map,
-        inputs=[webcam_input],
-        outputs=[captured_image, score_input] # 輸出到「右邊的截圖」和「分數選項」
+        fn=toggle_recording,
+        inputs=[recording_state],
+        outputs=[recording_state, detect_btn]
+    )
+
+    # ★ 事件 2：鏡頭串流 (Stream) -> 只要鏡頭有畫面就會觸發
+    # 它會檢查 recording_state，如果是 True 才會更新右邊的圖
+    webcam_input.stream(
+        fn=detect_emotion_stream,
+        inputs=[webcam_input, recording_state],
+        outputs=[captured_image, score_input]
     )
 
     # 提交按鈕