Update app.py

app.py

@@ -28,7 +28,8 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 # --- 1. 資料讀取 (主要資料庫 - 使用 restaurants.csv) ---
 df = pd.DataFrame()
 try:
-    df = pd.read_csv('restaurants.csv', engine='python')
+    # 確保你有 restaurants.csv 在專案根目錄
+    df = pd.read_csv('restaurants.csv', engine='python') 
     df.columns = df.columns.str.strip() # 清理欄位空白
 
     def classify_mood(row):
@@ -54,7 +55,8 @@ try:
     
     print("✅ 成功讀取 restaurants.csv 並建立 Mood_Tags")
 except Exception as e:
-    print(f"⚠️ 讀取 restaurants.csv 失敗: {e}")
+    # 如果找不到檔案或讀取失敗，這裡會提醒
+    print(f"⚠️ 讀取 restaurants.csv 失敗: {e}") 
 
 # --- 2. 資料讀取 (Prompt 資料庫) ---
 df_prompts = pd.DataFrame()
@@ -84,51 +86,12 @@ if not df.empty and 'RAG_Content' in df.columns:
         print(f"⚠️ RAG 初始化失敗: {e}")
 
 # ==========================================
-# 1. 小測驗設定
-# ==========================================
-custom_css = """
-.vertical-radio fieldset {
-    display: flex !important;
-    flex-direction: column !important;
-    align-items: flex-start !important;
-    gap: 12px !important;
-}
-.vertical-radio label {
-    width: 100% !important;
-    margin: 0 !important;
-    display: flex !important;
-}
-"""
-# 若無圖片自動生成空白圖避免報錯
-img_path_1 = "images/image_0.png"
-img_path_2 = "images/image_1.png"
-if not os.path.exists("images"): os.makedirs("images")
-try:
-    if not os.path.exists(img_path_1): Image.new('RGB', (400, 300), color='gray').save(img_path_1)
-    if not os.path.exists(img_path_2): Image.new('RGB', (400, 300), color='lightblue').save(img_path_2)
-except: pass
-
-q1_options_map = {
-    "不​要​再​問白癡​問題​了​我​只​想​吃飯​": "現實主義者 (只想吃飯)",
-    "去​過隱​居​生活": "隱士 (嚮往平靜)",
-    "加入​禁衛軍，​保衛​這個​世界": "守護者 (充滿正義感)",
-    "神​羅​天​征，毀滅這個​世界​": "破壞神 (心情可能很差或很中二)"
-}
-
-q2_options_map = {
-    "不​要​再​問白癡​問題​了​我​只​想​吃飯​": "無魔法 (飢餓度MAX)",
-    "麵​包形狀​的​魔法​炸彈": "爆炸魔法 (喜歡刺激/重口味)",
-    "毒氣​的​生化​魔法": "毒氣魔法 (可能想吃臭豆腐或特殊風味)",
-    "領域​展開:​無量​空處​": "領域展開 (思緒混亂或想放空)"
-}
-
-# ==========================================
-# 2. 核心功能函式 (餐廳推薦邏輯)
+# 1. 核心功能函式 (餐廳推薦邏輯)
 # ==========================================
 
 def get_restaurant_data(mood_score_str, food_choice):
     if df.empty: return None, True
-    # 定義分數與 Mood_Tags 的對應關係
+    # 定義分數與 Mood_Tags 的對應關係 (與舊版維持一致)
     score_map = {
         "1 (心情差)": ["傷心/疲憊", "生氣/發洩"],
         "2 (不太好)": ["傷心/疲憊", "生氣/發洩"],
@@ -157,23 +120,21 @@ def get_restaurant_data(mood_score_str, food_choice):
     if candidates.empty: return df.sample(1).iloc[0], True
     return candidates.sample(1).iloc[0], False
 
-def generate_content_with_groq(restaurant_name, restaurant_detail, user_diary, mood_score, quiz_result):
+# 注意：因為移除了測驗，這裡我們用 '無測驗' 作為預設結果
+def generate_content_with_groq(restaurant_name, restaurant_detail, user_diary, mood_score):
     if not GROQ_API_KEY: return "⚠️ 請設定 GROQ_API_KEY (系統變數)"
     client = OpenAI(api_key=GROQ_API_KEY, base_url="https://api.groq.com/openai/v1")
-    system_prompt = "你是一個幽默、懂吃且善解人意的 AI 朋友。請根據使用者的日記、心情以及「中二病測驗結果」來推薦餐廳。"
+    system_prompt = "你是一個幽默、懂吃且善解人意的 AI 朋友。請根據使用者的日記和心情來推薦餐廳。"
     user_msg = f"""
     【使用者狀態】
     心情分數：{mood_score}
     心情日記：{user_diary}
-    【前測小遊戲結果】
-    隱藏人設：{quiz_result.get('q1', '未知')}
-    魔法：{quiz_result.get('q2', '未知')}
     【推薦餐廳】
     名稱：{restaurant_name}
     資料：{restaurant_detail}
     請完成任務：
     【回應內容】：用繁體中文寫一段溫暖有趣的回覆。
-       - 結合「日記」與「前測人設」給予回應。
+       - 結合「日記」與「心情」給予回應。
        - 推薦上述餐廳，說明為什麼這家店適合他。
     """
     try:
@@ -193,7 +154,8 @@ def generate_image_huggingface(prompt):
     except: return None
 
 # --- 主邏輯 Agent ---
-def mood_agent_logic(score_input, food_input, diary_input, quiz_state, debug_mode):
+# 移除了 quiz_state 參數
+def mood_agent_logic(score_input, food_input, diary_input, debug_mode):
     # 確保 score_input 有值
     if not score_input:
         yield "⚠️ 請先選擇心情分數，或使用上方相機偵測！", None, ""
@@ -230,8 +192,8 @@ def mood_agent_logic(score_input, food_input, diary_input, quiz_state, debug_mod
                     prompt_source = "✅ CSV 檔案"
             except: pass
 
-    # 3. 呼叫 LLM
-    ai_text = generate_content_with_groq(name, rag_info, diary_input, score_input, quiz_state)
+    # 3. 呼叫 LLM (移除了 quiz_state)
+    ai_text = generate_content_with_groq(name, rag_info, diary_input, score_input)
     
     # 4. 組合回應
     debug_text = ""
@@ -246,7 +208,7 @@ def mood_agent_logic(score_input, food_input, diary_input, quiz_state, debug_mod
     yield final_response, image_output, map_html
 
 # ==========================================
-# 3. [新增] 情緒辨識整合模組
+# 2. 情緒辨識整合模組 (維持不變)
 # ==========================================
 
 # 定義中文字典 (繪圖用)
@@ -255,11 +217,10 @@ emotion_text_obj = {
     'happy': '開心', 'sad': '難過', 'surprise': '驚訝', 'neutral': '正常'
 }
 
-# 繪圖函式
+# 繪圖函式 (維持不變)
 def putText(img, x, y, text, size=50, color=(255, 255, 255)):
     try:
         fontpath = 'NotoSansTC-VariableFont_wght.ttf'
-        # 確保字型存在，若無則回傳原圖
         if not os.path.exists(fontpath): return img
         font = ImageFont.truetype(fontpath, size)
         imgPil = Image.fromarray(img)
@@ -270,13 +231,13 @@ 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
     
-    detected_emotion = "neutral" # 預設
-    mapped_score = "3 (普通)"    # 預設
+    detected_emotion = "neutral" 
+    mapped_score = "3 (普通)"    
 
     try:
         # 1. 辨識情緒
@@ -288,11 +249,6 @@ def detect_emotion_and_map(frame):
         frame = putText(frame, 20, 40, detected_emotion)
 
         # 3. 橋樑：將情緒轉換為餐廳系統的分數
-        # happy -> 5
-        # surprise -> 4
-        # neutral -> 3
-        # sad, fear -> 2
-        # angry, disgust -> 1
         if detected_emotion == 'happy':
             mapped_score = "5 (超棒)"
         elif detected_emotion == 'surprise':
@@ -311,63 +267,27 @@ def detect_emotion_and_map(frame):
     return frame, mapped_score
 
 # ==========================================
-# 4. 頁面切換與 UI 邏輯
-# ==========================================
-
-def handle_q1_change(selected_label, current_state):
-    if not selected_label: return current_state, gr.Button(interactive=False)
-    current_state["q1"] = q1_options_map.get(selected_label, selected_label)
-    return current_state, gr.Button(interactive=True, variant="primary")
-
-def handle_q2_change(selected_label, current_state):
-    if not selected_label: return current_state, gr.Button(interactive=False)
-    current_state["q2"] = q2_options_map.get(selected_label, selected_label)
-    return current_state, gr.Button(interactive=True, variant="primary", value="完成測驗 (前往點餐) ➔")
-
-def go_to_page_2(): return gr.update(visible=False), gr.update(visible=True)
-def go_to_main_app(): return gr.update(visible=False), gr.update(visible=True)
-
-# ==========================================
-# 5. Gradio 介面建構
+# 3. Gradio 介面建構 (移除前兩頁)
 # ==========================================
 
 with gr.Blocks(title="AI 心情食堂") as demo:
-    gr.HTML(f"<style>{custom_css}</style>")
-    quiz_state = gr.State(value={})
-
-    # Page 1
-    with gr.Column(visible=True) as page_1_col:
-        gr.Markdown("### 🔮 第一題：直覺測試")
-        gr.Image(value=img_path_1, type="filepath", label="請觀察圖片", height=300)
-        gr.Markdown("**問題：請觀察上方圖片，如果是你，你會怎麼做？**")
-        radio_q1 = gr.Radio(choices=list(q1_options_map.keys()), label="請選擇", elem_classes="vertical-radio")
-        btn_q1_next = gr.Button("下一頁 ➔", interactive=False)
-
-    # Page 2
-    with gr.Column(visible=False) as page_2_col:
-        gr.Markdown("### 🔮 第二題：魔法適性")
-        gr.Image(value=img_path_2, type="filepath", label="請觀察圖片", height=300)
-        gr.Markdown("**問題：身為魔導士的你，會選擇哪一個法術？**")
-        radio_q2 = gr.Radio(choices=list(q2_options_map.keys()), label="請選擇", elem_classes="vertical-radio")
-        btn_q2_finish = gr.Button("完成測驗 ➔", interactive=False)
-
-    # Page 3 (Main)
-    with gr.Column(visible=False) as main_app_col:
+    
+    # 只有一個主應用程式 Column，並設定為預設顯示
+    with gr.Column(visible=True) as main_app_col: 
         gr.Markdown(f"## 🍱 AI 心情食堂導航")
-        gr.Markdown("測驗結束！讓 AI 看看你的臉色，幫你決定吃什麼！")
+        gr.Markdown("請看著鏡頭，讓 AI 幫你判斷今天的心情分數！")
 
         with gr.Row():
             # 左側：輸入區
             with gr.Column(scale=1):
                 
-                # ★ 新增：情緒偵測區塊
+                # 情緒偵測區塊 (維持不變)
                 gr.Markdown("### 📸 步驟 1：心情偵測 (選用)")
                 with gr.Group():
                     webcam_input = gr.Image(sources=["webcam"], label="看著鏡頭", streaming=True)
                     detect_btn = gr.Button("📸 偵測心情並自動填入 👇", variant="secondary")
                 
                 gr.Markdown("### 📝 步驟 2：確認與補充")
-                # score_input 會被 detect_btn 更新
                 score_input = gr.Radio(
                     ["1 (心情差)", "2 (不太好)", "3 (普通)", "4 (不錯)", "5 (超棒)"], 
                     label="1. 心情分數 (可手動修改)", 
@@ -377,30 +297,29 @@ with gr.Blocks(title="AI 心情食堂") as demo:
                 diary_input = gr.Textbox(lines=3, label="3. 心情日記 (選填)", placeholder="例如：今天被老闆罵了，想吃點好料的...")
                 
                 debug_mode_btn = gr.Checkbox(label="🔧 顯示 Prompt 除錯資訊", value=False)
-                submit_btn = gr.Button("🚀 送出給 Agent", variant="primary")
+                # 移除了 quiz_state
+                submit_btn = gr.Button("🚀 送出給 Agent", variant="primary") 
 
-            # 右側：輸出區
+            # 右側：輸出區 (維持不變)
             with gr.Column(scale=1):
                 agent_output = gr.Markdown(label="AI 回應")
                 image_output = gr.Image(label="AI 推薦美食圖", type="pil", width=400)
                 map_output = gr.HTML(label="地圖導航")
 
-    # Events
-    radio_q1.change(fn=handle_q1_change, inputs=[radio_q1, quiz_state], outputs=[quiz_state, btn_q1_next])
-    btn_q1_next.click(fn=go_to_page_2, outputs=[page_1_col, page_2_col])
-    radio_q2.change(fn=handle_q2_change, inputs=[radio_q2, quiz_state], outputs=[quiz_state, btn_q2_finish])
-    btn_q2_finish.click(fn=go_to_main_app, outputs=[page_2_col, main_app_col])
-
-    # ★ 新增事件：點擊偵測按鈕 -> 執行 DeepFace -> 更新 score_input
+    # Events (只保留情緒偵測和提交)
+    
+    # 點擊偵測按鈕 -> 執行 DeepFace -> 更新 score_input
     detect_btn.click(
         fn=detect_emotion_and_map,
         inputs=[webcam_input],
         outputs=[webcam_input, score_input] # 同時更新圖片(畫字)和分數(Radio)
     )
 
+    # 提交按鈕
+    # 移除了 quiz_state 的輸入
     submit_btn.click(
         fn=mood_agent_logic,
-        inputs=[score_input, food_input, diary_input, quiz_state, debug_mode_btn],
+        inputs=[score_input, food_input, diary_input, debug_mode_btn],
         outputs=[agent_output, image_output, map_output]
     )